Only throw BatchUpdateException for statements for read

Author: zhicwu

clickhouse-jdbc/src/main/java/com/clickhouse/jdbc/internal/ClickHouseStatementImpl.java

@@ -772,7 +772,7 @@ public long[] executeLargeBatch() throws SQLException {
             int i = 0;
             for (ClickHouseSqlStatement s : batchStmts) {
                 try (ClickHouseResponse r = executeStatement(s, null, null, null); ResultSet rs = updateResult(s, r)) {
-                    if (rs != null) {
+                    if (rs != null && s.isQuery()) {
                         throw SqlExceptionUtils.queryInBatchError(results);
                     }
                     results[i] = currentUpdateCount <= 0L ? 0L : currentUpdateCount;

clickhouse-jdbc/src/main/java/com/clickhouse/jdbc/internal/SqlBasedPreparedStatement.java

@@ -161,7 +161,7 @@ protected long[] executeAny(boolean asBatch) throws SQLException {
             long rows = 0L;
             try {
                 r = executeStatement(builder.toString(), reparse);
-                if (updateResult(parsedStmt, r) != null && asBatch) {
+                if (updateResult(parsedStmt, r) != null && asBatch && parsedStmt.isQuery()) {
                     throw SqlExceptionUtils.queryInBatchError(results);
                 }
                 rows = r.getSummary().getWrittenRows();
@@ -208,7 +208,7 @@ protected long[] executeAny(boolean asBatch) throws SQLException {
                     preparedQuery.apply(builder, params);
                     try {
                         r = executeStatement(builder.toString(), reparse);
-                        if (updateResult(parsedStmt, r) != null && asBatch) {
+                        if (updateResult(parsedStmt, r) != null && asBatch && parsedStmt.isQuery()) {
                             throw SqlExceptionUtils.queryInBatchError(results);
                         }
                         int count = getUpdateCount();

clickhouse-jdbc/src/main/java/com/clickhouse/jdbc/internal/TableBasedPreparedStatement.java

@@ -101,7 +101,7 @@ public long[] executeAny(boolean asBatch) throws SQLException {
             for (List<ClickHouseExternalTable> list : batch) {
                 try (ClickHouseResponse r = executeStatement(sql, null, list, null);
                         ResultSet rs = updateResult(parsedStmt, r)) {
-                    if (asBatch && rs != null) {
+                    if (asBatch && rs != null && parsedStmt.isQuery()) {
                         throw SqlExceptionUtils.queryInBatchError(results);
                     }
                     long rows = getLargeUpdateCount();

clickhouse-jdbc/src/test/java/com/clickhouse/jdbc/ClickHousePreparedStatementTest.java

@@ -694,6 +694,24 @@ public void testInsertQueryDateTime64() throws SQLException {
         }
     }
 
+    @Test(groups = "integration")
+    public void testBatchDdl() throws SQLException {
+        Properties props = new Properties();
+        try (ClickHouseConnection conn = newConnection(props)) {
+            try (PreparedStatement stmt = conn.prepareStatement(
+                    "drop table if exists test_batch_dll_on_cluster on cluster test_shard_localhost")) {
+                stmt.addBatch();
+                stmt.addBatch();
+                Assert.assertEquals(stmt.executeBatch(), new int[] { 0, 0 });
+            }
+
+            try (PreparedStatement stmt = conn.prepareStatement("select 1")) {
+                stmt.addBatch();
+                Assert.assertThrows(BatchUpdateException.class, () -> stmt.executeBatch());
+            }
+        }
+    }
+
     @Test(groups = "integration")
     public void testBatchInsert() throws SQLException {
         try (ClickHouseConnection conn = newConnection(new Properties());

clickhouse-jdbc/src/test/java/com/clickhouse/jdbc/ClickHouseStatementTest.java

@@ -68,6 +68,22 @@ private Object[][] getConnectionProperties() {
                 new Object[] { emptyProps }, new Object[] { sessionProps } };
     }
 
+    @Test(groups = "integration")
+    public void testBatchUpdate() throws SQLException {
+        Properties props = new Properties();
+        try (ClickHouseConnection conn = newConnection(props); ClickHouseStatement stmt = conn.createStatement()) {
+            stmt.addBatch("drop table if exists test_batch_dll_on_cluster on cluster test_shard_localhost");
+            stmt.addBatch(
+                    "create table if not exists test_batch_dll_on_cluster on cluster test_shard_localhost(a Int64) Engine=MergeTree order by a;"
+                            + "drop table if exists test_batch_dll_on_cluster on cluster test_shard_localhost;");
+            Assert.assertEquals(stmt.executeBatch(), new int[] { 0, 0, 0 });
+
+            stmt.addBatch("drop table if exists test_batch_queries");
+            stmt.addBatch("select 1");
+            Assert.assertThrows(BatchUpdateException.class, () -> stmt.executeBatch());
+        }
+    }
+
     @Test(groups = "integration")
     public void testBitmap64() throws SQLException {
         Properties props = new Properties();