fix

databendlabs · Aug 1, 2024 · ba4c50a · ba4c50a
1 parent 326e7c3
commit ba4c50a
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/databend-jdbc/src/main/java/com/databend/jdbc/cloud/DatabendPresignClientV1.java b/databend-jdbc/src/main/java/com/databend/jdbc/cloud/DatabendPresignClientV1.java
@@ -233,6 +233,7 @@ public MediaType contentType() {
     @Override
     public long contentLength() {
         return fileSize; // return the actual file size
+//        return inputStream.available() == 0 ? -1 : inputStream.available();
     }
 
     @Override

diff --git a/databend-jdbc/src/test/java/com/databend/jdbc/TestFileTransfer.java b/databend-jdbc/src/test/java/com/databend/jdbc/TestFileTransfer.java
@@ -57,6 +57,7 @@ private Connection createConnection()
             throws SQLException {
         String url = "jdbc:databend://localhost:8000/default";
         return DriverManager.getConnection(url, "databend", "databend");
+
     }
 
     private Connection createConnection(boolean presignDisabled) throws SQLException {
@@ -84,6 +85,28 @@ private String generateRandomCSV(int lines) {
         return csvPath;
     }
 
+    private String generateLargeCSV() {
+        String tmpDir = System.getProperty("java.io.tmpdir");
+        String csvPath = tmpDir + "/large_test.csv";
+        long fileSizeInBytes = 0;
+        File f = new File(csvPath);
+        try {
+            FileWriter writer = new FileWriter(f);
+            while (fileSizeInBytes < 2L * 1024 * 1024 * 1024) { // 2GB
+                for (int i = 0; i < 1000; i++) { // write 1000 lines at a time
+                    int num = (int) (Math.random() * 1000);
+                    writer.write("a,b,c," + num + "\n");
+                }
+                writer.flush();
+                fileSizeInBytes = f.length();
+            }
+            writer.close();
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        }
+        return csvPath;
+    }
+
     private String generateRandomCSVComplex(int lines) {
         if (lines <= 0) {
             return "";