ZhouXY108/simple-jdbc
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

格式化代码。

Browse Source
This commit is contained in:
zhouxy108 2023-08-11 01:34:37 +08:00
parent 19906f2bd7
commit 03dd1af07e
2 changed files with 23 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/main/java/xyz/zhouxy/jdbc/SimpleJdbcTemplate.java
View File

@ -138,26 +138,26 @@ public class SimpleJdbcTemplate {
} }
public Optional<String> queryToString(String sql, Object... params) throws SQLException { public Optional<String> queryToString(String sql, Object... params) throws SQLException {
return queryFirst(sql, params, (ResultSet rs, int rowNumber) -> rs.getString(1)); 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, (ResultSet rs, int rowNumber) -> rs.getInt(1)); Optional<Integer> result = queryFirst(sql, params, (rs, rowNumber) -> 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, int rowNumber) -> rs.getLong(1)); Optional<Long> result = queryFirst(sql, params, (rs, rowNumber) -> rs.getLong(1));
return OptionalUtil.toOptionalLong(result); return OptionalUtil.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, (ResultSet rs, int rowNumber) -> rs.getDouble(1)); Optional<Double> result = queryFirst(sql, params, (rs, rowNumber) -> rs.getDouble(1));
return OptionalUtil.toOptionalDouble(result); 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, int rowNumber) -> rs.getBigDecimal(1)); 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 {
@ -201,7 +201,8 @@ public class SimpleJdbcTemplate {
atom.execute(); atom.execute();
conn.commit(); conn.commit();
conn.setAutoCommit(true); conn.setAutoCommit(true);
} catch (Exception e) { }
catch (Exception e) {
conn.rollback(); conn.rollback();
conn.setAutoCommit(true); conn.setAutoCommit(true);
throw e; throw e;
@ -240,13 +241,13 @@ public class SimpleJdbcTemplate {
.toArray(); .toArray();
} }
public static <T> List<Object[]> buildBatchParams(final Collection<T> c, final Function<T, Object[]> function) { 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(c, "The collection can not be null.");
Preconditions.checkNotNull(function, "The function can not be null."); Preconditions.checkNotNull(func, "The func can not be null.");
if (MoreCollections.isEmpty(c)) { if (MoreCollections.isEmpty(c)) {
return Collections.emptyList(); return Collections.emptyList();
} }
return c.stream().map(function).collect(Collectors.toList()); return c.stream().map(func).collect(Collectors.toList());
} }
private ParamBuilder() { private ParamBuilder() {

21
src/test/java/xyz/zhouxy/jdbc/SimpleJdbcTemplateTests.java
View File

@ -26,7 +26,7 @@ class SimpleJdbcTemplateTests {
private static final Logger log = LoggerFactory.getLogger(SimpleJdbcTemplateTests.class); private static final Logger log = LoggerFactory.getLogger(SimpleJdbcTemplateTests.class);
final DataSource dataSource; private static final DataSource dataSource;
String[] cStruct = { String[] cStruct = {
"id", "id",
@ -37,26 +37,28 @@ class SimpleJdbcTemplateTests {
"status" "status"
}; };
SimpleJdbcTemplateTests() { static {
HikariConfig config = new HikariConfig(); HikariConfig config = new HikariConfig();
config.setJdbcUrl("jdbc:postgresql://localhost:5432/plusone"); config.setJdbcUrl("jdbc:postgresql://localhost:5432/jdbctest");
config.setUsername("postgres"); config.setUsername("postgres");
config.setPassword("zhouxy108"); config.setPassword("zhouxy108");
this.dataSource = new HikariDataSource(config); config.setMaximumPoolSize(800);
config.setConnectionTimeout(1000000);
dataSource = new HikariDataSource(config);
} }
@Test @Test
void testQuery() throws SQLException { void testQuery() throws SQLException {
try (Connection conn = this.dataSource.getConnection()) { try (Connection conn = dataSource.getConnection()) {
Object[] params = buildParams("501533", "501554", "544599"); Object[] ids = buildParams("501533", "501554", "544599");
String sql = SQL.newJdbcSql() String sql = SQL.newJdbcSql()
.SELECT("*") .SELECT("*")
.FROM("test_table") .FROM("test_table")
.WHERE(IN("id", params)) .WHERE(IN("id", ids))
.toString(); .toString();
log.info(sql); log.info(sql);
List<DbRecord> rs = SimpleJdbcTemplate.connect(conn) List<DbRecord> rs = SimpleJdbcTemplate.connect(conn)
.queryToRecordList(sql, params); .queryToRecordList(sql, ids);
assertNotNull(rs); assertNotNull(rs);
for (DbRecord baseEntity : rs) { for (DbRecord baseEntity : rs) {
// log.info("id: {}", baseEntity.getValueAsString("id")); // log.info("id: {}", baseEntity.getValueAsString("id"));
@ -78,7 +80,8 @@ class SimpleJdbcTemplateTests {
log.info("first: {}", first); log.info("first: {}", first);
throw new NullPointerException(); throw new NullPointerException();
}); });
} catch (Exception e) { }
catch (Exception e) {
log.error(e.getMessage(), e); log.error(e.getMessage(), e);
} }
} }