Feature/td 5265 (#6878)

* [TD-5265]<feature>: add test case for executeBatch in Statement

* [TD-5265]<feature>: support continue process in Statement.executeBatch method

* change

* change

* change

* change

src/connector/jdbc/src/main/java/com/taosdata/jdbc/AbstractStatement.java

@@ -2,6 +2,7 @@ package com.taosdata.jdbc;
 
 import java.sql.*;
 import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.List;
 
 public abstract class AbstractStatement extends WrapperImpl implements Statement {
@@ -196,13 +197,44 @@ public abstract class AbstractStatement extends WrapperImpl implements Statement
         if (batchedArgs == null || batchedArgs.isEmpty())
             throw TSDBError.createSQLException(TSDBErrorNumbers.ERROR_BATCH_IS_EMPTY);
 
+        String clientInfo = getConnection().getClientInfo(TSDBDriver.PROPERTY_KEY_BATCH_ERROR_IGNORE);
+        boolean batchErrorIgnore = clientInfo == null ? TSDBConstants.DEFAULT_BATCH_ERROR_IGNORE : Boolean.parseBoolean(clientInfo);
+
+        if (batchErrorIgnore) {
+            return executeBatchIgnoreException();
+        }
+        return executeBatchThrowException();
+    }
+
+    private int[] executeBatchIgnoreException() {
+        return batchedArgs.stream().mapToInt(sql -> {
+            try {
+                boolean isSelect = execute(sql);
+                if (isSelect) {
+                    return SUCCESS_NO_INFO;
+                } else {
+                    return getUpdateCount();
+                }
+            } catch (SQLException e) {
+                return EXECUTE_FAILED;
+            }
+        }).toArray();
+    }
+
+    private int[] executeBatchThrowException() throws BatchUpdateException {
         int[] res = new int[batchedArgs.size()];
         for (int i = 0; i < batchedArgs.size(); i++) {
-            boolean isSelect = execute(batchedArgs.get(i));
-            if (isSelect) {
-                res[i] = SUCCESS_NO_INFO;
-            } else {
-                res[i] = getUpdateCount();
+            try {
+                boolean isSelect = execute(batchedArgs.get(i));
+                if (isSelect) {
+                    res[i] = SUCCESS_NO_INFO;
+                } else {
+                    res[i] = getUpdateCount();
+                }
+            } catch (SQLException e) {
+                String reason = e.getMessage();
+                int[] updateCounts = Arrays.copyOfRange(res, 0, i);
+                throw new BatchUpdateException(reason, updateCounts, e);
             }
         }
         return res;

src/connector/jdbc/src/main/java/com/taosdata/jdbc/TSDBConstants.java

@@ -74,6 +74,8 @@ public abstract class TSDBConstants {
 
     public static final String DEFAULT_PRECISION = "ms";
 
+    public static final boolean DEFAULT_BATCH_ERROR_IGNORE = false;
+
     public static int typeName2JdbcType(String type) {
         switch (type.toUpperCase()) {
             case "TIMESTAMP":

src/connector/jdbc/src/main/java/com/taosdata/jdbc/TSDBDriver.java

@@ -100,6 +100,11 @@ public class TSDBDriver extends AbstractDriver {
      */
     public static final String PROPERTY_KEY_TIMESTAMP_FORMAT = "timestampFormat";
 
+    /**
+     * continue process commands in executeBatch
+     */
+    public static final String PROPERTY_KEY_BATCH_ERROR_IGNORE = "batchErrorIgnore";
+
     private TSDBDatabaseMetaData dbMetaData = null;
 
     static {

src/connector/jdbc/src/test/java/com/taosdata/jdbc/cases/BatchErrorIgnoreTest.java

+package com.taosdata.jdbc.cases;
+
+import org.junit.*;
+
+import java.sql.*;
+import java.util.stream.IntStream;
+
+public class BatchErrorIgnoreTest {
+
+    private static final String host = "127.0.0.1";
+
+    @Test
+    public void batchErrorThrowException() throws SQLException {
+        // given
+        Connection conn = DriverManager.getConnection("jdbc:TAOS://" + host + ":6030/?user=root&password=taosdata");
+
+        // when
+        try (Statement stmt = conn.createStatement()) {
+            IntStream.range(1, 6).mapToObj(i -> "insert into test.t" + i + " values(now, " + i + ")").forEach(sql -> {
+                try {
+                    stmt.addBatch(sql);
+                } catch (SQLException e) {
+                    e.printStackTrace();
+                }
+            });
+            stmt.addBatch("insert into t11 values(now, 11)");
+            IntStream.range(6, 11).mapToObj(i -> "insert into test.t" + i + " values(now, " + i + "),(now + 1s, " + (10 * i) + ")").forEach(sql -> {
+                try {
+                    stmt.addBatch(sql);
+                } catch (SQLException e) {
+                    e.printStackTrace();
+                }
+            });
+            stmt.addBatch("select count(*) from test.weather");
+
+            stmt.executeBatch();
+        } catch (BatchUpdateException e) {
+            int[] updateCounts = e.getUpdateCounts();
+            Assert.assertEquals(5, updateCounts.length);
+            Assert.assertEquals(1, updateCounts[0]);
+            Assert.assertEquals(1, updateCounts[1]);
+            Assert.assertEquals(1, updateCounts[2]);
+            Assert.assertEquals(1, updateCounts[3]);
+            Assert.assertEquals(1, updateCounts[4]);
+        }
+
+    }
+
+    @Test
+    public void batchErrorIgnore() throws SQLException {
+        // given
+        Connection conn = DriverManager.getConnection("jdbc:TAOS://" + host + ":6030/?user=root&password=taosdata&batchErrorIgnore=true");
+
+        // when
+        int[] results = null;
+        try (Statement stmt = conn.createStatement()) {
+            IntStream.range(1, 6).mapToObj(i -> "insert into test.t" + i + " values(now, " + i + ")").forEach(sql -> {
+                try {
+                    stmt.addBatch(sql);
+                } catch (SQLException e) {
+                    e.printStackTrace();
+                }
+            });
+            stmt.addBatch("insert into t11 values(now, 11)");
+            IntStream.range(6, 11).mapToObj(i -> "insert into test.t" + i + " values(now, " + i + "),(now + 1s, " + (10 * i) + ")").forEach(sql -> {
+                try {
+                    stmt.addBatch(sql);
+                } catch (SQLException e) {
+                    e.printStackTrace();
+                }
+            });
+            stmt.addBatch("select count(*) from test.weather");
+
+            results = stmt.executeBatch();
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+
+        // then
+        assert results != null;
+        Assert.assertEquals(12, results.length);
+        Assert.assertEquals(1, results[0]);
+        Assert.assertEquals(1, results[1]);
+        Assert.assertEquals(1, results[2]);
+        Assert.assertEquals(1, results[3]);
+        Assert.assertEquals(1, results[4]);
+        Assert.assertEquals(Statement.EXECUTE_FAILED, results[5]);
+        Assert.assertEquals(2, results[6]);
+        Assert.assertEquals(2, results[7]);
+        Assert.assertEquals(2, results[8]);
+        Assert.assertEquals(2, results[9]);
+        Assert.assertEquals(2, results[10]);
+        Assert.assertEquals(Statement.SUCCESS_NO_INFO, results[11]);
+    }
+
+    @Before
+    public void before() {
+        try {
+            Connection conn = DriverManager.getConnection("jdbc:TAOS://" + host + ":6030/?user=root&password=taosdata");
+            Statement stmt = conn.createStatement();
+            stmt.execute("use test");
+            stmt.execute("drop table if exists weather");
+            stmt.execute("create table weather (ts timestamp, f1 float) tags(t1 int)");
+            IntStream.range(1, 11).mapToObj(i -> "create table t" + i + " using weather tags(" + i + ")").forEach(sql -> {
+                try {
+                    stmt.execute(sql);
+                } catch (SQLException e) {
+                    e.printStackTrace();
+                }
+            });
+            stmt.close();
+            conn.close();
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+    }
+
+    @BeforeClass
+    public static void beforeClass() {
+        try {
+            Connection conn = DriverManager.getConnection("jdbc:TAOS://" + host + ":6030/?user=root&password=taosdata");
+            Statement stmt = conn.createStatement();
+            stmt.execute("drop database if exists test");
+            stmt.execute("create database if not exists test");
+            stmt.close();
+            conn.close();
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+    }
+
+    @AfterClass
+    public static void afterClass() {
+        try {
+            Connection conn = DriverManager.getConnection("jdbc:TAOS://" + host + ":6030/?user=root&password=taosdata");
+            Statement stmt = conn.createStatement();
+            stmt.execute("drop database if exists test");
+            stmt.close();
+            conn.close();
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+    }
+}