添加 commitIfTrue。

pom.xml

@@ -21,6 +21,12 @@
             <version>0.1.0-SNAPSHOT</version>
         </dependency>
 
+        <dependency>
+            <groupId>org.apache.commons</groupId>
+            <artifactId>commons-lang3</artifactId>
+            <version>3.16.0</version>
+        </dependency>
+
         <dependency>
             <groupId>org.junit.jupiter</groupId>
             <artifactId>junit-jupiter-api</artifactId>

src/main/java/xyz/zhouxy/jdbc/ParamBuilder.java

@@ -0,0 +1,58 @@
+package xyz.zhouxy.jdbc;
+
+import java.util.Arrays;
+import java.util.Collection;
+import java.util.Collections;
+import java.util.List;
+import java.util.Optional;
+import java.util.OptionalDouble;
+import java.util.OptionalInt;
+import java.util.OptionalLong;
+import java.util.function.Function;
+import java.util.stream.Collectors;
+
+import com.google.common.base.Preconditions;
+
+import xyz.zhouxy.plusone.commons.collection.CollectionTools;
+import xyz.zhouxy.plusone.commons.util.ArrayTools;
+import xyz.zhouxy.plusone.commons.util.OptionalTools;
+
+public class ParamBuilder {
+    public static final Object[] EMPTY_OBJECT_ARRAY = {};
+
+    public static Object[] buildParams(final Object... params) {
+        if (ArrayTools.isNullOrEmpty(params)) {
+            return EMPTY_OBJECT_ARRAY;
+        }
+        return Arrays.stream(params)
+                .map(param -> {
+                    if (param instanceof Optional) {
+                        return OptionalTools.orElseNull((Optional<?>) param);
+                    }
+                    if (param instanceof OptionalInt) {
+                        return OptionalTools.toInteger(((OptionalInt) param));
+                    }
+                    if (param instanceof OptionalLong) {
+                        return OptionalTools.toLong(((OptionalLong) param));
+                    }
+                    if (param instanceof OptionalDouble) {
+                        return OptionalTools.toDouble(((OptionalDouble) param));
+                    }
+                    return param;
+                })
+                .toArray();
+    }
+
+    public static <T> List<Object[]> buildBatchParams(final Collection<T> c, final Function<T, Object[]> func) {
+        Preconditions.checkNotNull(c, "The collection can not be null.");
+        Preconditions.checkNotNull(func, "The func can not be null.");
+        if (CollectionTools.isEmpty(c)) {
+            return Collections.emptyList();
+        }
+        return c.stream().map(func).collect(Collectors.toList());
+    }
+
+    private ParamBuilder() {
+        throw new IllegalStateException("Utility class");
+    }
+}

src/main/java/xyz/zhouxy/jdbc/ResultMap.java

@@ -16,15 +16,12 @@
 
 package xyz.zhouxy.jdbc;
 
-import com.google.common.annotations.Beta;
-
 import java.sql.ResultSet;
 import java.sql.ResultSetMetaData;
 import java.sql.SQLException;
 import java.util.HashMap;
 import java.util.Map;
 
-@Beta
 @FunctionalInterface
 public interface ResultMap<T> {
     T map(ResultSet rs, int rowNumber) throws SQLException;

src/main/java/xyz/zhouxy/jdbc/SimpleJdbcTemplate.java

@@ -25,27 +25,19 @@ import java.sql.Statement;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Collection;
-import java.util.Collections;
 import java.util.List;
 import java.util.Map;
 import java.util.Optional;
 import java.util.OptionalDouble;
 import java.util.OptionalInt;
 import java.util.OptionalLong;
-import java.util.function.Function;
-import java.util.stream.Collectors;
-
 import javax.annotation.Nonnull;
 
-import com.google.common.annotations.Beta;
 import com.google.common.base.Preconditions;
 import com.google.common.collect.Lists;
 
-import xyz.zhouxy.plusone.commons.collection.CollectionTools;
-import xyz.zhouxy.plusone.commons.util.ArrayTools;
 import xyz.zhouxy.plusone.commons.util.OptionalTools;
 
-@Beta
 public class SimpleJdbcTemplate {
 
     public static JdbcExecutor connect(final Connection conn) {
@@ -86,7 +78,7 @@ public class SimpleJdbcTemplate {
 
         private final Connection conn;
 
-        public JdbcExecutor(Connection conn) {
+        private JdbcExecutor(Connection conn) {
             this.conn = conn;
         }
 
@@ -109,46 +101,46 @@ public class SimpleJdbcTemplate {
             return query(sql, params, resultMap).stream().findFirst();
         }
 
-        public List<Map<String, Object>> query(String sql, Object... params) throws SQLException {
+        public List<Map<String, Object>> query(String sql, Object[] params) throws SQLException {
             return query(sql, params, ResultMap.mapResultMap);
         }
 
-        public Optional<Map<String, Object>> queryFirst(String sql, Object... params) throws SQLException {
+        public Optional<Map<String, Object>> queryFirst(String sql, Object[] params) throws SQLException {
             return queryFirst(sql, params, ResultMap.mapResultMap);
         }
 
-        public List<DbRecord> queryToRecordList(String sql, Object... params) throws SQLException {
+        public List<DbRecord> queryToRecordList(String sql, Object[] params) throws SQLException {
             return query(sql, params, ResultMap.recordResultMap);
         }
 
-        public Optional<DbRecord> queryFirstRecord(String sql, Object... params) throws SQLException {
+        public Optional<DbRecord> queryFirstRecord(String sql, Object[] params) throws SQLException {
             return queryFirst(sql, params, ResultMap.recordResultMap);
         }
 
-        public Optional<String> queryToString(String sql, Object... params) throws SQLException {
+        public Optional<String> queryToString(String sql, Object[] params) throws SQLException {
             return queryFirst(sql, params, (rs, rowNumber) -> rs.getString(1));
         }
 
-        public OptionalInt queryToInt(String sql, Object... params) throws SQLException {
+        public OptionalInt queryToInt(String sql, Object[] params) throws SQLException {
             Optional<Integer> result = queryFirst(sql, params, (rs, rowNumber) -> rs.getInt(1));
             return OptionalTools.toOptionalInt(result);
         }
 
-        public OptionalLong queryToLong(String sql, Object... params) throws SQLException {
+        public OptionalLong queryToLong(String sql, Object[] params) throws SQLException {
             Optional<Long> result = queryFirst(sql, params, (rs, rowNumber) -> rs.getLong(1));
             return OptionalTools.toOptionalLong(result);
         }
 
-        public OptionalDouble queryToDouble(String sql, Object... params) throws SQLException {
+        public OptionalDouble queryToDouble(String sql, Object[] params) throws SQLException {
             Optional<Double> result = queryFirst(sql, params, (rs, rowNumber) -> rs.getDouble(1));
             return OptionalTools.toOptionalDouble(result);
         }
 
-        public Optional<BigDecimal> queryToBigDecimal(String sql, Object... params) throws SQLException {
+        public Optional<BigDecimal> queryToBigDecimal(String sql, Object[] params) throws SQLException {
             return queryFirst(sql, params, (rs, rowNumber) -> rs.getBigDecimal(1));
         }
 
-        public int update(String sql, Object... params) throws SQLException {
+        public int update(String sql, Object[] params) throws SQLException {
             try (PreparedStatement stmt = this.conn.prepareStatement(sql)) {
                 fillStatement(stmt, params);
                 return stmt.executeUpdate();
@@ -185,7 +177,7 @@ public class SimpleJdbcTemplate {
             }
         }
 
-        public int[] batchUpdate(String sql, Collection<Object[]> params, int batchSize) throws SQLException {
+        public List<int[]> batchUpdate(String sql, Collection<Object[]> params, int batchSize) throws SQLException {
             int executeCount = params.size() / batchSize;
             executeCount = (params.size() % batchSize == 0) ? executeCount : (executeCount + 1);
             List<int[]> result = Lists.newArrayListWithCapacity(executeCount);
@@ -194,9 +186,7 @@ public class SimpleJdbcTemplate {
                 int i = 0;
                 for (Object[] ps : params) {
                     i++;
-                    for (int j = 0; j < ps.length; j++) {
+                    fillStatement(stmt, ps);
-                        stmt.setObject(j + 1, ps[j]);
-                    }
                     stmt.addBatch();
                     if (i % batchSize == 0 || i >= params.size()) {
                         int[] n = stmt.executeBatch();
@@ -204,16 +194,17 @@ public class SimpleJdbcTemplate {
                         stmt.clearBatch();
                     }
                 }
-                return ArrayTools.concatIntArray(result);
+                return result;
             }
         }
 
-        public <E extends Exception> void tx(final IAtom<E> atom) throws SQLException, E {
+        public <E extends Exception> void executeTransaction(@Nonnull final DbOperations<E> operations)
-            Preconditions.checkNotNull(atom, "Atom can not be null.");
+                throws SQLException, E {
+            Preconditions.checkNotNull(operations, "Operations can not be null.");
             final boolean autoCommit = this.conn.getAutoCommit();
             try {
                 this.conn.setAutoCommit(false);
-                atom.execute(this);
+                operations.execute(this);
                 this.conn.commit();
             }
             catch (Exception e) {
@@ -225,9 +216,36 @@ public class SimpleJdbcTemplate {
             }
         }
 
+        public <E extends Exception> void commitIfTrue(@Nonnull final PredicateWithThrowable<E> operations)
+                throws SQLException, E {
+            Preconditions.checkNotNull(operations, "Operations can not be null.");
+            final boolean autoCommit = this.conn.getAutoCommit();
+            try {
+                this.conn.setAutoCommit(false);
+                if (operations.test(this)) {
+                    this.conn.commit();
+                }
+                else {
+                    this.conn.rollback();
+                }
+            }
+            catch (Exception e) {
+                this.conn.rollback();
+                throw e;
+            }
+            finally {
+                this.conn.setAutoCommit(autoCommit);
+            }
+        }
+
         @FunctionalInterface
-        public interface IAtom<E extends Exception> {
+        public interface DbOperations<E extends Exception> {
-            void execute(JdbcExecutor jdbcExecutor) throws SQLException, E;
+            void execute(JdbcExecutor jdbcExecutor) throws E;
+        }
+
+        @FunctionalInterface
+        public interface PredicateWithThrowable<E extends Throwable> {
+            boolean test(JdbcExecutor jdbcExecutor) throws E;
         }
 
         private static void fillStatement(PreparedStatement stmt, Object[] params) throws SQLException {
@@ -251,45 +269,4 @@ public class SimpleJdbcTemplate {
             }
         }
     }
-
-    public static class ParamBuilder {
-
-        public static final Object[] EMPTY_OBJECT_ARRAY = {};
-
-        public static Object[] buildParams(final Object... params) {
-            if (ArrayTools.isNullOrEmpty(params)) {
-                return EMPTY_OBJECT_ARRAY;
-            }
-            return Arrays.stream(params)
-                    .map(param -> {
-                        if (param instanceof Optional) {
-                            return OptionalTools.orElseNull((Optional<?>) param);
-                        }
-                        if (param instanceof OptionalInt) {
-                            return OptionalTools.toInteger(((OptionalInt) param));
-                        }
-                        if (param instanceof OptionalLong) {
-                            return OptionalTools.toLong(((OptionalLong) param));
-                        }
-                        if (param instanceof OptionalDouble) {
-                            return OptionalTools.toDouble(((OptionalDouble) param));
-                        }
-                        return param;
-                    })
-                    .toArray();
-        }
-
-        public static <T> List<Object[]> buildBatchParams(final Collection<T> c, final Function<T, Object[]> func) {
-            Preconditions.checkNotNull(c, "The collection can not be null.");
-            Preconditions.checkNotNull(func, "The func can not be null.");
-            if (CollectionTools.isEmpty(c)) {
-                return Collections.emptyList();
-            }
-            return c.stream().map(func).collect(Collectors.toList());
-        }
-
-        private ParamBuilder() {
-            throw new IllegalStateException("Utility class");
-        }
-    }
 }

src/test/java/xyz/zhouxy/jdbc/SimpleJdbcTemplateTests.java

@@ -3,16 +3,18 @@ package xyz.zhouxy.jdbc;
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertNotNull;
 import static org.junit.jupiter.api.Assertions.assertTrue;
-import static xyz.zhouxy.jdbc.SimpleJdbcTemplate.ParamBuilder.*;
+import static xyz.zhouxy.jdbc.ParamBuilder.*;
 import static xyz.zhouxy.plusone.commons.sql.JdbcSql.IN;
 
 import java.sql.Connection;
 import java.sql.SQLException;
+import java.time.LocalDate;
 import java.time.LocalDateTime;
-import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
 import java.util.Optional;
+import java.util.Random;
+import java.util.concurrent.ThreadLocalRandom;
 
 import javax.sql.DataSource;
 
@@ -20,13 +22,16 @@ import org.junit.jupiter.api.Test;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
+import com.google.common.collect.Lists;
 import com.zaxxer.hikari.HikariConfig;
 import com.zaxxer.hikari.HikariDataSource;
 
 import xyz.zhouxy.jdbc.SimpleJdbcTemplate.JdbcExecutor;
 import xyz.zhouxy.plusone.commons.sql.SQL;
+import xyz.zhouxy.plusone.commons.util.ArrayTools;
 import xyz.zhouxy.plusone.commons.util.IdGenerator;
 import xyz.zhouxy.plusone.commons.util.IdWorker;
+import xyz.zhouxy.plusone.commons.util.Numbers;
 
 class SimpleJdbcTemplateTests {
 
@@ -48,14 +53,13 @@ class SimpleJdbcTemplateTests {
         config.setJdbcUrl("jdbc:postgresql://localhost:5432/plusone");
         config.setUsername("postgres");
         config.setPassword("zhouxy108");
-        config.setMaximumPoolSize(800);
+        config.setMaximumPoolSize(8);
         config.setConnectionTimeout(1000000);
         dataSource = new HikariDataSource(config);
     }
 
     @Test
     void testQuery() throws SQLException {
-        try (Connection conn = dataSource.getConnection()) {
         Object[] ids = buildParams("501533", "501554", "544599");
         String sql = SQL.newJdbcSql()
                 .SELECT("*")
@@ -63,11 +67,12 @@ class SimpleJdbcTemplateTests {
                 .WHERE(IN("id", ids))
                 .toString();
         log.info(sql);
+        try (Connection conn = dataSource.getConnection()) {
             List<DbRecord> rs = SimpleJdbcTemplate.connect(conn)
                     .queryToRecordList(sql, ids);
             assertNotNull(rs);
             for (DbRecord baseEntity : rs) {
-                // log.info("id: {}", baseEntity.getValueAsString("id"));
+                // log.info("id: {}", baseEntity.getValueAsString("id")); // NOSONAR
                 log.info(baseEntity.toString());
                 assertEquals(Optional.empty(), baseEntity.getValueAsString("updated_by"));
             }
@@ -77,19 +82,26 @@ class SimpleJdbcTemplateTests {
     @Test
     void testInsert() throws SQLException {
         try (Connection conn = dataSource.getConnection()) {
-            List<Map<String, Object>> keys = new ArrayList<>();
+            List<DbRecord> keys = SimpleJdbcTemplate.connect(conn).update(
-            SimpleJdbcTemplate.connect(conn).update("INSERT INTO base_table(status, created_by) VALUES (?, ?)",
+                    "INSERT INTO base_table(status, created_by) VALUES (?, ?)",
-                    buildParams(1, 886), keys);
+                    buildParams(1, 886L),
+                    ResultMap.recordResultMap);
             log.info("keys: {}", keys);
+            assertEquals(1, keys.size());
+            DbRecord result = keys.get(0);
+            assertEquals(1, result.getValueAsInt("status").getAsInt());
+            assertEquals(886L, result.getValueAsLong("created_by").getAsLong());
+            assertTrue(result.get("id").isPresent());
         }
     }
 
     @Test
     void testUpdate() throws SQLException {
         try (Connection conn = dataSource.getConnection()) {
-            List<Map<String, Object>> keys = new ArrayList<>();
+            List<DbRecord> keys = SimpleJdbcTemplate.connect(conn).update(
-            SimpleJdbcTemplate.connect(conn).update("UPDATE base_table SET status = ?, version = version + 1, update_time = now(), updated_by = ? WHERE id = ?",
+                    "UPDATE base_table SET status = ?, version = version + 1, update_time = now(), updated_by = ? WHERE id = ? AND version = ?",
-                    buildParams(2, 886, 9), keys);
+                    buildParams(2, 886, 9, 0),
+                    ResultMap.recordResultMap);
             log.info("keys: {}", keys);
         }
     }
@@ -101,21 +113,13 @@ class SimpleJdbcTemplateTests {
         try (Connection conn = dataSource.getConnection()) {
             long id = this.idGenerator.nextId();
             JdbcExecutor jdbcExecutor = SimpleJdbcTemplate.connect(conn);
-            try {
+            jdbcExecutor.executeTransaction(jdbc -> {
-                jdbcExecutor.tx(jdbc -> {
                 jdbc.update("INSERT INTO base_table (id, created_by, create_time, status) VALUES (?, ?, ?, ?)",
                         buildParams(id, 585757, LocalDateTime.now(), 0));
                 throw new NullPointerException();
             });
-            }
-            catch (NullPointerException e) {
-                // ignore
-            }
-            catch (Exception e) {
-                e.printStackTrace();
-            }
             Optional<Map<String, Object>> first = jdbcExecutor
-                    .queryFirst("SELECT * FROM base_table WHERE id = ?", id);
+                    .queryFirst("SELECT * FROM base_table WHERE id = ?", buildParams(id));
             log.info("first: {}", first);
             assertTrue(!first.isPresent());
         }
@@ -123,23 +127,95 @@ class SimpleJdbcTemplateTests {
         try (Connection conn = dataSource.getConnection()) {
             long id = this.idGenerator.nextId();
             JdbcExecutor jdbcExecutor = SimpleJdbcTemplate.connect(conn);
-            try {
+            jdbcExecutor.executeTransaction(jdbc -> {
-                jdbcExecutor.tx(jdbc -> {
                 jdbc.update("INSERT INTO base_table (id, created_by, create_time, status) VALUES (?, ?, ?, ?)",
                         buildParams(id, 585757, LocalDateTime.now(), 0));
-                    // throw new NullPointerException();
+                // throw new NullPointerException(); // NOSONAR
             });
-            }
-            catch (NullPointerException e) {
-                // ignore
-            }
-            catch (Exception e) {
-                e.printStackTrace();
-            }
             Optional<Map<String, Object>> first = jdbcExecutor
-                    .queryFirst("SELECT * FROM base_table WHERE id = ?", id);
+                    .queryFirst("SELECT * FROM base_table WHERE id = ?", buildParams(id));
+            log.info("first: {}", first);
+            assertTrue(first.isPresent());
+        }
+
+        try (Connection conn = dataSource.getConnection()) {
+            long id = this.idGenerator.nextId();
+            JdbcExecutor jdbcExecutor = SimpleJdbcTemplate.connect(conn);
+            jdbcExecutor.commitIfTrue(jdbc -> {
+                jdbc.update("INSERT INTO base_table (id, created_by, create_time, status) VALUES (?, ?, ?, ?)",
+                        buildParams(id, 585757, LocalDateTime.now(), 0));
+                throw new NullPointerException();
+            });
+            Optional<Map<String, Object>> first = jdbcExecutor
+                    .queryFirst("SELECT * FROM base_table WHERE id = ?", buildParams(id));
+            log.info("first: {}", first);
+            assertTrue(!first.isPresent());
+        }
+
+        try (Connection conn = dataSource.getConnection()) {
+            long id = this.idGenerator.nextId();
+            JdbcExecutor jdbcExecutor = SimpleJdbcTemplate.connect(conn);
+            jdbcExecutor.commitIfTrue(jdbc -> {
+                jdbc.update("INSERT INTO base_table (id, created_by, create_time, status) VALUES (?, ?, ?, ?)",
+                        buildParams(id, 585757, LocalDateTime.now(), 0));
+                return false;
+            });
+            Optional<Map<String, Object>> first = jdbcExecutor
+                    .queryFirst("SELECT * FROM base_table WHERE id = ?", buildParams(id));
+            log.info("first: {}", first);
+            assertTrue(!first.isPresent());
+        }
+
+        try (Connection conn = dataSource.getConnection()) {
+            long id = this.idGenerator.nextId();
+            JdbcExecutor jdbcExecutor = SimpleJdbcTemplate.connect(conn);
+            jdbcExecutor.commitIfTrue(jdbc -> {
+                jdbc.update("INSERT INTO base_table (id, created_by, create_time, status) VALUES (?, ?, ?, ?)",
+                        buildParams(id, 585757, LocalDateTime.now(), 0));
+                return true;
+            });
+            Optional<Map<String, Object>> first = jdbcExecutor
+                    .queryFirst("SELECT * FROM base_table WHERE id = ?", buildParams(id));
             log.info("first: {}", first);
             assertTrue(first.isPresent());
         }
     }
+
+    @Test
+    void testBatch() throws Exception {
+
+        Random random = ThreadLocalRandom.current();
+
+        LocalDate handleDate = LocalDate.of(1949, 10, 1);
+        List<DbRecord> datas = Lists.newArrayList();
+        while (handleDate.isBefore(LocalDate.of(2008, 8, 8))) {
+            DbRecord r1 = new DbRecord();
+            r1.put("username", "张三2");
+            r1.put("usage_date", handleDate);
+            r1.put("usage_duration", random.nextInt(500));
+            datas.add(r1);
+            DbRecord r2 = new DbRecord();
+            r2.put("username", "李四2");
+            r2.put("usage_date", handleDate);
+            r2.put("usage_duration", random.nextInt(500));
+            datas.add(r2);
+
+            handleDate = handleDate.plusDays(1L);
+        }
+        try (Connection conn = dataSource.getConnection()) {
+            List<int[]> result = SimpleJdbcTemplate.connect(conn)
+                    .batchUpdate("insert into test_table (username, usage_date, usage_duration) values (?,?,?)",
+                            buildBatchParams(datas, item -> buildParams(
+                                    item.getValueAsString("username"),
+                                    item.getValueAsString("usage_date"),
+                                    item.getValueAsString("usage_duration"))),
+                            400);
+            long sum = Numbers.sum(ArrayTools.concatIntArray(result));
+            assertEquals(datas.size(), sum);
+            log.info("sum: {}", sum);
+        } catch (Exception e) {
+            e.printStackTrace();
+            throw e;
+        }
+    }
 }