ZhouXY108
/
plusone-commons
forked from plusone/plusone-commons
1
0
Fork 0
Code Pull Requests Activity

优化代码。

feature/net-util
ZhouXY108 2023-06-20 20:03:33 +08:00
parent e0ee719fe1
commit 7d4ae57842
1 changed files with 15 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/main/java/xyz/zhouxy/plusone/commons/jdbc/SimpleJdbcTemplate.java
View File

@ -35,11 +35,11 @@ import java.util.OptionalLong;
import java.util.function.Function; import java.util.function.Function;
import java.util.stream.Collectors; import java.util.stream.Collectors;
import org.apache.commons.lang3.ArrayUtils;
import com.google.common.annotations.Beta; import com.google.common.annotations.Beta;
import com.google.common.collect.Lists;
import xyz.zhouxy.plusone.commons.util.Assert; import xyz.zhouxy.plusone.commons.util.Assert;
import xyz.zhouxy.plusone.commons.util.MoreArrays;
import xyz.zhouxy.plusone.commons.util.OptionalUtil; import xyz.zhouxy.plusone.commons.util.OptionalUtil;
@Beta @Beta
@ -180,15 +180,20 @@ public class SimpleJdbcTemplate {
} }
public OptionalInt queryToInt(String sql, Object... params) throws SQLException { public OptionalInt queryToInt(String sql, Object... params) throws SQLException {
Optional<Integer> result = queryFirst(sql, params, (ResultSet rs) -> rs.getBigDecimal(1).intValue()); Optional<Integer> result = queryFirst(sql, params, (ResultSet rs) -> rs.getInt(1));
return OptionalUtil.toOptionalInt(result); return OptionalUtil.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, (ResultSet rs) -> rs.getBigDecimal(1).longValue()); Optional<Long> result = queryFirst(sql, params, (ResultSet rs) -> rs.getLong(1));
return OptionalUtil.toOptionalLong(result); return OptionalUtil.toOptionalLong(result);
} }
public OptionalDouble queryToDouble(String sql, Object... params) throws SQLException {
Optional<Double> result = queryFirst(sql, params, (ResultSet rs) -> rs.getDouble(1));
return OptionalUtil.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, (ResultSet rs) -> rs.getBigDecimal(1)); return queryFirst(sql, params, (ResultSet rs) -> rs.getBigDecimal(1));
} }
@ -205,7 +210,10 @@ public class SimpleJdbcTemplate {
} }
public int[] batchUpdate(String sql, Collection<Object[]> params, int batchSize) throws SQLException { public int[] batchUpdate(String sql, Collection<Object[]> params, int batchSize) throws SQLException {
int[] result = {}; int executeCount = params.size() / batchSize;
executeCount = (params.size() % batchSize == 0) ? executeCount : (executeCount + 1);
List<int[]> result = Lists.newArrayListWithCapacity(executeCount);
try (PreparedStatement stmt = this.conn.prepareStatement(sql)) { try (PreparedStatement stmt = this.conn.prepareStatement(sql)) {
int i = 0; int i = 0;
for (Object[] ps : params) { for (Object[] ps : params) {
@ -216,11 +224,11 @@ public class SimpleJdbcTemplate {
stmt.addBatch(); stmt.addBatch();
if (i % batchSize == 0 || i >= params.size()) { if (i % batchSize == 0 || i >= params.size()) {
int[] n = stmt.executeBatch(); int[] n = stmt.executeBatch();
result = ArrayUtils.addAll(result, n); result.add(n);
stmt.clearBatch(); stmt.clearBatch();
} }
} }
return result; return MoreArrays.concatIntArray(result);
} }
} }