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 bc990178..fc88bb19 100644 --- a/databend-jdbc/src/main/java/com/databend/jdbc/DatabendConnection.java +++ b/databend-jdbc/src/main/java/com/databend/jdbc/DatabendConnection.java @@ -702,12 +702,13 @@ public void uploadStream(String stageName, String destPrefix, InputStream inputS logger.info("upload cost time: " + (uploadEndTime - uploadStartTime) / 1000000.0 + "ms"); } } - } catch (JsonProcessingException e) { + } catch (RuntimeException e) { System.out.println(e.getMessage()); // For datax batch insert test, do not throw exception throw new SQLException(e); } catch (IOException e) { logger.warning("failed to upload input stream, file size is:" + fileSize / 1024.0 + e.getMessage()); + throw new SQLException(e); } } @@ -721,7 +722,7 @@ public InputStream downloadStream(String stageName, String sourceFileName, boole Headers h = ctx.getHeaders(); String presignUrl = ctx.getUrl(); return cli.presignDownloadStream(h, presignUrl); - } catch (JsonProcessingException e) { + } catch (RuntimeException e) { throw new SQLException(e); } } diff --git a/databend-jdbc/src/main/java/com/databend/jdbc/PresignContext.java b/databend-jdbc/src/main/java/com/databend/jdbc/PresignContext.java index 071a58cf..2a69ff07 100644 --- a/databend-jdbc/src/main/java/com/databend/jdbc/PresignContext.java +++ b/databend-jdbc/src/main/java/com/databend/jdbc/PresignContext.java @@ -40,7 +40,7 @@ public static void dropStageIfExists(DatabendConnection connection, String stage } public static PresignContext getPresignContext(DatabendConnection connection, PresignMethod method, String stageName, String fileName) - throws SQLException, JsonProcessingException { + throws SQLException { requireNonNull(connection, "connection is null"); requireNonNull(method, "method is null"); Statement statement = connection.createStatement();