forked from plusone/simple-jdbc
Compare commits
10 Commits
a135292909
...
e4e94d5686
Author | SHA1 | Date | |
---|---|---|---|
e4e94d5686 | |||
9e035d1b53 | |||
2ab0492e8c | |||
c6582ecf6b | |||
3ad9c9fddb | |||
7f31d477d6 | |||
fda69cea6b | |||
3ca9ad2be1 | |||
354fe42e0d | |||
0fea97f2a1 |
202
LICENSE.txt
Normal file
202
LICENSE.txt
Normal file
@@ -0,0 +1,202 @@
|
|||||||
|
|
||||||
|
Apache License
|
||||||
|
Version 2.0, January 2004
|
||||||
|
http://www.apache.org/licenses/
|
||||||
|
|
||||||
|
TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
|
||||||
|
|
||||||
|
1. Definitions.
|
||||||
|
|
||||||
|
"License" shall mean the terms and conditions for use, reproduction,
|
||||||
|
and distribution as defined by Sections 1 through 9 of this document.
|
||||||
|
|
||||||
|
"Licensor" shall mean the copyright owner or entity authorized by
|
||||||
|
the copyright owner that is granting the License.
|
||||||
|
|
||||||
|
"Legal Entity" shall mean the union of the acting entity and all
|
||||||
|
other entities that control, are controlled by, or are under common
|
||||||
|
control with that entity. For the purposes of this definition,
|
||||||
|
"control" means (i) the power, direct or indirect, to cause the
|
||||||
|
direction or management of such entity, whether by contract or
|
||||||
|
otherwise, or (ii) ownership of fifty percent (50%) or more of the
|
||||||
|
outstanding shares, or (iii) beneficial ownership of such entity.
|
||||||
|
|
||||||
|
"You" (or "Your") shall mean an individual or Legal Entity
|
||||||
|
exercising permissions granted by this License.
|
||||||
|
|
||||||
|
"Source" form shall mean the preferred form for making modifications,
|
||||||
|
including but not limited to software source code, documentation
|
||||||
|
source, and configuration files.
|
||||||
|
|
||||||
|
"Object" form shall mean any form resulting from mechanical
|
||||||
|
transformation or translation of a Source form, including but
|
||||||
|
not limited to compiled object code, generated documentation,
|
||||||
|
and conversions to other media types.
|
||||||
|
|
||||||
|
"Work" shall mean the work of authorship, whether in Source or
|
||||||
|
Object form, made available under the License, as indicated by a
|
||||||
|
copyright notice that is included in or attached to the work
|
||||||
|
(an example is provided in the Appendix below).
|
||||||
|
|
||||||
|
"Derivative Works" shall mean any work, whether in Source or Object
|
||||||
|
form, that is based on (or derived from) the Work and for which the
|
||||||
|
editorial revisions, annotations, elaborations, or other modifications
|
||||||
|
represent, as a whole, an original work of authorship. For the purposes
|
||||||
|
of this License, Derivative Works shall not include works that remain
|
||||||
|
separable from, or merely link (or bind by name) to the interfaces of,
|
||||||
|
the Work and Derivative Works thereof.
|
||||||
|
|
||||||
|
"Contribution" shall mean any work of authorship, including
|
||||||
|
the original version of the Work and any modifications or additions
|
||||||
|
to that Work or Derivative Works thereof, that is intentionally
|
||||||
|
submitted to Licensor for inclusion in the Work by the copyright owner
|
||||||
|
or by an individual or Legal Entity authorized to submit on behalf of
|
||||||
|
the copyright owner. For the purposes of this definition, "submitted"
|
||||||
|
means any form of electronic, verbal, or written communication sent
|
||||||
|
to the Licensor or its representatives, including but not limited to
|
||||||
|
communication on electronic mailing lists, source code control systems,
|
||||||
|
and issue tracking systems that are managed by, or on behalf of, the
|
||||||
|
Licensor for the purpose of discussing and improving the Work, but
|
||||||
|
excluding communication that is conspicuously marked or otherwise
|
||||||
|
designated in writing by the copyright owner as "Not a Contribution."
|
||||||
|
|
||||||
|
"Contributor" shall mean Licensor and any individual or Legal Entity
|
||||||
|
on behalf of whom a Contribution has been received by Licensor and
|
||||||
|
subsequently incorporated within the Work.
|
||||||
|
|
||||||
|
2. Grant of Copyright License. Subject to the terms and conditions of
|
||||||
|
this License, each Contributor hereby grants to You a perpetual,
|
||||||
|
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||||
|
copyright license to reproduce, prepare Derivative Works of,
|
||||||
|
publicly display, publicly perform, sublicense, and distribute the
|
||||||
|
Work and such Derivative Works in Source or Object form.
|
||||||
|
|
||||||
|
3. Grant of Patent License. Subject to the terms and conditions of
|
||||||
|
this License, each Contributor hereby grants to You a perpetual,
|
||||||
|
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||||
|
(except as stated in this section) patent license to make, have made,
|
||||||
|
use, offer to sell, sell, import, and otherwise transfer the Work,
|
||||||
|
where such license applies only to those patent claims licensable
|
||||||
|
by such Contributor that are necessarily infringed by their
|
||||||
|
Contribution(s) alone or by combination of their Contribution(s)
|
||||||
|
with the Work to which such Contribution(s) was submitted. If You
|
||||||
|
institute patent litigation against any entity (including a
|
||||||
|
cross-claim or counterclaim in a lawsuit) alleging that the Work
|
||||||
|
or a Contribution incorporated within the Work constitutes direct
|
||||||
|
or contributory patent infringement, then any patent licenses
|
||||||
|
granted to You under this License for that Work shall terminate
|
||||||
|
as of the date such litigation is filed.
|
||||||
|
|
||||||
|
4. Redistribution. You may reproduce and distribute copies of the
|
||||||
|
Work or Derivative Works thereof in any medium, with or without
|
||||||
|
modifications, and in Source or Object form, provided that You
|
||||||
|
meet the following conditions:
|
||||||
|
|
||||||
|
(a) You must give any other recipients of the Work or
|
||||||
|
Derivative Works a copy of this License; and
|
||||||
|
|
||||||
|
(b) You must cause any modified files to carry prominent notices
|
||||||
|
stating that You changed the files; and
|
||||||
|
|
||||||
|
(c) You must retain, in the Source form of any Derivative Works
|
||||||
|
that You distribute, all copyright, patent, trademark, and
|
||||||
|
attribution notices from the Source form of the Work,
|
||||||
|
excluding those notices that do not pertain to any part of
|
||||||
|
the Derivative Works; and
|
||||||
|
|
||||||
|
(d) If the Work includes a "NOTICE" text file as part of its
|
||||||
|
distribution, then any Derivative Works that You distribute must
|
||||||
|
include a readable copy of the attribution notices contained
|
||||||
|
within such NOTICE file, excluding those notices that do not
|
||||||
|
pertain to any part of the Derivative Works, in at least one
|
||||||
|
of the following places: within a NOTICE text file distributed
|
||||||
|
as part of the Derivative Works; within the Source form or
|
||||||
|
documentation, if provided along with the Derivative Works; or,
|
||||||
|
within a display generated by the Derivative Works, if and
|
||||||
|
wherever such third-party notices normally appear. The contents
|
||||||
|
of the NOTICE file are for informational purposes only and
|
||||||
|
do not modify the License. You may add Your own attribution
|
||||||
|
notices within Derivative Works that You distribute, alongside
|
||||||
|
or as an addendum to the NOTICE text from the Work, provided
|
||||||
|
that such additional attribution notices cannot be construed
|
||||||
|
as modifying the License.
|
||||||
|
|
||||||
|
You may add Your own copyright statement to Your modifications and
|
||||||
|
may provide additional or different license terms and conditions
|
||||||
|
for use, reproduction, or distribution of Your modifications, or
|
||||||
|
for any such Derivative Works as a whole, provided Your use,
|
||||||
|
reproduction, and distribution of the Work otherwise complies with
|
||||||
|
the conditions stated in this License.
|
||||||
|
|
||||||
|
5. Submission of Contributions. Unless You explicitly state otherwise,
|
||||||
|
any Contribution intentionally submitted for inclusion in the Work
|
||||||
|
by You to the Licensor shall be under the terms and conditions of
|
||||||
|
this License, without any additional terms or conditions.
|
||||||
|
Notwithstanding the above, nothing herein shall supersede or modify
|
||||||
|
the terms of any separate license agreement you may have executed
|
||||||
|
with Licensor regarding such Contributions.
|
||||||
|
|
||||||
|
6. Trademarks. This License does not grant permission to use the trade
|
||||||
|
names, trademarks, service marks, or product names of the Licensor,
|
||||||
|
except as required for reasonable and customary use in describing the
|
||||||
|
origin of the Work and reproducing the content of the NOTICE file.
|
||||||
|
|
||||||
|
7. Disclaimer of Warranty. Unless required by applicable law or
|
||||||
|
agreed to in writing, Licensor provides the Work (and each
|
||||||
|
Contributor provides its Contributions) on an "AS IS" BASIS,
|
||||||
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
||||||
|
implied, including, without limitation, any warranties or conditions
|
||||||
|
of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
|
||||||
|
PARTICULAR PURPOSE. You are solely responsible for determining the
|
||||||
|
appropriateness of using or redistributing the Work and assume any
|
||||||
|
risks associated with Your exercise of permissions under this License.
|
||||||
|
|
||||||
|
8. Limitation of Liability. In no event and under no legal theory,
|
||||||
|
whether in tort (including negligence), contract, or otherwise,
|
||||||
|
unless required by applicable law (such as deliberate and grossly
|
||||||
|
negligent acts) or agreed to in writing, shall any Contributor be
|
||||||
|
liable to You for damages, including any direct, indirect, special,
|
||||||
|
incidental, or consequential damages of any character arising as a
|
||||||
|
result of this License or out of the use or inability to use the
|
||||||
|
Work (including but not limited to damages for loss of goodwill,
|
||||||
|
work stoppage, computer failure or malfunction, or any and all
|
||||||
|
other commercial damages or losses), even if such Contributor
|
||||||
|
has been advised of the possibility of such damages.
|
||||||
|
|
||||||
|
9. Accepting Warranty or Additional Liability. While redistributing
|
||||||
|
the Work or Derivative Works thereof, You may choose to offer,
|
||||||
|
and charge a fee for, acceptance of support, warranty, indemnity,
|
||||||
|
or other liability obligations and/or rights consistent with this
|
||||||
|
License. However, in accepting such obligations, You may act only
|
||||||
|
on Your own behalf and on Your sole responsibility, not on behalf
|
||||||
|
of any other Contributor, and only if You agree to indemnify,
|
||||||
|
defend, and hold each Contributor harmless for any liability
|
||||||
|
incurred by, or claims asserted against, such Contributor by reason
|
||||||
|
of your accepting any such warranty or additional liability.
|
||||||
|
|
||||||
|
END OF TERMS AND CONDITIONS
|
||||||
|
|
||||||
|
APPENDIX: How to apply the Apache License to your work.
|
||||||
|
|
||||||
|
To apply the Apache License to your work, attach the following
|
||||||
|
boilerplate notice, with the fields enclosed by brackets "[]"
|
||||||
|
replaced with your own identifying information. (Don't include
|
||||||
|
the brackets!) The text should be enclosed in the appropriate
|
||||||
|
comment syntax for the file format. We also recommend that a
|
||||||
|
file or class name and description of purpose be included on the
|
||||||
|
same "printed page" as the copyright notice for easier
|
||||||
|
identification within third-party archives.
|
||||||
|
|
||||||
|
Copyright [yyyy] [name of copyright owner]
|
||||||
|
|
||||||
|
Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
you may not use this file except in compliance with the License.
|
||||||
|
You may obtain a copy of the License at
|
||||||
|
|
||||||
|
http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
|
||||||
|
Unless required by applicable law or agreed to in writing, software
|
||||||
|
distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
See the License for the specific language governing permissions and
|
||||||
|
limitations under the License.
|
6
README.md
Normal file
6
README.md
Normal file
@@ -0,0 +1,6 @@
|
|||||||
|
# SimpleJDBC
|
||||||
|
对 JDBC 的简单封装。
|
||||||
|
|
||||||
|
之前遇到的一个老项目,没有引入任何 ORM 框架,使用的 JDK7 明明支持泛型,所依赖的 spring-jdbc 居然是没有泛型的远古版本,该项目又不允许随意添加依赖,对数据库的操作几乎都在写原生 JDBC。故自己写了几个工具类,对 JDBC 进行简单封装,后来逐渐改进完善。
|
||||||
|
|
||||||
|
本项目不比成熟的工具,如若使用请自行承担风险。建议仅作为 JDBC 的学习参考。
|
2
pom.xml
2
pom.xml
@@ -6,7 +6,7 @@
|
|||||||
|
|
||||||
<groupId>xyz.zhouxy.jdbc</groupId>
|
<groupId>xyz.zhouxy.jdbc</groupId>
|
||||||
<artifactId>simple-jdbc</artifactId>
|
<artifactId>simple-jdbc</artifactId>
|
||||||
<version>0.1.0-SNAPSHOT</version>
|
<version>1.0.0-SNAPSHOT</version>
|
||||||
|
|
||||||
<properties>
|
<properties>
|
||||||
<maven.compiler.source>8</maven.compiler.source>
|
<maven.compiler.source>8</maven.compiler.source>
|
||||||
|
@@ -1,5 +1,5 @@
|
|||||||
/*
|
/*
|
||||||
* Copyright 2022-2023 the original author or authors.
|
* Copyright 2022-2024 the original author or authors.
|
||||||
*
|
*
|
||||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
* you may not use this file except in compliance with the License.
|
* you may not use this file except in compliance with the License.
|
||||||
@@ -18,9 +18,9 @@ package xyz.zhouxy.jdbc;
|
|||||||
|
|
||||||
import com.google.common.annotations.Beta;
|
import com.google.common.annotations.Beta;
|
||||||
import com.google.common.base.Preconditions;
|
import com.google.common.base.Preconditions;
|
||||||
import org.apache.commons.lang3.StringUtils;
|
|
||||||
import xyz.zhouxy.plusone.commons.collection.AbstractMapWrapper;
|
import xyz.zhouxy.plusone.commons.collection.AbstractMapWrapper;
|
||||||
import xyz.zhouxy.plusone.commons.util.OptionalTools;
|
import xyz.zhouxy.plusone.commons.util.OptionalTools;
|
||||||
|
import xyz.zhouxy.plusone.commons.util.StringTools;
|
||||||
|
|
||||||
import java.util.*;
|
import java.util.*;
|
||||||
|
|
||||||
@@ -28,11 +28,11 @@ import java.util.*;
|
|||||||
public class DbRecord extends AbstractMapWrapper<String, Object, DbRecord> {
|
public class DbRecord extends AbstractMapWrapper<String, Object, DbRecord> {
|
||||||
|
|
||||||
public DbRecord() {
|
public DbRecord() {
|
||||||
super(new HashMap<>(), k -> Preconditions.checkArgument(StringUtils.isNotBlank(k), "Key must has text."), null);
|
super(new HashMap<>(), k -> Preconditions.checkArgument(StringTools.isNotBlank(k), "Key must has text."), null);
|
||||||
}
|
}
|
||||||
|
|
||||||
public DbRecord(Map<String, Object> map) {
|
public DbRecord(Map<String, Object> map) {
|
||||||
super(map, k -> Preconditions.checkArgument(StringUtils.isNotBlank(k), "Key must has text."), null);
|
super(map, k -> Preconditions.checkArgument(StringTools.isNotBlank(k), "Key must has text."), null);
|
||||||
}
|
}
|
||||||
|
|
||||||
public Optional<String> getValueAsString(String key) {
|
public Optional<String> getValueAsString(String key) {
|
||||||
|
110
src/main/java/xyz/zhouxy/jdbc/DefaultBeanRowMapper.java
Normal file
110
src/main/java/xyz/zhouxy/jdbc/DefaultBeanRowMapper.java
Normal file
@@ -0,0 +1,110 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2022-2024 the original author or authors.
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* https://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package xyz.zhouxy.jdbc;
|
||||||
|
|
||||||
|
import java.beans.BeanInfo;
|
||||||
|
import java.beans.IntrospectionException;
|
||||||
|
import java.beans.Introspector;
|
||||||
|
import java.beans.PropertyDescriptor;
|
||||||
|
import java.lang.reflect.Constructor;
|
||||||
|
import java.lang.reflect.InvocationTargetException;
|
||||||
|
import java.lang.reflect.Method;
|
||||||
|
import java.sql.ResultSet;
|
||||||
|
import java.sql.ResultSetMetaData;
|
||||||
|
import java.sql.SQLException;
|
||||||
|
import java.util.Arrays;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.function.Function;
|
||||||
|
import java.util.stream.Collectors;
|
||||||
|
|
||||||
|
import javax.annotation.Nullable;
|
||||||
|
|
||||||
|
import com.google.common.base.CaseFormat;
|
||||||
|
|
||||||
|
public class DefaultBeanRowMapper<T> implements RowMapper<T> {
|
||||||
|
|
||||||
|
private final Constructor<T> constructor;
|
||||||
|
private final Map<String, PropertyDescriptor> colPropertyMap;
|
||||||
|
|
||||||
|
private DefaultBeanRowMapper(Constructor<T> constructor, Map<String, PropertyDescriptor> colPropertyMap) {
|
||||||
|
this.constructor = constructor;
|
||||||
|
this.colPropertyMap = colPropertyMap;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static <T> DefaultBeanRowMapper<T> of(Class<T> beanType) throws SQLException {
|
||||||
|
return of(beanType, null);
|
||||||
|
}
|
||||||
|
|
||||||
|
public static <T> DefaultBeanRowMapper<T> of(Class<T> beanType, @Nullable Map<String, String> propertyColMap)
|
||||||
|
throws SQLException {
|
||||||
|
try {
|
||||||
|
// 获取无参构造器
|
||||||
|
Constructor<T> constructor = beanType.getDeclaredConstructor();
|
||||||
|
constructor.setAccessible(true); // NOSONAR
|
||||||
|
|
||||||
|
// 构建 column name 和 PropertyDescriptor 的 映射
|
||||||
|
BeanInfo beanInfo = Introspector.getBeanInfo(beanType);
|
||||||
|
PropertyDescriptor[] propertyDescriptors = beanInfo.getPropertyDescriptors();
|
||||||
|
|
||||||
|
Function<? super PropertyDescriptor, String> keyMapper;
|
||||||
|
if (propertyColMap == null || propertyColMap.isEmpty()) {
|
||||||
|
keyMapper = p -> CaseFormat.LOWER_CAMEL.to(CaseFormat.LOWER_UNDERSCORE, p.getName());
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
keyMapper = p -> {
|
||||||
|
String propertyName = p.getName();
|
||||||
|
String colName = propertyColMap.get(propertyName);
|
||||||
|
return colName != null ? colName
|
||||||
|
: CaseFormat.LOWER_CAMEL.to(CaseFormat.LOWER_UNDERSCORE, propertyName);
|
||||||
|
};
|
||||||
|
}
|
||||||
|
Map<String, PropertyDescriptor> colPropertyMap = Arrays.stream(propertyDescriptors).collect(
|
||||||
|
Collectors.toMap(keyMapper, Function.identity(), (a, b) -> b));
|
||||||
|
return new DefaultBeanRowMapper<>(constructor, colPropertyMap);
|
||||||
|
}
|
||||||
|
catch (IntrospectionException e) {
|
||||||
|
throw new SQLException("There is an exception occurs during introspection.", e);
|
||||||
|
}
|
||||||
|
catch (NoSuchMethodException e) {
|
||||||
|
throw new SQLException("Could not find a no-args constructor in " + beanType.getName(), e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public T mapRow(ResultSet rs, int rowNumber) throws SQLException {
|
||||||
|
try {
|
||||||
|
T newInstance = this.constructor.newInstance();
|
||||||
|
ResultSetMetaData metaData = rs.getMetaData();
|
||||||
|
for (int i = 1; i <= metaData.getColumnCount(); i++) {
|
||||||
|
String colName = metaData.getColumnName(i);
|
||||||
|
PropertyDescriptor propertyDescriptor = this.colPropertyMap.get(colName);
|
||||||
|
if (propertyDescriptor != null) {
|
||||||
|
Method setter = propertyDescriptor.getWriteMethod();
|
||||||
|
if (setter != null) {
|
||||||
|
Class<?> propertyType = propertyDescriptor.getPropertyType();
|
||||||
|
setter.setAccessible(true); // NOSONAR
|
||||||
|
setter.invoke(newInstance, rs.getObject(colName, propertyType));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return newInstance;
|
||||||
|
}
|
||||||
|
catch (IllegalAccessException | InstantiationException | InvocationTargetException e) {
|
||||||
|
throw new SQLException(e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
74
src/main/java/xyz/zhouxy/jdbc/ParamBuilder.java
Normal file
74
src/main/java/xyz/zhouxy/jdbc/ParamBuilder.java
Normal file
@@ -0,0 +1,74 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2022-2024 the original author or authors.
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* https://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
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");
|
||||||
|
}
|
||||||
|
}
|
@@ -1,5 +1,5 @@
|
|||||||
/*
|
/*
|
||||||
* Copyright 2022-2023 the original author or authors.
|
* Copyright 2024 the original author or authors.
|
||||||
*
|
*
|
||||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
* you may not use this file except in compliance with the License.
|
* you may not use this file except in compliance with the License.
|
||||||
@@ -16,13 +16,10 @@
|
|||||||
|
|
||||||
package xyz.zhouxy.jdbc;
|
package xyz.zhouxy.jdbc;
|
||||||
|
|
||||||
import com.google.common.annotations.Beta;
|
|
||||||
|
|
||||||
import java.sql.ResultSet;
|
import java.sql.ResultSet;
|
||||||
import java.sql.SQLException;
|
import java.sql.SQLException;
|
||||||
|
|
||||||
@Beta
|
|
||||||
@FunctionalInterface
|
@FunctionalInterface
|
||||||
public interface ResultMap<T> {
|
public interface ResultHandler<T> {
|
||||||
T map(ResultSet rs, int rowNumber) throws SQLException;
|
T handle(ResultSet resultSet) throws SQLException;
|
||||||
}
|
}
|
51
src/main/java/xyz/zhouxy/jdbc/RowMapper.java
Normal file
51
src/main/java/xyz/zhouxy/jdbc/RowMapper.java
Normal file
@@ -0,0 +1,51 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2022-2024 the original author or authors.
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* https://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package xyz.zhouxy.jdbc;
|
||||||
|
|
||||||
|
import java.sql.ResultSet;
|
||||||
|
import java.sql.ResultSetMetaData;
|
||||||
|
import java.sql.SQLException;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
|
@FunctionalInterface
|
||||||
|
public interface RowMapper<T> {
|
||||||
|
T mapRow(ResultSet rs, int rowNumber) throws SQLException;
|
||||||
|
|
||||||
|
public static final RowMapper<Map<String, Object>> HASH_MAP_MAPPER = (rs, rowNumber) -> {
|
||||||
|
Map<String, Object> result = new HashMap<>();
|
||||||
|
ResultSetMetaData metaData = rs.getMetaData();
|
||||||
|
int columnCount = metaData.getColumnCount();
|
||||||
|
for (int i = 1; i <= columnCount; i++) {
|
||||||
|
String colName = metaData.getColumnName(i);
|
||||||
|
result.put(colName, rs.getObject(colName));
|
||||||
|
}
|
||||||
|
return result;
|
||||||
|
};
|
||||||
|
|
||||||
|
public static final RowMapper<DbRecord> RECORD_MAPPER =
|
||||||
|
(rs, rowNumber) -> new DbRecord(HASH_MAP_MAPPER.mapRow(rs, rowNumber));
|
||||||
|
|
||||||
|
public static <T> RowMapper<T> beanRowMapper(Class<T> beanType) throws SQLException {
|
||||||
|
return DefaultBeanRowMapper.of(beanType);
|
||||||
|
}
|
||||||
|
|
||||||
|
public static <T> RowMapper<T> beanRowMapper(Class<T> beanType, Map<String, String> propertyColMap)
|
||||||
|
throws SQLException {
|
||||||
|
return DefaultBeanRowMapper.of(beanType, propertyColMap);
|
||||||
|
}
|
||||||
|
}
|
@@ -1,5 +1,5 @@
|
|||||||
/*
|
/*
|
||||||
* Copyright 2022-2023 the original author or authors.
|
* Copyright 2022-2024 the original author or authors.
|
||||||
*
|
*
|
||||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
* you may not use this file except in compliance with the License.
|
* you may not use this file except in compliance with the License.
|
||||||
@@ -20,176 +20,771 @@ import java.math.BigDecimal;
|
|||||||
import java.sql.Connection;
|
import java.sql.Connection;
|
||||||
import java.sql.PreparedStatement;
|
import java.sql.PreparedStatement;
|
||||||
import java.sql.ResultSet;
|
import java.sql.ResultSet;
|
||||||
import java.sql.ResultSetMetaData;
|
|
||||||
import java.sql.SQLException;
|
import java.sql.SQLException;
|
||||||
|
import java.sql.Statement;
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
import java.util.Arrays;
|
|
||||||
import java.util.Collection;
|
import java.util.Collection;
|
||||||
import java.util.Collections;
|
import java.util.Collections;
|
||||||
import java.util.HashMap;
|
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
import java.util.Optional;
|
import java.util.Optional;
|
||||||
import java.util.OptionalDouble;
|
import java.util.OptionalDouble;
|
||||||
import java.util.OptionalInt;
|
import java.util.OptionalInt;
|
||||||
import java.util.OptionalLong;
|
import java.util.OptionalLong;
|
||||||
import java.util.function.Function;
|
|
||||||
import java.util.stream.Collectors;
|
|
||||||
|
|
||||||
import org.apache.commons.lang3.ArrayUtils;
|
import javax.annotation.Nonnull;
|
||||||
|
import javax.annotation.Nullable;
|
||||||
|
import javax.sql.DataSource;
|
||||||
|
|
||||||
import com.google.common.annotations.Beta;
|
|
||||||
import com.google.common.base.Preconditions;
|
|
||||||
import com.google.common.collect.Lists;
|
import com.google.common.collect.Lists;
|
||||||
|
|
||||||
import xyz.zhouxy.plusone.commons.collection.CollectionTools;
|
import xyz.zhouxy.plusone.commons.collection.CollectionTools;
|
||||||
import xyz.zhouxy.plusone.commons.util.ArrayTools;
|
import xyz.zhouxy.plusone.commons.util.AssertTools;
|
||||||
import xyz.zhouxy.plusone.commons.util.OptionalTools;
|
import xyz.zhouxy.plusone.commons.util.OptionalTools;
|
||||||
|
|
||||||
@Beta
|
|
||||||
public class SimpleJdbcTemplate {
|
public class SimpleJdbcTemplate {
|
||||||
|
|
||||||
public static JdbcExecutor connect(final Connection conn) {
|
@Nonnull
|
||||||
return new JdbcExecutor(conn);
|
private final DataSource dataSource;
|
||||||
|
|
||||||
|
public SimpleJdbcTemplate(@Nonnull DataSource dataSource) {
|
||||||
|
AssertTools.checkNotNull(dataSource);
|
||||||
|
this.dataSource = dataSource;
|
||||||
}
|
}
|
||||||
|
|
||||||
public static String paramsToString(Object[] params) {
|
// #region - query
|
||||||
return Arrays.toString(params);
|
|
||||||
|
public <T> T query(String sql, Object[] params, ResultHandler<T> resultHandler)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.query(conn, sql, params, resultHandler);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public static String paramsToString(final Collection<Object[]> params) {
|
public <T> T query(String sql, ResultHandler<T> resultHandler)
|
||||||
if (params == null) {
|
throws SQLException {
|
||||||
return "null";
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.query(conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY, resultHandler);
|
||||||
}
|
}
|
||||||
if (params.isEmpty()) {
|
|
||||||
return "[]";
|
|
||||||
}
|
|
||||||
int iMax = params.size() - 1;
|
|
||||||
StringBuilder b = new StringBuilder();
|
|
||||||
b.append('[');
|
|
||||||
int i = 0;
|
|
||||||
for (Object[] p : params) {
|
|
||||||
b.append(Arrays.toString(p));
|
|
||||||
if (i == iMax) {
|
|
||||||
return b.append(']').toString();
|
|
||||||
}
|
|
||||||
b.append(',');
|
|
||||||
i++;
|
|
||||||
}
|
|
||||||
return b.append(']').toString();
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private SimpleJdbcTemplate() {
|
// #endregion
|
||||||
throw new IllegalStateException("Utility class");
|
|
||||||
|
// #region - queryList
|
||||||
|
|
||||||
|
public <T> List<T> queryList(String sql, Object[] params, RowMapper<T> rowMapper)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryList(conn, sql, params, rowMapper);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public static class JdbcExecutor {
|
public <T> List<T> queryList(String sql, Object[] params, Class<T> clazz)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryList(conn, sql, params, clazz);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public List<Map<String, Object>> queryList(String sql, Object[] params)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryList(conn, sql, params, RowMapper.HASH_MAP_MAPPER);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public List<DbRecord> queryRecordList(String sql, Object[] params)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryList(conn, sql, params, RowMapper.RECORD_MAPPER);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public <T> List<T> queryList(String sql, RowMapper<T> rowMapper)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryList(conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY, rowMapper);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public <T> List<T> queryList(String sql, Class<T> clazz)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryList(conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY, clazz);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public List<Map<String, Object>> queryList(String sql)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryList(conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY, RowMapper.HASH_MAP_MAPPER);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public List<DbRecord> queryRecordList(String sql)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryList(conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY, RowMapper.RECORD_MAPPER);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// #endregion
|
||||||
|
|
||||||
|
// #region - queryFirst
|
||||||
|
|
||||||
|
public <T> Optional<T> queryFirst(String sql, Object[] params, RowMapper<T> rowMapper)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryFirst(conn, sql, params, rowMapper);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public <T> Optional<T> queryFirst(String sql, Object[] params, Class<T> clazz)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryFirst(conn, sql, params, clazz);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public Optional<Map<String, Object>> queryFirst(String sql, Object[] params)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryFirst(conn, sql, params, RowMapper.HASH_MAP_MAPPER);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public Optional<DbRecord> queryFirstRecord(String sql, Object[] params)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryFirst(conn, sql, params, RowMapper.RECORD_MAPPER);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public Optional<String> queryFirstString(String sql, Object[] params)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryFirstString(conn, sql, params);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public OptionalInt queryFirstInt(String sql, Object[] params)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryFirstInt(conn, sql, params);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public OptionalLong queryFirstLong(String sql, Object[] params)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryFirstLong(conn, sql, params);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public OptionalDouble queryFirstDouble(String sql, Object[] params)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryFirstDouble(conn, sql, params);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public Optional<BigDecimal> queryFirstBigDecimal(String sql, Object[] params)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryFirstBigDecimal(conn, sql, params);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public <T> Optional<T> queryFirst(String sql, RowMapper<T> rowMapper)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryFirst(conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY, rowMapper);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public <T> Optional<T> queryFirst(String sql, Class<T> clazz)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryFirst(conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY, clazz);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public Optional<Map<String, Object>> queryFirst(String sql)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryFirst(conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY, RowMapper.HASH_MAP_MAPPER);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public Optional<DbRecord> queryFirstRecord(String sql)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryFirst(conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY, RowMapper.RECORD_MAPPER);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public Optional<String> queryFirstString(String sql)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryFirstString(conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public OptionalInt queryFirstInt(String sql)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryFirstInt(conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public OptionalLong queryFirstLong(String sql)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryFirstLong(conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public OptionalDouble queryFirstDouble(String sql)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryFirstDouble(conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public Optional<BigDecimal> queryFirstBigDecimal(String sql)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.queryFirstBigDecimal(conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// #endregion
|
||||||
|
|
||||||
|
// #region - update & batchUpdate
|
||||||
|
|
||||||
|
public int update(String sql, Object[] params)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.update(conn, sql, params);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public int update(String sql)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.update(conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 执行 SQL 并更新后的数据
|
||||||
|
*
|
||||||
|
* @param sql 要执行的 SQL 语句
|
||||||
|
* @param params 参数
|
||||||
|
* @param rowMapper 结果映射规则
|
||||||
|
*
|
||||||
|
* @return 更新的数据
|
||||||
|
* @throws SQLException 执行 SQL 遇到异常情况将抛出
|
||||||
|
*/
|
||||||
|
public <T> List<T> update(String sql, Object[] params, RowMapper<T> rowMapper)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.update(conn, sql, params, rowMapper);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 执行 SQL 并更新后的数据
|
||||||
|
*
|
||||||
|
* @param sql 要执行的 SQL 语句
|
||||||
|
* @param params 参数
|
||||||
|
* @param rowMapper 结果映射规则
|
||||||
|
*
|
||||||
|
* @return 更新的数据
|
||||||
|
* @throws SQLException 执行 SQL 遇到异常情况将抛出
|
||||||
|
*/
|
||||||
|
public <T> List<T> update(String sql, RowMapper<T> rowMapper)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.update(conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY, rowMapper);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public List<int[]> batchUpdate(String sql, @Nullable Collection<Object[]> params, int batchSize)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.batchUpdate(conn, sql, params, batchSize);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public List<int[]> batchUpdateAndIgnoreException(String sql, @Nullable Collection<Object[]> params,
|
||||||
|
int batchSize, List<Exception> exceptions)
|
||||||
|
throws SQLException {
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
return JdbcExecutor.batchUpdateAndIgnoreException(conn, sql, params, batchSize, exceptions);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// #endregion
|
||||||
|
|
||||||
|
// #region - transaction
|
||||||
|
|
||||||
|
public <E extends Exception> void executeTransaction(@Nonnull final DbOperations<E> operations)
|
||||||
|
throws SQLException, E {
|
||||||
|
AssertTools.checkNotNull(operations, "Operations can not be null.");
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
final boolean autoCommit = conn.getAutoCommit();
|
||||||
|
try {
|
||||||
|
conn.setAutoCommit(false);
|
||||||
|
operations.execute(new JdbcExecutor(conn));
|
||||||
|
conn.commit();
|
||||||
|
}
|
||||||
|
catch (Exception e) {
|
||||||
|
conn.rollback();
|
||||||
|
throw e;
|
||||||
|
}
|
||||||
|
finally {
|
||||||
|
conn.setAutoCommit(autoCommit);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public <E extends Exception> void commitIfTrue(@Nonnull final PredicateWithThrowable<E> operations)
|
||||||
|
throws SQLException, E {
|
||||||
|
AssertTools.checkNotNull(operations, "Operations can not be null.");
|
||||||
|
try (Connection conn = this.dataSource.getConnection()) {
|
||||||
|
final boolean autoCommit = conn.getAutoCommit();
|
||||||
|
try {
|
||||||
|
conn.setAutoCommit(false);
|
||||||
|
if (operations.test(new JdbcExecutor(conn))) {
|
||||||
|
conn.commit();
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
conn.rollback();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (Exception e) {
|
||||||
|
conn.rollback();
|
||||||
|
throw e;
|
||||||
|
}
|
||||||
|
finally {
|
||||||
|
conn.setAutoCommit(autoCommit);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@FunctionalInterface
|
||||||
|
public interface DbOperations<E extends Exception> {
|
||||||
|
void execute(JdbcExecutor jdbcExecutor) throws E;
|
||||||
|
}
|
||||||
|
|
||||||
|
@FunctionalInterface
|
||||||
|
public interface PredicateWithThrowable<E extends Throwable> {
|
||||||
|
boolean test(JdbcExecutor jdbcExecutor) throws E;
|
||||||
|
}
|
||||||
|
|
||||||
|
// #endregion
|
||||||
|
|
||||||
|
public static final class JdbcExecutor {
|
||||||
|
|
||||||
private final Connection conn;
|
private final Connection conn;
|
||||||
|
|
||||||
public JdbcExecutor(Connection conn) {
|
private JdbcExecutor(Connection conn) {
|
||||||
this.conn = conn;
|
this.conn = conn;
|
||||||
}
|
}
|
||||||
|
|
||||||
public <T> List<T> query(String sql, Object[] params, ResultMap<T> resultMap) throws SQLException {
|
// #region - query
|
||||||
try (PreparedStatement stmt = this.conn.prepareStatement(sql)) {
|
|
||||||
|
public <T> T query(String sql, Object[] params, ResultHandler<T> resulthHandler)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.query(this.conn, sql, params, resulthHandler);
|
||||||
|
}
|
||||||
|
|
||||||
|
public <T> T query(String sql, ResultHandler<T> resulthHandler)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.query(this.conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY, resulthHandler);
|
||||||
|
}
|
||||||
|
|
||||||
|
// #endregion
|
||||||
|
|
||||||
|
// #region - queryList
|
||||||
|
|
||||||
|
public <T> List<T> queryList(String sql, Object[] params, RowMapper<T> rowMapper)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryList(this.conn, sql, params, rowMapper);
|
||||||
|
}
|
||||||
|
|
||||||
|
public <T> List<T> queryList(String sql, Object[] params, Class<T> clazz)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryList(this.conn, sql, params, clazz);
|
||||||
|
}
|
||||||
|
|
||||||
|
public List<Map<String, Object>> queryList(String sql, Object[] params)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryList(this.conn, sql, params, RowMapper.HASH_MAP_MAPPER);
|
||||||
|
}
|
||||||
|
|
||||||
|
public List<DbRecord> queryRecordList(String sql, Object[] params)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryList(this.conn, sql, params, RowMapper.RECORD_MAPPER);
|
||||||
|
}
|
||||||
|
|
||||||
|
public <T> List<T> queryList(String sql, RowMapper<T> rowMapper)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryList(this.conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY, rowMapper);
|
||||||
|
}
|
||||||
|
|
||||||
|
public <T> List<T> queryList(String sql, Class<T> clazz)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryList(this.conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY, clazz);
|
||||||
|
}
|
||||||
|
|
||||||
|
public List<Map<String, Object>> queryList(String sql)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryList(this.conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY, RowMapper.HASH_MAP_MAPPER);
|
||||||
|
}
|
||||||
|
|
||||||
|
public List<DbRecord> queryRecordList(String sql)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryList(this.conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY, RowMapper.RECORD_MAPPER);
|
||||||
|
}
|
||||||
|
|
||||||
|
// #endregion
|
||||||
|
|
||||||
|
// #region - queryFirst
|
||||||
|
|
||||||
|
public <T> Optional<T> queryFirst(String sql, Object[] params, RowMapper<T> rowMapper)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryFirst(this.conn, sql, params, rowMapper);
|
||||||
|
}
|
||||||
|
|
||||||
|
public <T> Optional<T> queryFirst(String sql, Object[] params, Class<T> clazz)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryFirst(this.conn, sql, params, clazz);
|
||||||
|
}
|
||||||
|
|
||||||
|
public Optional<Map<String, Object>> queryFirst(String sql, Object[] params)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryFirst(this.conn, sql, params, RowMapper.HASH_MAP_MAPPER);
|
||||||
|
}
|
||||||
|
|
||||||
|
public Optional<DbRecord> queryFirstRecord(String sql, Object[] params)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryFirst(this.conn, sql, params, RowMapper.RECORD_MAPPER);
|
||||||
|
}
|
||||||
|
|
||||||
|
public Optional<String> queryFirstString(String sql, Object[] params)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryFirstString(this.conn, sql, params);
|
||||||
|
}
|
||||||
|
|
||||||
|
public OptionalInt queryFirstInt(String sql, Object[] params)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryFirstInt(this.conn, sql, params);
|
||||||
|
}
|
||||||
|
|
||||||
|
public OptionalLong queryFirstLong(String sql, Object[] params)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryFirstLong(this.conn, sql, params);
|
||||||
|
}
|
||||||
|
|
||||||
|
public OptionalDouble queryFirstDouble(String sql, Object[] params)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryFirstDouble(this.conn, sql, params);
|
||||||
|
}
|
||||||
|
|
||||||
|
public Optional<BigDecimal> queryFirstBigDecimal(String sql, Object[] params)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryFirstBigDecimal(this.conn, sql, params);
|
||||||
|
}
|
||||||
|
|
||||||
|
public <T> Optional<T> queryFirst(String sql, RowMapper<T> rowMapper)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryFirst(this.conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY, rowMapper);
|
||||||
|
}
|
||||||
|
|
||||||
|
public <T> Optional<T> queryFirst(String sql, Class<T> clazz)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryFirst(this.conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY, clazz);
|
||||||
|
}
|
||||||
|
|
||||||
|
public Optional<Map<String, Object>> queryFirst(String sql)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryFirst(this.conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY, RowMapper.HASH_MAP_MAPPER);
|
||||||
|
}
|
||||||
|
|
||||||
|
public Optional<DbRecord> queryFirstRecord(String sql)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryFirst(this.conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY, RowMapper.RECORD_MAPPER);
|
||||||
|
}
|
||||||
|
|
||||||
|
public Optional<String> queryFirstString(String sql)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryFirstString(this.conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY);
|
||||||
|
}
|
||||||
|
|
||||||
|
public OptionalInt queryFirstInt(String sql)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryFirstInt(this.conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY);
|
||||||
|
}
|
||||||
|
|
||||||
|
public OptionalLong queryFirstLong(String sql)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryFirstLong(this.conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY);
|
||||||
|
}
|
||||||
|
|
||||||
|
public OptionalDouble queryFirstDouble(String sql)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryFirstDouble(this.conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY);
|
||||||
|
}
|
||||||
|
|
||||||
|
public Optional<BigDecimal> queryFirstBigDecimal(String sql)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.queryFirstBigDecimal(this.conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY);
|
||||||
|
}
|
||||||
|
|
||||||
|
// #endregion
|
||||||
|
|
||||||
|
// #region - update & batchUpdate
|
||||||
|
|
||||||
|
public int update(String sql, Object[] params)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.update(this.conn, sql, params);
|
||||||
|
}
|
||||||
|
|
||||||
|
public int update(String sql)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.update(this.conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 执行 SQL 并更新后的数据
|
||||||
|
*
|
||||||
|
* @param sql 要执行的 SQL 语句
|
||||||
|
* @param params 参数
|
||||||
|
* @param rowMapper 结果映射规则
|
||||||
|
*
|
||||||
|
* @return 更新的数据
|
||||||
|
* @throws SQLException 执行 SQL 遇到异常情况将抛出
|
||||||
|
*/
|
||||||
|
public <T> List<T> update(String sql, Object[] params, RowMapper<T> rowMapper)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.update(this.conn, sql, params, rowMapper);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 执行 SQL 并更新后的数据
|
||||||
|
*
|
||||||
|
* @param sql 要执行的 SQL 语句
|
||||||
|
* @param params 参数
|
||||||
|
* @param rowMapper 结果映射规则
|
||||||
|
*
|
||||||
|
* @return 更新的数据
|
||||||
|
* @throws SQLException 执行 SQL 遇到异常情况将抛出
|
||||||
|
*/
|
||||||
|
public <T> List<T> update(String sql, RowMapper<T> rowMapper)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.update(this.conn, sql, ParamBuilder.EMPTY_OBJECT_ARRAY, rowMapper);
|
||||||
|
}
|
||||||
|
|
||||||
|
public List<int[]> batchUpdate(String sql, @Nullable Collection<Object[]> params, int batchSize)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.batchUpdate(this.conn, sql, params, batchSize);
|
||||||
|
}
|
||||||
|
|
||||||
|
public List<int[]> batchUpdateAndIgnoreException(String sql, @Nullable Collection<Object[]> params,
|
||||||
|
int batchSize, List<Exception> exceptions)
|
||||||
|
throws SQLException {
|
||||||
|
return JdbcExecutor.batchUpdateAndIgnoreException(this.conn, sql, params, batchSize, exceptions);
|
||||||
|
}
|
||||||
|
|
||||||
|
// #endregion
|
||||||
|
|
||||||
|
// #region - internal
|
||||||
|
|
||||||
|
private static <T> T queryInternal(@Nonnull Connection conn,
|
||||||
|
@Nonnull String sql,
|
||||||
|
@Nullable Object[] params,
|
||||||
|
@Nonnull ResultHandler<T> resultHandler)
|
||||||
|
throws SQLException {
|
||||||
|
try (PreparedStatement stmt = conn.prepareStatement(sql)) {
|
||||||
fillStatement(stmt, params);
|
fillStatement(stmt, params);
|
||||||
try (ResultSet rs = stmt.executeQuery()) {
|
try (ResultSet rs = stmt.executeQuery()) {
|
||||||
|
return resultHandler.handle(rs);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private static <T> List<T> queryListInternal(@Nonnull Connection conn,
|
||||||
|
@Nonnull String sql,
|
||||||
|
@Nullable Object[] params,
|
||||||
|
@Nonnull RowMapper<T> rowMapper)
|
||||||
|
throws SQLException {
|
||||||
|
return queryInternal(conn, sql, params, rs -> {
|
||||||
List<T> result = new ArrayList<>();
|
List<T> result = new ArrayList<>();
|
||||||
int rowNumber = 0;
|
int rowNumber = 0;
|
||||||
while (rs.next()) {
|
while (rs.next()) {
|
||||||
T e = resultMap.map(rs, rowNumber++);
|
T e = rowMapper.mapRow(rs, rowNumber++);
|
||||||
result.add(e);
|
result.add(e);
|
||||||
}
|
}
|
||||||
return result;
|
return result;
|
||||||
}
|
});
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
public <T> Optional<T> queryFirst(String sql, Object[] params, ResultMap<T> resultMap) throws SQLException {
|
private static <T> Optional<T> queryFirstInternal(@Nonnull Connection conn,
|
||||||
return query(sql, params, resultMap).stream().findFirst();
|
@Nonnull String sql,
|
||||||
|
@Nullable Object[] params,
|
||||||
|
@Nonnull RowMapper<T> rowMapper)
|
||||||
|
throws SQLException {
|
||||||
|
return queryInternal(conn, sql, params, rs -> {
|
||||||
|
if (rs.next()) {
|
||||||
|
return Optional.ofNullable(rowMapper.mapRow(rs, 0));
|
||||||
|
}
|
||||||
|
return Optional.empty();
|
||||||
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
public static final ResultMap<Map<String, Object>> mapResultMap = (rs, rowNumber) -> {
|
// #endregion
|
||||||
Map<String, Object> result = new HashMap<>();
|
|
||||||
ResultSetMetaData metaData = rs.getMetaData();
|
|
||||||
int columnCount = metaData.getColumnCount();
|
|
||||||
for (int i = 1; i <= columnCount; i++) {
|
|
||||||
String colName = metaData.getColumnName(i);
|
|
||||||
result.put(colName, rs.getObject(colName));
|
|
||||||
}
|
|
||||||
return result;
|
|
||||||
};
|
|
||||||
|
|
||||||
public List<Map<String, Object>> query(String sql, Object... params) throws SQLException {
|
// #region - query
|
||||||
return query(sql, params, mapResultMap);
|
|
||||||
|
private static <T> T query(Connection conn, String sql, Object[] params, ResultHandler<T> resultHandler)
|
||||||
|
throws SQLException {
|
||||||
|
assertConnectionNotNull(conn);
|
||||||
|
assertSqlNotNull(sql);
|
||||||
|
assertResultHandlerNotNull(resultHandler);
|
||||||
|
return queryInternal(conn, sql, params, resultHandler);
|
||||||
}
|
}
|
||||||
|
|
||||||
public Optional<Map<String, Object>> queryFirst(String sql, Object... params) throws SQLException {
|
// #endregion
|
||||||
return queryFirst(sql, params, mapResultMap);
|
|
||||||
|
// #region - queryList
|
||||||
|
|
||||||
|
private static <T> List<T> queryList(Connection conn, String sql, Object[] params, RowMapper<T> rowMapper)
|
||||||
|
throws SQLException {
|
||||||
|
assertConnectionNotNull(conn);
|
||||||
|
assertSqlNotNull(sql);
|
||||||
|
assertRowMapperNotNull(rowMapper);
|
||||||
|
return queryListInternal(conn, sql, params, rowMapper);
|
||||||
}
|
}
|
||||||
|
|
||||||
public static final ResultMap<DbRecord> recordResultMap = (rs, rowNumber) -> {
|
private static <T> List<T> queryList(Connection conn, String sql, Object[] params, Class<T> clazz)
|
||||||
DbRecord result = new DbRecord();
|
throws SQLException {
|
||||||
ResultSetMetaData metaData = rs.getMetaData();
|
assertConnectionNotNull(conn);
|
||||||
int columnCount = metaData.getColumnCount();
|
assertSqlNotNull(sql);
|
||||||
for (int i = 1; i <= columnCount; i++) {
|
assertClazzNotNull(clazz);
|
||||||
String colName = metaData.getColumnName(i);
|
return queryListInternal(conn, sql, params, (rs, rowNumber) -> rs.getObject(1, clazz));
|
||||||
result.put(colName, rs.getObject(colName));
|
|
||||||
}
|
|
||||||
return result;
|
|
||||||
};
|
|
||||||
|
|
||||||
public List<DbRecord> queryToRecordList(String sql, Object... params) throws SQLException {
|
|
||||||
return query(sql, params, recordResultMap);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
public Optional<DbRecord> queryFirstRecord(String sql, Object... params) throws SQLException {
|
// #endregion
|
||||||
return queryFirst(sql, params, recordResultMap);
|
|
||||||
|
// #region - queryFirst
|
||||||
|
|
||||||
|
private static <T> Optional<T> queryFirst(Connection conn, String sql, Object[] params, RowMapper<T> rowMapper)
|
||||||
|
throws SQLException {
|
||||||
|
assertConnectionNotNull(conn);
|
||||||
|
assertSqlNotNull(sql);
|
||||||
|
assertRowMapperNotNull(rowMapper);
|
||||||
|
return queryFirstInternal(conn, sql, params, rowMapper);
|
||||||
}
|
}
|
||||||
|
|
||||||
public Optional<String> queryToString(String sql, Object... params) throws SQLException {
|
private static <T> Optional<T> queryFirst(Connection conn, String sql, Object[] params, Class<T> clazz)
|
||||||
return queryFirst(sql, params, (rs, rowNumber) -> rs.getString(1));
|
throws SQLException {
|
||||||
|
assertConnectionNotNull(conn);
|
||||||
|
assertSqlNotNull(sql);
|
||||||
|
assertClazzNotNull(clazz);
|
||||||
|
return queryFirstInternal(conn, sql, params, (rs, rowNumber) -> rs.getObject(1, clazz));
|
||||||
}
|
}
|
||||||
|
|
||||||
public OptionalInt queryToInt(String sql, Object... params) throws SQLException {
|
private static Optional<String> queryFirstString(Connection conn, String sql, Object[] params)
|
||||||
Optional<Integer> result = queryFirst(sql, params, (rs, rowNumber) -> rs.getInt(1));
|
throws SQLException {
|
||||||
|
return queryFirst(conn, sql, params, (rs, rowNumber) -> rs.getString(1));
|
||||||
|
}
|
||||||
|
|
||||||
|
private static OptionalInt queryFirstInt(Connection conn, String sql, Object[] params)
|
||||||
|
throws SQLException {
|
||||||
|
Optional<Integer> result = queryFirst(conn, sql, params, (rs, rowNumber) -> rs.getInt(1));
|
||||||
return OptionalTools.toOptionalInt(result);
|
return OptionalTools.toOptionalInt(result);
|
||||||
}
|
}
|
||||||
|
|
||||||
public OptionalLong queryToLong(String sql, Object... params) throws SQLException {
|
private static OptionalLong queryFirstLong(Connection conn, String sql, Object[] params)
|
||||||
Optional<Long> result = queryFirst(sql, params, (rs, rowNumber) -> rs.getLong(1));
|
throws SQLException {
|
||||||
|
Optional<Long> result = queryFirst(conn, sql, params, (rs, rowNumber) -> rs.getLong(1));
|
||||||
return OptionalTools.toOptionalLong(result);
|
return OptionalTools.toOptionalLong(result);
|
||||||
}
|
}
|
||||||
|
|
||||||
public OptionalDouble queryToDouble(String sql, Object... params) throws SQLException {
|
private static OptionalDouble queryFirstDouble(Connection conn, String sql, Object[] params)
|
||||||
Optional<Double> result = queryFirst(sql, params, (rs, rowNumber) -> rs.getDouble(1));
|
throws SQLException {
|
||||||
|
Optional<Double> result = queryFirst(conn, sql, params, (rs, rowNumber) -> rs.getDouble(1));
|
||||||
return OptionalTools.toOptionalDouble(result);
|
return OptionalTools.toOptionalDouble(result);
|
||||||
}
|
}
|
||||||
|
|
||||||
public Optional<BigDecimal> queryToBigDecimal(String sql, Object... params) throws SQLException {
|
private static Optional<BigDecimal> queryFirstBigDecimal(Connection conn, String sql, Object[] params)
|
||||||
return queryFirst(sql, params, (rs, rowNumber) -> rs.getBigDecimal(1));
|
throws SQLException {
|
||||||
|
return queryFirst(conn, sql, params, (rs, rowNumber) -> rs.getBigDecimal(1));
|
||||||
}
|
}
|
||||||
|
|
||||||
public int update(String sql, Object... params) throws SQLException {
|
// #endregion
|
||||||
try (PreparedStatement stmt = this.conn.prepareStatement(sql)) {
|
|
||||||
|
// #region - update & batchUpdate
|
||||||
|
|
||||||
|
private static int update(Connection conn, String sql, Object[] params)
|
||||||
|
throws SQLException {
|
||||||
|
assertConnectionNotNull(conn);
|
||||||
|
assertSqlNotNull(sql);
|
||||||
|
try (PreparedStatement stmt = conn.prepareStatement(sql)) {
|
||||||
fillStatement(stmt, params);
|
fillStatement(stmt, params);
|
||||||
return stmt.executeUpdate();
|
return stmt.executeUpdate();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public int[] batchUpdate(String sql, Collection<Object[]> params, int batchSize) throws SQLException {
|
/**
|
||||||
|
* 执行 SQL 并更新后的数据
|
||||||
|
*
|
||||||
|
* @param sql 要执行的 SQL 语句
|
||||||
|
* @param params 参数
|
||||||
|
* @param rowMapper 结果映射规则
|
||||||
|
*
|
||||||
|
* @return 更新的数据
|
||||||
|
* @throws SQLException 执行 SQL 遇到异常情况将抛出
|
||||||
|
*/
|
||||||
|
private static <T> List<T> update(Connection conn, String sql, Object[] params, RowMapper<T> rowMapper)
|
||||||
|
throws SQLException {
|
||||||
|
assertConnectionNotNull(conn);
|
||||||
|
assertSqlNotNull(sql);
|
||||||
|
assertRowMapperNotNull(rowMapper);
|
||||||
|
final List<T> result = new ArrayList<>();
|
||||||
|
try (PreparedStatement stmt = conn.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS)) {
|
||||||
|
fillStatement(stmt, params);
|
||||||
|
stmt.executeUpdate();
|
||||||
|
try (ResultSet generatedKeys = stmt.getGeneratedKeys();) {
|
||||||
|
int rowNumber = 0;
|
||||||
|
while (generatedKeys.next()) {
|
||||||
|
T e = rowMapper.mapRow(generatedKeys, rowNumber++);
|
||||||
|
result.add(e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private static List<int[]> batchUpdate(Connection conn, String sql, Collection<Object[]> params, int batchSize)
|
||||||
|
throws SQLException {
|
||||||
|
assertConnectionNotNull(conn);
|
||||||
|
assertSqlNotNull(sql);
|
||||||
|
|
||||||
|
if (params == null || params.isEmpty()) {
|
||||||
|
return Collections.emptyList();
|
||||||
|
}
|
||||||
int executeCount = params.size() / batchSize;
|
int executeCount = params.size() / batchSize;
|
||||||
executeCount = (params.size() % batchSize == 0) ? executeCount : (executeCount + 1);
|
executeCount = (params.size() % batchSize == 0) ? executeCount : (executeCount + 1);
|
||||||
List<int[]> result = Lists.newArrayListWithCapacity(executeCount);
|
List<int[]> result = Lists.newArrayListWithCapacity(executeCount);
|
||||||
|
|
||||||
try (PreparedStatement stmt = this.conn.prepareStatement(sql)) {
|
try (PreparedStatement stmt = conn.prepareStatement(sql)) {
|
||||||
int i = 0;
|
int i = 0;
|
||||||
for (Object[] ps : params) {
|
for (Object[] ps : params) {
|
||||||
i++;
|
i++;
|
||||||
for (int j = 0; j < ps.length; j++) {
|
fillStatement(stmt, ps);
|
||||||
stmt.setObject(j + 1, ps[j]);
|
|
||||||
}
|
|
||||||
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();
|
||||||
@@ -197,33 +792,53 @@ public class SimpleJdbcTemplate {
|
|||||||
stmt.clearBatch();
|
stmt.clearBatch();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return ArrayTools.concatIntArray(result);
|
return result;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public <E extends Exception> void tx(final IAtom<E> atom) throws SQLException, E {
|
private static List<int[]> batchUpdateAndIgnoreException(Connection conn,
|
||||||
Preconditions.checkNotNull(atom, "Atom can not be null.");
|
String sql, @Nullable Collection<Object[]> params, int batchSize,
|
||||||
final boolean autoCommit = this.conn.getAutoCommit();
|
List<Exception> exceptions)
|
||||||
|
throws SQLException {
|
||||||
|
assertConnectionNotNull(conn);
|
||||||
|
assertSqlNotNull(sql);
|
||||||
|
AssertTools.checkArgument(CollectionTools.isNotEmpty(exceptions),
|
||||||
|
"The list used to store exceptions should be non-null and empty.");
|
||||||
|
if (params == null || params.isEmpty()) {
|
||||||
|
return Collections.emptyList();
|
||||||
|
}
|
||||||
|
int executeCount = params.size() / batchSize;
|
||||||
|
executeCount = (params.size() % batchSize == 0) ? executeCount : (executeCount + 1);
|
||||||
|
List<int[]> result = Lists.newArrayListWithCapacity(executeCount);
|
||||||
|
|
||||||
|
try (PreparedStatement stmt = conn.prepareStatement(sql)) {
|
||||||
|
int i = 0;
|
||||||
|
for (Object[] ps : params) {
|
||||||
|
i++;
|
||||||
|
fillStatement(stmt, ps);
|
||||||
|
stmt.addBatch();
|
||||||
|
final int batchIndex = i % batchSize;
|
||||||
|
if (batchIndex == 0 || i >= params.size()) {
|
||||||
try {
|
try {
|
||||||
this.conn.setAutoCommit(false);
|
int[] n = stmt.executeBatch();
|
||||||
atom.execute(this);
|
result.add(n);
|
||||||
this.conn.commit();
|
stmt.clearBatch();
|
||||||
}
|
}
|
||||||
catch (Exception e) {
|
catch (Exception e) {
|
||||||
this.conn.rollback();
|
int n = (i >= params.size() && batchIndex != 0) ? batchIndex : batchSize;
|
||||||
throw e;
|
result.add(new int[n]);
|
||||||
|
stmt.clearBatch();
|
||||||
|
// 收集异常信息
|
||||||
|
exceptions.add(e);
|
||||||
}
|
}
|
||||||
finally {
|
}
|
||||||
this.conn.setAutoCommit(autoCommit);
|
}
|
||||||
|
return result;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@FunctionalInterface
|
private static void fillStatement(@Nonnull PreparedStatement stmt, @Nullable Object[] params)
|
||||||
public interface IAtom<E extends Exception> {
|
throws SQLException {
|
||||||
void execute(JdbcExecutor jdbcExecutor) throws SQLException, E;
|
|
||||||
}
|
|
||||||
|
|
||||||
private static void fillStatement(PreparedStatement stmt, Object[] params) throws SQLException {
|
|
||||||
if (params != null && params.length > 0) {
|
if (params != null && params.length > 0) {
|
||||||
Object param;
|
Object param;
|
||||||
for (int i = 0; i < params.length; i++) {
|
for (int i = 0; i < params.length; i++) {
|
||||||
@@ -243,44 +858,29 @@ public class SimpleJdbcTemplate {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// #region - Asserts
|
||||||
|
|
||||||
|
private static void assertConnectionNotNull(Connection conn) {
|
||||||
|
AssertTools.checkArgumentNotNull(conn, "The argument \"conn\" could not be null.");
|
||||||
}
|
}
|
||||||
|
|
||||||
public static class ParamBuilder {
|
private static void assertSqlNotNull(String sql) {
|
||||||
|
AssertTools.checkArgumentNotNull(sql, "The argument \"sql\" could not be null.");
|
||||||
public static Object[] buildParams(final Object... params) {
|
|
||||||
if (ArrayUtils.isEmpty(params)) {
|
|
||||||
return ArrayUtils.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) {
|
private static void assertRowMapperNotNull(RowMapper<?> rowMapper) {
|
||||||
Preconditions.checkNotNull(c, "The collection can not be null.");
|
AssertTools.checkArgumentNotNull(rowMapper, "The argument \"rowMapper\" could 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() {
|
private static void assertResultHandlerNotNull(ResultHandler<?> resultHandler) {
|
||||||
throw new IllegalStateException("Utility class");
|
AssertTools.checkArgumentNotNull(resultHandler, "The argument \"resultHandler\" could not be null.");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private static void assertClazzNotNull(Class<?> clazz) {
|
||||||
|
AssertTools.checkArgumentNotNull(clazz, "The argument \"clazz\" could not be null.");
|
||||||
|
}
|
||||||
|
|
||||||
|
// #endregion
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@@ -1,124 +0,0 @@
|
|||||||
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.plusone.commons.sql.JdbcSql.IN;
|
|
||||||
|
|
||||||
import java.sql.Connection;
|
|
||||||
import java.sql.SQLException;
|
|
||||||
import java.time.LocalDateTime;
|
|
||||||
import java.util.List;
|
|
||||||
import java.util.Map;
|
|
||||||
import java.util.Optional;
|
|
||||||
|
|
||||||
import javax.sql.DataSource;
|
|
||||||
|
|
||||||
import org.junit.jupiter.api.Test;
|
|
||||||
import org.slf4j.Logger;
|
|
||||||
import org.slf4j.LoggerFactory;
|
|
||||||
|
|
||||||
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.IdGenerator;
|
|
||||||
import xyz.zhouxy.plusone.commons.util.IdWorker;
|
|
||||||
|
|
||||||
class SimpleJdbcTemplateTests {
|
|
||||||
|
|
||||||
private static final Logger log = LoggerFactory.getLogger(SimpleJdbcTemplateTests.class);
|
|
||||||
|
|
||||||
private static final DataSource dataSource;
|
|
||||||
|
|
||||||
String[] cStruct = {
|
|
||||||
"id",
|
|
||||||
"created_by",
|
|
||||||
"create_time",
|
|
||||||
"updated_by",
|
|
||||||
"update_time",
|
|
||||||
"status"
|
|
||||||
};
|
|
||||||
|
|
||||||
static {
|
|
||||||
HikariConfig config = new HikariConfig();
|
|
||||||
config.setJdbcUrl("jdbc:postgresql://localhost:5432/plusone");
|
|
||||||
config.setUsername("postgres");
|
|
||||||
config.setPassword("zhouxy108");
|
|
||||||
config.setMaximumPoolSize(800);
|
|
||||||
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("*")
|
|
||||||
.FROM("test_table")
|
|
||||||
.WHERE(IN("id", ids))
|
|
||||||
.toString();
|
|
||||||
log.info(sql);
|
|
||||||
List<DbRecord> rs = SimpleJdbcTemplate.connect(conn)
|
|
||||||
.queryToRecordList(sql, ids);
|
|
||||||
assertNotNull(rs);
|
|
||||||
for (DbRecord baseEntity : rs) {
|
|
||||||
// log.info("id: {}", baseEntity.getValueAsString("id"));
|
|
||||||
log.info(baseEntity.toString());
|
|
||||||
assertEquals(Optional.empty(), baseEntity.getValueAsString("updated_by"));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
final IdWorker idGenerator = IdGenerator.getSnowflakeIdGenerator(0);
|
|
||||||
|
|
||||||
@Test
|
|
||||||
void testTransaction() throws SQLException {
|
|
||||||
try (Connection conn = dataSource.getConnection()) {
|
|
||||||
long id = this.idGenerator.nextId();
|
|
||||||
JdbcExecutor jdbcExecutor = SimpleJdbcTemplate.connect(conn);
|
|
||||||
try {
|
|
||||||
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);
|
|
||||||
log.info("first: {}", first);
|
|
||||||
assertTrue(!first.isPresent());
|
|
||||||
}
|
|
||||||
|
|
||||||
try (Connection conn = dataSource.getConnection()) {
|
|
||||||
long id = this.idGenerator.nextId();
|
|
||||||
JdbcExecutor jdbcExecutor = SimpleJdbcTemplate.connect(conn);
|
|
||||||
try {
|
|
||||||
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);
|
|
||||||
log.info("first: {}", first);
|
|
||||||
assertTrue(first.isPresent());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
277
src/test/java/xyz/zhouxy/jdbc/test/SimpleJdbcTemplateTests.java
Normal file
277
src/test/java/xyz/zhouxy/jdbc/test/SimpleJdbcTemplateTests.java
Normal file
@@ -0,0 +1,277 @@
|
|||||||
|
package xyz.zhouxy.jdbc.test;
|
||||||
|
|
||||||
|
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.ParamBuilder.*;
|
||||||
|
import static xyz.zhouxy.plusone.commons.sql.JdbcSql.IN;
|
||||||
|
|
||||||
|
import java.sql.SQLException;
|
||||||
|
import java.time.LocalDate;
|
||||||
|
import java.time.LocalDateTime;
|
||||||
|
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;
|
||||||
|
|
||||||
|
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.DbRecord;
|
||||||
|
import xyz.zhouxy.jdbc.RowMapper;
|
||||||
|
import xyz.zhouxy.jdbc.SimpleJdbcTemplate;
|
||||||
|
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 {
|
||||||
|
|
||||||
|
private static final Logger log = LoggerFactory.getLogger(SimpleJdbcTemplateTests.class);
|
||||||
|
|
||||||
|
private static final DataSource dataSource;
|
||||||
|
|
||||||
|
private static final SimpleJdbcTemplate jdbcTemplate;
|
||||||
|
|
||||||
|
static {
|
||||||
|
HikariConfig config = new HikariConfig();
|
||||||
|
config.setJdbcUrl("jdbc:postgresql://localhost:5432/plusone");
|
||||||
|
config.setUsername("postgres");
|
||||||
|
config.setPassword("zhouxy108");
|
||||||
|
config.setMaximumPoolSize(8);
|
||||||
|
config.setConnectionTimeout(1000000);
|
||||||
|
dataSource = new HikariDataSource(config);
|
||||||
|
jdbcTemplate = new SimpleJdbcTemplate(dataSource);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void testQuery() throws SQLException {
|
||||||
|
Object[] ids = buildParams(22915, 22916, 22917, 22918, 22919, 22920, 22921);
|
||||||
|
String sql = SQL.newJdbcSql()
|
||||||
|
.SELECT("*")
|
||||||
|
.FROM("test_table")
|
||||||
|
.WHERE(IN("id", ids))
|
||||||
|
.toString();
|
||||||
|
log.info(sql);
|
||||||
|
List<DbRecord> rs = jdbcTemplate.queryRecordList(sql, ids);
|
||||||
|
assertNotNull(rs);
|
||||||
|
for (DbRecord baseEntity : rs) {
|
||||||
|
// log.info("id: {}", baseEntity.getValueAsString("id")); // NOSONAR
|
||||||
|
log.info(baseEntity.toString());
|
||||||
|
assertTrue(baseEntity.getValueAsString("username").isPresent());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void testInsert() throws SQLException {
|
||||||
|
List<DbRecord> keys = jdbcTemplate.update(
|
||||||
|
"INSERT INTO base_table(status, created_by) VALUES (?, ?)",
|
||||||
|
buildParams(1, 886L),
|
||||||
|
RowMapper.RECORD_MAPPER);
|
||||||
|
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 {
|
||||||
|
List<DbRecord> keys = jdbcTemplate.update(
|
||||||
|
"UPDATE base_table SET status = ?, version = version + 1, update_time = now(), updated_by = ? WHERE id = ? AND version = ?",
|
||||||
|
buildParams(2, 886, 571328822575109L, 0),
|
||||||
|
RowMapper.RECORD_MAPPER);
|
||||||
|
log.info("keys: {}", keys);
|
||||||
|
}
|
||||||
|
|
||||||
|
final IdWorker idGenerator = IdGenerator.getSnowflakeIdGenerator(0);
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void testTransaction() throws SQLException {
|
||||||
|
// 抛异常,回滚
|
||||||
|
{
|
||||||
|
long id = this.idGenerator.nextId();
|
||||||
|
try {
|
||||||
|
jdbcTemplate.executeTransaction((JdbcExecutor jdbc) -> {
|
||||||
|
jdbc.update("INSERT INTO base_table (id, created_by, create_time, status) VALUES (?, ?, ?, ?)",
|
||||||
|
buildParams(id, 100, LocalDateTime.now(), 0));
|
||||||
|
throw new NullPointerException();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
catch (NullPointerException e) {
|
||||||
|
// ignore
|
||||||
|
}
|
||||||
|
Optional<Map<String, Object>> first = jdbcTemplate
|
||||||
|
.queryFirst("SELECT * FROM base_table WHERE id = ?", buildParams(id));
|
||||||
|
log.info("first: {}", first);
|
||||||
|
assertTrue(!first.isPresent());
|
||||||
|
}
|
||||||
|
|
||||||
|
// 没有异常,提交事务
|
||||||
|
{
|
||||||
|
long id = this.idGenerator.nextId();
|
||||||
|
jdbcTemplate.executeTransaction(jdbc -> {
|
||||||
|
jdbc.update("INSERT INTO base_table (id, created_by, create_time, status) VALUES (?, ?, ?, ?)",
|
||||||
|
buildParams(id, 101, LocalDateTime.now(), 0));
|
||||||
|
});
|
||||||
|
|
||||||
|
Optional<Map<String, Object>> first = jdbcTemplate
|
||||||
|
.queryFirst("SELECT * FROM base_table WHERE id = ?", buildParams(id));
|
||||||
|
log.info("first: {}", first);
|
||||||
|
assertTrue(first.isPresent());
|
||||||
|
}
|
||||||
|
|
||||||
|
// 抛异常,回滚
|
||||||
|
{
|
||||||
|
long id = this.idGenerator.nextId();
|
||||||
|
try {
|
||||||
|
jdbcTemplate.commitIfTrue(jdbc -> {
|
||||||
|
jdbc.update("INSERT INTO base_table (id, created_by, create_time, status) VALUES (?, ?, ?, ?)",
|
||||||
|
buildParams(id, 102, LocalDateTime.now(), 0));
|
||||||
|
throw new NullPointerException();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
catch (NullPointerException e) {
|
||||||
|
// ignore
|
||||||
|
}
|
||||||
|
Optional<Map<String, Object>> first = jdbcTemplate
|
||||||
|
.queryFirst("SELECT * FROM base_table WHERE id = ?", buildParams(id));
|
||||||
|
log.info("first: {}", first);
|
||||||
|
assertTrue(!first.isPresent());
|
||||||
|
}
|
||||||
|
|
||||||
|
// 返回 false,回滚
|
||||||
|
{
|
||||||
|
long id = this.idGenerator.nextId();
|
||||||
|
jdbcTemplate.commitIfTrue(jdbc -> {
|
||||||
|
jdbc.update("INSERT INTO base_table (id, created_by, create_time, status) VALUES (?, ?, ?, ?)",
|
||||||
|
buildParams(id, 103, LocalDateTime.now(), 0));
|
||||||
|
return false;
|
||||||
|
});
|
||||||
|
|
||||||
|
Optional<Map<String, Object>> first = jdbcTemplate
|
||||||
|
.queryFirst("SELECT * FROM base_table WHERE id = ?", buildParams(id));
|
||||||
|
log.info("first: {}", first);
|
||||||
|
assertTrue(!first.isPresent());
|
||||||
|
}
|
||||||
|
|
||||||
|
// 返回 true,提交事务
|
||||||
|
{
|
||||||
|
long id = this.idGenerator.nextId();
|
||||||
|
jdbcTemplate.commitIfTrue(jdbc -> {
|
||||||
|
jdbc.update("INSERT INTO base_table (id, created_by, create_time, status) VALUES (?, ?, ?, ?)",
|
||||||
|
buildParams(id, 104, LocalDateTime.now(), 0));
|
||||||
|
return true;
|
||||||
|
});
|
||||||
|
|
||||||
|
Optional<Map<String, Object>> first = jdbcTemplate
|
||||||
|
.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 {
|
||||||
|
List<int[]> result = jdbcTemplate.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;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void testBean() throws Exception {
|
||||||
|
Optional<TestBean> t = jdbcTemplate.queryFirst(
|
||||||
|
"SELECT * FROM test_table WHERE id = ?",
|
||||||
|
buildParams(22915),
|
||||||
|
RowMapper.beanRowMapper(TestBean.class));
|
||||||
|
log.info("t: {}", t);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
class TestBean {
|
||||||
|
Long id;
|
||||||
|
String username;
|
||||||
|
LocalDate usageDate;
|
||||||
|
Long usageDuration;
|
||||||
|
|
||||||
|
public Long getId() {
|
||||||
|
return id;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setId(Long id) {
|
||||||
|
this.id = id;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getUsername() {
|
||||||
|
return username;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setUsername(String username) {
|
||||||
|
this.username = username;
|
||||||
|
}
|
||||||
|
|
||||||
|
public LocalDate getUsageDate() {
|
||||||
|
return usageDate;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setUsageDate(LocalDate usageDate) {
|
||||||
|
this.usageDate = usageDate;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Long getUsageDuration() {
|
||||||
|
return usageDuration;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setUsageDuration(Long usageDuration) {
|
||||||
|
this.usageDuration = usageDuration;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String toString() {
|
||||||
|
return "TestBean [id=" + id + ", username=" + username + ", usageDate=" + usageDate + ", usageDuration="
|
||||||
|
+ usageDuration + "]";
|
||||||
|
}
|
||||||
|
}
|
Reference in New Issue
Block a user