diff --git a/databend-client/src/test/java/com/databend/client/TestClientIT.java b/databend-client/src/test/java/com/databend/client/TestClientIT.java index 537e6a80..58784c4c 100644 --- a/databend-client/src/test/java/com/databend/client/TestClientIT.java +++ b/databend-client/src/test/java/com/databend/client/TestClientIT.java @@ -76,7 +76,7 @@ public void testConnectionRefused() { @Test(groups = {"it"}) public void testBasicQueryIDHeader() { OkHttpClient client = new OkHttpClient.Builder().addInterceptor(OkHttpUtils.basicAuthInterceptor("databend", "databend")).build(); - String expectedUUID = UUID.randomUUID().toString(); + String expectedUUID = UUID.randomUUID().toString().replace("-",""); AtomicReference lastNodeID = new AtomicReference<>(); Map additionalHeaders = new HashMap<>(); @@ -85,7 +85,7 @@ public void testBasicQueryIDHeader() { DatabendClient cli = new DatabendClientV1(client, "select 1", settings, null, lastNodeID); Assert.assertEquals(cli.getAdditionalHeaders().get(X_Databend_Query_ID), expectedUUID); - String expectedUUID1 = UUID.randomUUID().toString(); + String expectedUUID1 = UUID.randomUUID().toString().replace("-", ""); Map additionalHeaders1 = new HashMap<>(); additionalHeaders1.put(X_Databend_Query_ID, expectedUUID1); ClientSettings settings1 = new ClientSettings(DATABEND_HOST, DatabendSession.createDefault(), DEFAULT_QUERY_TIMEOUT, DEFAULT_CONNECTION_TIMEOUT, DEFAULT_SOCKET_TIMEOUT, PaginationOptions.defaultPaginationOptions(), additionalHeaders1, null, DEFAULT_RETRY_ATTEMPTS); @@ -104,7 +104,7 @@ public void testBasicQueryIDHeader() { @Test(groups = {"it"}) public void testDiscoverNodes() { OkHttpClient client = new OkHttpClient.Builder().addInterceptor(OkHttpUtils.basicAuthInterceptor("databend", "databend")).build(); - String expectedUUID = UUID.randomUUID().toString(); + String expectedUUID = UUID.randomUUID().toString().replace("-", ""); Map additionalHeaders = new HashMap<>(); additionalHeaders.put(X_Databend_Query_ID, expectedUUID); @@ -119,7 +119,7 @@ public void testDiscoverNodes() { @Test(groups = {"it"}) public void testDiscoverNodesUnSupported() { OkHttpClient client = new OkHttpClient.Builder().addInterceptor(OkHttpUtils.basicAuthInterceptor("databend", "databend")).build(); - String expectedUUID = UUID.randomUUID().toString(); + String expectedUUID = UUID.randomUUID().toString().replace("-", ""); Map additionalHeaders = new HashMap<>(); additionalHeaders.put(X_Databend_Query_ID, expectedUUID); diff --git a/databend-jdbc/src/main/java/com/databend/jdbc/DatabendConnection.java b/databend-jdbc/src/main/java/com/databend/jdbc/DatabendConnection.java index a326c652..2891f19c 100644 --- a/databend-jdbc/src/main/java/com/databend/jdbc/DatabendConnection.java +++ b/databend-jdbc/src/main/java/com/databend/jdbc/DatabendConnection.java @@ -706,7 +706,7 @@ DatabendClient startQueryWithFailover(String sql, StageAttachment attach) throws for (int attempt = 0; attempt <= maxRetries; attempt++) { try { - String queryId = UUID.randomUUID().toString(); + String queryId = UUID.randomUUID().toString().replace("-", "");; String candidateHost = selectHostForQuery(queryId); // configure the client settings diff --git a/databend-jdbc/src/main/java/com/databend/jdbc/DatabendPreparedStatement.java b/databend-jdbc/src/main/java/com/databend/jdbc/DatabendPreparedStatement.java index 13db41d8..b2929eb5 100644 --- a/databend-jdbc/src/main/java/com/databend/jdbc/DatabendPreparedStatement.java +++ b/databend-jdbc/src/main/java/com/databend/jdbc/DatabendPreparedStatement.java @@ -164,7 +164,7 @@ private DatabendCopyParams uploadBatchesForCopyInto() throws SQLException { File saved = batchInsertUtils.get().saveBatchToCSV(batchValues); try (FileInputStream fis = new FileInputStream(saved);) { DatabendConnection c = (DatabendConnection) getConnection(); - String uuid = UUID.randomUUID().toString(); + String uuid = UUID.randomUUID().toString().replace("-", ""); // format %Y/%m/%d/%H/%M/%S/fileName.csv String stagePrefix = String.format("%s/%s/%s/%s/%s/%s/%s/", LocalDateTime.now().getYear(), @@ -205,7 +205,7 @@ private StageAttachment uploadBatches() throws SQLException { File saved = batchInsertUtils.get().saveBatchToCSV(batchValues); try (FileInputStream fis = new FileInputStream(saved);) { DatabendConnection c = (DatabendConnection) getConnection(); - String uuid = UUID.randomUUID().toString(); + String uuid = UUID.randomUUID().toString().replace("-", ""); // format %Y/%m/%d/%H/%M/%S/fileName.csv String stagePrefix = String.format("%s/%s/%s/%s/%s/%s/%s/", LocalDateTime.now().getYear(), diff --git a/databend-jdbc/src/main/java/com/databend/jdbc/StatementInfoWrapper.java b/databend-jdbc/src/main/java/com/databend/jdbc/StatementInfoWrapper.java index 5814b119..fbd992b3 100644 --- a/databend-jdbc/src/main/java/com/databend/jdbc/StatementInfoWrapper.java +++ b/databend-jdbc/src/main/java/com/databend/jdbc/StatementInfoWrapper.java @@ -30,7 +30,7 @@ public class StatementInfoWrapper { * @return the statement that will be sent to the server */ public static StatementInfoWrapper of(@NonNull RawStatement rawStatement) { - return of(rawStatement, UUID.randomUUID().toString()); + return of(rawStatement, UUID.randomUUID().toString().replace("-", "")); } /** diff --git a/databend-jdbc/src/main/java/com/databend/jdbc/StatementUtil.java b/databend-jdbc/src/main/java/com/databend/jdbc/StatementUtil.java index 8f80f9a9..4426722a 100644 --- a/databend-jdbc/src/main/java/com/databend/jdbc/StatementUtil.java +++ b/databend-jdbc/src/main/java/com/databend/jdbc/StatementUtil.java @@ -281,7 +281,7 @@ public List replaceParameterMarksWithValues(@NonNull Map additionalParams = subQuery.getStatementType() == StatementType.PARAM_SETTING ? ((SetParamRawStatement) subQuery).getAdditionalProperty() : null; - subQueries.add(new StatementInfoWrapper(subQueryWithParams, UUID.randomUUID().toString(), + subQueries.add(new StatementInfoWrapper(subQueryWithParams, UUID.randomUUID().toString().replace("-", ""), subQuery.getStatementType(), additionalParams, subQuery)); } diff --git a/databend-jdbc/src/main/java/com/databend/jdbc/parser/BatchInsertUtils.java b/databend-jdbc/src/main/java/com/databend/jdbc/parser/BatchInsertUtils.java index 835027e7..cbbb551a 100644 --- a/databend-jdbc/src/main/java/com/databend/jdbc/parser/BatchInsertUtils.java +++ b/databend-jdbc/src/main/java/com/databend/jdbc/parser/BatchInsertUtils.java @@ -87,7 +87,7 @@ public String[] getValues() { public File saveBatchToCSV(List values) { // get a temporary directory - String id = UUID.randomUUID().toString(); + String id = UUID.randomUUID().toString().replace("-",""); File tempDir = new File(System.getProperty("java.io.tmpdir")); File tempFile = new File(tempDir, "databend_batch_insert_" + id + ".csv"); return saveBatchToCSV(values, tempFile);