threadLocalConnection = new ThreadLocal<>();
+
+ private final DSLoader.DSInfo dsInfo;
+
+ public static DewDB pick(String dsCode) {
+ if (!DBS.containsKey(dsCode)) {
+ synchronized (DBS) {
+ if (!DBS.containsKey(dsCode)) {
+ DBS.put(dsCode, new DewDB(DSLoader.getDSInfo(dsCode)));
+ }
+ }
+ }
+ return DBS.get(dsCode);
+ }
+
+ DewDB(DSLoader.DSInfo dsInfo) {
+ this.dsInfo = dsInfo;
+ }
+
+ public DSLoader.DSInfo getDsInfo() {
+ return dsInfo;
+ }
+
+ /**
+ * 创建表.
+ *
+ *
+ * @param tableName 表名
+ * @param tableDesc 表说明
+ * @param fields 表字段（字段名 - 类型）
+ * @param fieldsDesc 字段说明
+ * @param indexFields 索引字段
+ * @param uniqueFields 唯一值字段
+ * @param pkField 主键字段
+ * @throws SQLException SQL错误
+ * @deprecated - 此功能存在一定限制，建议使用 {@link #ddl(String)} 建表
+ */
+ @Deprecated
+ public void createTableIfNotExist(String tableName, String tableDesc,
+ Map fields,
+ Map fieldsDesc,
+ List indexFields,
+ List uniqueFields,
+ String pkField) throws SQLException {
+ tableName = tableName.toLowerCase();
+ DBExecutor.ddl(
+ dsInfo.getDialect().createTableIfNotExist(tableName, tableDesc,
+ fields, fieldsDesc, indexFields, uniqueFields, pkField),
+ getConnection(), isCloseConnection()
+ );
+ }
+
+ /**
+ * DDL操作.
+ *
+ * @param ddl DDL语句
+ * @throws SQLException SQL错误
+ */
+ public void ddl(String ddl) throws SQLException {
+ DBExecutor.ddl(ddl, getConnection(), isCloseConnection());
+ }
+
+ /**
+ * 获取单条记录.
+ *
+ * @param tableName 表名
+ * @param pkField 主键字段
+ * @param pkValue 主键值
+ * @param clazz 对象类
+ * @param 对象
+ * @return java对象
+ * @throws SQLException SQL错误
+ */
+ public E getByPk(String tableName, String pkField, Object pkValue, Class clazz) throws SQLException {
+ return get("SELECT * FROM " + tableName + " WHERE " + pkField + " = ?", clazz, pkValue);
+ }
+
+ /**
+ * 获取单个对象.
+ *
+ * @param sql SQL
+ * @param params 参数
+ * @param clazz 对象类
+ * @param 对象
+ * @return java对象
+ * @throws SQLException SQL错误
+ */
+ public E get(String sql, Class clazz, Object... params) throws SQLException {
+ return DBExecutor.get(sql, params, clazz, getConnection(), isCloseConnection());
+ }
+
+ /**
+ * 获取多个对象.
+ *
+ * @param sql SQL
+ * @param params 参数
+ * @param clazz 对象类
+ * @param 对象
+ * @return java对象
+ * @throws SQLException SQL错误
+ */
+ public List find(String sql, Class clazz, Object... params) throws SQLException {
+ return DBExecutor.find(sql, params, clazz, getConnection(), isCloseConnection());
+ }
+
+
+ /**
+ * 获取多个对象（带分页）.
+ *
+ * @param sql SQL
+ * @param params 参数
+ * @param pageNumber 页码（从1开始）
+ * @param pageSize 每页条数
+ * @param clazz 对象类
+ * @param 对象
+ * @return 多个对象（带分页）
+ * @throws SQLException SQL错误
+ */
+ public Page page(String sql, long pageNumber, long pageSize, Class clazz, Object... params) throws SQLException {
+ return DBExecutor.page(sql, params, pageNumber, pageSize, clazz, getConnection(), isCloseConnection(), dsInfo.getDialect());
+ }
+
+ /**
+ * 判断记录是否存在.
+ *
+ * @param tableName 表名
+ * @param pkField 主键字段
+ * @param pkValue 主键值
+ * @return 是否存在
+ * @throws SQLException SQL错误
+ */
+ public boolean exits(String tableName, String pkField, Object pkValue) throws SQLException {
+ return get("SELECT id FROM " + tableName + " WHERE " + pkField + " = ?", new Object[]{pkValue}).size() != 0;
+ }
+
+ /**
+ * 判断记录是否存在.
+ *
+ * @param sql SQL
+ * @param params 参数
+ * @return 是否存在
+ * @throws SQLException SQL错误
+ */
+ public boolean exits(String sql, Object... params) throws SQLException {
+ return count(sql, params) != 0;
+ }
+
+ /**
+ * 获取单条记录.
+ *
+ * @param tableName 表名
+ * @param pkField 主键字段
+ * @param pkValue 主键值
+ * @return 单条记录
+ * @throws SQLException SQL错误
+ */
+ public Map getByPk(String tableName, String pkField, Object pkValue) throws SQLException {
+ return get("SELECT * FROM " + tableName + " WHERE " + pkField + " = ?", pkValue);
+ }
+
+ /**
+ * 获取单条记录.
+ *
+ * @param sql SQL
+ * @param params 参数
+ * @return 单条记录
+ * @throws SQLException SQL错误
+ */
+ public Map get(String sql, Object... params) throws SQLException {
+ return DBExecutor.get(sql, params, getConnection(), isCloseConnection());
+ }
+
+ /**
+ * 获取多条记录.
+ *
+ * @param sql SQL
+ * @param params 参数
+ * @return 多条记录（带分页）
+ * @throws SQLException SQL错误
+ */
+ public List> find(String sql, Object... params) throws SQLException {
+ return DBExecutor.find(sql, params, getConnection(), isCloseConnection());
+ }
+
+ /**
+ * 获取多条记录（带分页）.
+ *
+ * @param sql SQL
+ * @param params 参数
+ * @param pageNumber 页码（从1开始）
+ * @param pageSize 每页条数
+ * @return 多条记录（带分页）
+ * @throws SQLException SQL错误
+ */
+ public Page> page(String sql, int pageNumber, int pageSize, Object... params) throws SQLException {
+ return DBExecutor.page(sql, params, pageNumber, pageSize, getConnection(), isCloseConnection(), dsInfo.getDialect());
+ }
+
+ /**
+ * 获取记录数.
+ *
+ * @param sql SQL
+ * @param params 参数
+ * @return 记录数
+ * @throws SQLException SQL错误
+ */
+ public long count(String sql, Object... params) throws SQLException {
+ return DBExecutor.count(sql, params, getConnection(), isCloseConnection(), dsInfo.getDialect());
+ }
+
+ /**
+ * 添加记录.
+ *
+ * @param tableName 表名
+ * @param values 值列表
+ * @return 影响行数
+ * @throws SQLException SQL错误
+ */
+ public int insert(String tableName, Map values) throws SQLException {
+ return DBExecutor.insert(tableName, values, getConnection(), isCloseConnection(), dsInfo.getDialect());
+ }
+
+ /**
+ * 修改记录.
+ *
+ * @param tableName 表名
+ * @param pkField 主键字段
+ * @param pkValue 主键值
+ * @param values 值列表
+ * @return 影响行数
+ * @throws SQLException SQL错误
+ */
+ public int modify(String tableName, String pkField, Object pkValue, Map values) throws SQLException {
+ return DBExecutor.modify(tableName, pkField, pkValue, values, getConnection(), isCloseConnection(), dsInfo.getDialect());
+ }
+
+ /**
+ * 更新记录.
+ *
+ * @param sql SQL
+ * @param params 参数
+ * @return 影响行数
+ * @throws SQLException SQL错误
+ */
+ public int update(String sql, Object... params) throws SQLException {
+ return DBExecutor.update(sql, params, getConnection(), isCloseConnection(), dsInfo.getDialect());
+ }
+
+ /**
+ * 批量更新记录.
+ *
+ * @param sql SQL
+ * @param params 参数
+ * @return 影响行数
+ * @throws SQLException SQL错误
+ */
+ public int[] batch(String sql, Object[][] params) throws SQLException {
+ return DBExecutor.batch(sql, params, getConnection(), isCloseConnection(), dsInfo.getDialect());
+ }
+
+ /**
+ * 批量更新记录.
+ *
+ * @param sqls SQL
+ * @throws SQLException SQL错误
+ */
+ public void batch(Map sqls) throws SQLException {
+ DBExecutor.batch(sqls, getConnection(), isCloseConnection(), dsInfo.getDialect());
+ }
+
+ /**
+ * 删除单条记录.
+ *
+ * @param tableName 表名
+ * @param pkField 主键字段
+ * @param pkValue 主键值
+ * @return 影响行数
+ * @throws SQLException SQL错误
+ */
+ public Integer delete(String tableName, String pkField, Object pkValue) throws SQLException {
+ return update("DELETE FROM " + tableName + " WHERE " + pkField + " = ?", pkValue);
+ }
+
+ /**
+ * 删除所有记录.
+ *
+ * @param tableName 表名
+ * @return 单条记录
+ * @throws SQLException SQL错误
+ */
+ public Integer deleteAll(String tableName) throws SQLException {
+ return update("DELETE FROM " + tableName);
+ }
+
+ /**
+ * 获取Meta信息.
+ *
+ * @param tableName 表名
+ * @return Meta信息
+ * @throws SQLException SQL错误
+ */
+ public List getMetaData(String tableName) throws SQLException {
+ return DBExecutor.getMetaData(tableName, getConnection());
+ }
+
+ /**
+ * 获取Meta信息.
+ *
+ * @param tableName 表名
+ * @param fieldName 指定的字段名
+ * @return Meta信息
+ * @throws SQLException SQL错误
+ */
+ public Meta getMetaData(String tableName, String fieldName) throws SQLException {
+ return DBExecutor.getMetaData(tableName, fieldName, getConnection());
+ }
+
+ /**
+ * 打开事务.
+ */
+ public void open() {
+ Connection conn = threadLocalConnection.get();
+ try {
+ if (null == conn) {
+ conn = getConnection();
+ threadLocalConnection.set(conn);
+ }
+ conn.setAutoCommit(false);
+ } catch (SQLException e) {
+ logger.error("[DewDBUtils]Connection open error", e);
+ }
+ }
+
+ /**
+ * 提交事务.
+ */
+ public void commit() {
+ Connection conn = threadLocalConnection.get();
+ if (null != conn) {
+ try {
+ conn.commit();
+ } catch (SQLException e) {
+ logger.error("[DewDBUtils]Connection commit error", e);
+ }
+ }
+ close();
+ }
+
+ /**
+ * 显式回滚事务.
+ *
+ * 发生SQL错误时会自动回滚，但业务错误需要调用此方法手工回滚.
+ */
+ public void rollback() {
+ Connection conn = threadLocalConnection.get();
+ if (null != conn) {
+ try {
+ conn.rollback();
+ } catch (SQLException e) {
+ logger.error("[DewDBUtils]Connection rollback error", e);
+ }
+ }
+ close();
+ }
+
+ private boolean isCloseConnection() {
+ return null == threadLocalConnection.get();
+ }
+
+ private void close() {
+ Connection conn = threadLocalConnection.get();
+ if (null != conn) {
+ try {
+ if (!conn.isClosed()) {
+ conn.close();
+ }
+ threadLocalConnection.set(null);
+ } catch (SQLException e) {
+ logger.error("[DewDBUtils]Connection close error", e);
+ }
+ }
+ }
+
+ @SneakyThrows
+ private Connection getConnection() {
+ try {
+ if (threadLocalConnection.get() != null && !threadLocalConnection.get().isClosed()) {
+ return threadLocalConnection.get();
+ }
+ Connection conn = dsInfo.getDataSource().getConnection();
+ if (!conn.isClosed()) {
+ return conn;
+ }
+ //Re-setting connection when connection was close.
+ synchronized (DSLoader.class) {
+ logger.warn("[DewDBUtils]Connection info [{}] was close", conn.toString());
+ DSLoader.loadPool(dsInfo.getDsConfig(), dsInfo.getDialect());
+ return dsInfo.getDataSource().getConnection();
+ }
+ } catch (SQLException e) {
+ logger.error("[DewDBUtils]Connection get error.", e);
+ throw e;
+ }
+ }
+
+}
diff --git a/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/DewDBUtils.java b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/DewDBUtils.java
new file mode 100644
index 00000000..e40a152a
--- /dev/null
+++ b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/DewDBUtils.java
@@ -0,0 +1,83 @@
+/*
+ * Copyright 2020. 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
+ *
+ * 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.
+ */
+
+package group.idealworld.dew.core.dbutils;
+
+import group.idealworld.dew.core.dbutils.dto.DBUtilsConfig;
+import group.idealworld.dew.core.dbutils.dto.DSConfig;
+import group.idealworld.dew.core.dbutils.process.DSLoader;
+
+/**
+ * 操作入口类.
+ *
+ * @author gudaoxuri
+ */
+public class DewDBUtils {
+
+ /**
+ * 初始化数据源.
+ *
+ * @param configPath 配置文件路径
+ */
+ public static void init(String configPath) {
+ DSLoader.load(configPath);
+ }
+
+ public static void init2(DBUtilsConfig dbUtilsConfig){
+ DSLoader.load2(dbUtilsConfig);
+ }
+
+ /**
+ * 添加数据源.
+ *
+ * @param dsConfig 配置文件
+ */
+ public static void addDS(DSConfig dsConfig) {
+ DSLoader.addDS(dsConfig);
+ }
+
+ /**
+ * 删除数据源.
+ *
+ * @param dsCode 数据源编码
+ */
+ public static void removeDS(String dsCode) {
+ DSLoader.removeDS(dsCode);
+ }
+
+ /**
+ * 删除数据源.
+ *
+ * @param dewDB DB实例
+ */
+ public static void removeDS(DewDB dewDB) {
+ if (dewDB != null
+ && dewDB.getDsInfo() != null
+ && dewDB.getDsInfo().getDsConfig() != null) {
+ DSLoader.removeDS(dewDB.getDsInfo().getDsConfig().getCode());
+ }
+ }
+
+ /**
+ * 选择DB实例.
+ *
+ * @param dsCode 数据源编码
+ * @return DB实例
+ */
+ public static DewDB use(String dsCode) {
+ return DewDB.pick(dsCode);
+ }
+}
diff --git a/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dialect/Dialect.java b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dialect/Dialect.java
new file mode 100644
index 00000000..d7e0c251
--- /dev/null
+++ b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dialect/Dialect.java
@@ -0,0 +1,40 @@
+/*
+ * Copyright 2020. 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
+ *
+ * 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.
+ */
+
+package group.idealworld.dew.core.dbutils.dialect;
+
+import java.sql.SQLException;
+import java.util.List;
+import java.util.Map;
+
+public interface Dialect {
+
+ String paging(String sql, long pageNumber, long pageSize) throws SQLException;
+
+ String count(String sql) throws SQLException;
+
+ String getTableInfo(String tableName) throws SQLException;
+
+ @Deprecated
+ String createTableIfNotExist(String tableName, String tableDesc, Map fields, Map fieldsDesc,
+ List indexFields, List uniqueFields, String pkField) throws SQLException;
+
+ String validationQuery();
+
+ String getDriver();
+
+ DialectType getDialectType();
+}
diff --git a/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dialect/DialectFactory.java b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dialect/DialectFactory.java
new file mode 100644
index 00000000..8e1badee
--- /dev/null
+++ b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dialect/DialectFactory.java
@@ -0,0 +1,35 @@
+/*
+ * Copyright 2020. 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
+ *
+ * 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.
+ */
+
+package group.idealworld.dew.core.dbutils.dialect;
+
+
+public class DialectFactory {
+
+ public static Dialect parseDialect(String url) {
+ if (url.startsWith("jdbc:h2")) {
+ return new H2Dialect();
+ } else if (url.startsWith("jdbc:mysql")) {
+ return new MySQLDialect();
+ } else if (url.startsWith("jdbc:postgresql")) {
+ return new PostgresDialect();
+ } else if (url.startsWith("jdbc:hive2")) {
+ return new HiveDialect();
+ }
+ return null;
+ }
+
+}
diff --git a/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dialect/DialectType.java b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dialect/DialectType.java
new file mode 100644
index 00000000..dc73c4a9
--- /dev/null
+++ b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dialect/DialectType.java
@@ -0,0 +1,23 @@
+/*
+ * Copyright 2020. 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
+ *
+ * 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.
+ */
+
+package group.idealworld.dew.core.dbutils.dialect;
+
+public enum DialectType {
+
+ MYSQL,POSTGRE,H2,HIVE
+
+}
diff --git a/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dialect/H2Dialect.java b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dialect/H2Dialect.java
new file mode 100644
index 00000000..26c47823
--- /dev/null
+++ b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dialect/H2Dialect.java
@@ -0,0 +1,116 @@
+/*
+ * Copyright 2020. 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
+ *
+ * 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.
+ */
+
+package group.idealworld.dew.core.dbutils.dialect;
+
+import java.sql.SQLException;
+import java.util.List;
+import java.util.Map;
+import java.util.Objects;
+
+class H2Dialect implements Dialect {
+
+ @Override
+ public String paging(String sql, long pageNumber, long pageSize) throws SQLException {
+ return sql + " LIMIT " + pageSize + " OFFSET " + (pageNumber - 1) * pageSize;
+ }
+
+ @Override
+ public String count(String sql) throws SQLException {
+ return "SELECT COUNT(1) FROM ( " + sql + " ) ";
+ }
+
+ @Override
+ public String getTableInfo(String tableName) throws SQLException {
+ return "SELECT * FROM INFORMATION_SCHEMA.TABLES t WHERE t.table_name = '" + tableName + "'";
+ }
+
+ @Override
+ public String createTableIfNotExist(String tableName, String tableDesc, Map fields, Map fieldsDesc, List indexFields, List uniqueFields, String pkField) throws SQLException {
+ StringBuilder sb = new StringBuilder("CREATE TABLE IF NOT EXISTS " + tableName + " ( ");
+ for (Map.Entry field : fields.entrySet()) {
+ String f = field.getValue().toLowerCase();
+ String t;
+ switch (f) {
+ case "int":
+ case "integer":
+ t = "INT";
+ break;
+ case "long":
+ t = "BIGINT";
+ break;
+ case "short":
+ t = "SMALLINT";
+ break;
+ case "string":
+ t = "VARCHAR(65535)";
+ break;
+ case "text":
+ t = "TEXT";
+ break;
+ case "bool":
+ case "boolean":
+ t = "BOOLEAN";
+ break;
+ case "float":
+ // https://www.h2database.com/html/faq.html?highlight=float&search=FLOAT#float_is_double
+ t = "FLOAT(24)";
+ break;
+ case "double":
+ t = "DOUBLE";
+ break;
+ case "char":
+ t = "CHAR";
+ break;
+ case "date":
+ t = "TIMESTAMP";
+ break;
+ case "uuid":
+ t = "UUID";
+ break;
+ case "bigdecimal":
+ case "decimal":
+ t = "DECIMAL";
+ break;
+ default:
+ throw new SQLException("Not support type:" + f);
+ }
+ sb.append(field.getKey()).append(" ").append(t).append(" ,");
+ }
+ if (pkField != null && !Objects.equals(pkField.trim(), "")) {
+ return sb.append("primary key(").append(pkField.trim()).append(") )").toString();
+ } else {
+ return sb.substring(0, sb.length() - 1) + ")";
+ }
+ //TODO
+ }
+
+ @Override
+ public String validationQuery() {
+ return "SELECT 1";
+ }
+
+ @Override
+ public String getDriver() {
+ return "org.h2.Driver";
+ }
+
+ @Override
+ public DialectType getDialectType() {
+ return DialectType.H2;
+ }
+
+}
diff --git a/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dialect/HiveDialect.java b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dialect/HiveDialect.java
new file mode 100644
index 00000000..79c0c1b3
--- /dev/null
+++ b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dialect/HiveDialect.java
@@ -0,0 +1,104 @@
+/*
+ * Copyright 2020. 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
+ *
+ * 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.
+ */
+
+package group.idealworld.dew.core.dbutils.dialect;
+
+import java.sql.SQLException;
+import java.util.List;
+import java.util.Map;
+
+class HiveDialect implements Dialect {
+
+ @Override
+ public String paging(String sql, long pageNumber, long pageSize) throws SQLException {
+ return sql + " LIMIT " + (pageNumber - 1) * pageSize + ", " + pageSize;
+ }
+
+ @Override
+ public String count(String sql) throws SQLException {
+ return "SELECT COUNT(1) FROM ( " + sql + " ) _" + System.currentTimeMillis();
+ }
+
+ @Override
+ public String getTableInfo(String tableName) throws SQLException {
+ //TODO
+ throw new RuntimeException("NotImplementedException");
+ }
+
+ @Override
+ public String createTableIfNotExist(String tableName, String tableDesc, Map fields, Map fieldsDesc, List indexFields, List uniqueFields, String pkField) throws SQLException {
+ //TODO
+ throw new RuntimeException("NotImplementedException");
+ /*StringBuilder sb = new StringBuilder("CREATE TABLE IF NOT EXISTS " + tableName + " ( ");
+ for (Map.Entry field : fields.entrySet()) {
+ String f = field.getValue().toLowerCase();
+ String t;
+ switch (f) {
+ case "int":
+ case "integer":
+ t = "INT";
+ break;
+ case "long":
+ t = "BIGINT";
+ break;
+ case "short":
+ t = "SMALLINT";
+ break;
+ case "string":
+ t = "STRING";
+ break;
+ case "bool":
+ case "boolean":
+ t = "BOOLEAN";
+ break;
+ case "float":
+ t = "FLOAT";
+ break;
+ case "double":
+ t = "DOUBLE";
+ break;
+ case "char":
+ t = "CHAR";
+ break;
+ case "date":
+ t = "TIMESTAMP";
+ break;
+ case "decimal":
+ t = "DECIMAL";
+ break;
+ default:
+ throw new SQLException("Not support type:" + f);
+ }
+ sb.append(field.getKey()).append(" ").append(t).append(" ,");
+ }
+ return sb.substring(0, sb.length() - 1) + ")";*/
+ }
+
+ @Override
+ public String validationQuery() {
+ return "SELECT 1";
+ }
+
+ @Override
+ public String getDriver() {
+ return "org.apache.hive.jdbc.HiveDriver";
+ }
+
+ @Override
+ public DialectType getDialectType() {
+ return DialectType.HIVE;
+ }
+}
diff --git a/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dialect/MySQLDialect.java b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dialect/MySQLDialect.java
new file mode 100644
index 00000000..01cd1b79
--- /dev/null
+++ b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dialect/MySQLDialect.java
@@ -0,0 +1,116 @@
+/*
+ * Copyright 2020. 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
+ *
+ * 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.
+ */
+
+package group.idealworld.dew.core.dbutils.dialect;
+
+import java.sql.SQLException;
+import java.util.List;
+import java.util.Map;
+import java.util.Objects;
+
+class MySQLDialect implements Dialect {
+
+ @Override
+ public String paging(String sql, long pageNumber, long pageSize) throws SQLException {
+ return sql + " LIMIT " + (pageNumber - 1) * pageSize + ", " + pageSize;
+ }
+
+ @Override
+ public String count(String sql) throws SQLException {
+ return "SELECT COUNT(1) FROM ( " + sql + " ) _" + System.currentTimeMillis();
+ }
+
+ @Override
+ public String getTableInfo(String tableName) throws SQLException {
+ //TODO
+ throw new RuntimeException("NotImplementedException");
+ }
+
+ @Override
+ public String createTableIfNotExist(String tableName, String tableDesc, Map fields, Map fieldsDesc, List indexFields, List uniqueFields, String pkField) throws SQLException {
+ //TODO
+ StringBuilder sb = new StringBuilder("CREATE TABLE IF NOT EXISTS " + tableName + " ( ");
+ for (Map.Entry field : fields.entrySet()) {
+ String f = field.getValue().toLowerCase();
+ String t;
+ switch (f) {
+ case "int":
+ case "integer":
+ t = "INT";
+ break;
+ case "long":
+ t = "BIGINT";
+ break;
+ case "short":
+ t = "SMALLINT";
+ break;
+ case "string":
+ t = "VARCHAR(2000)";
+ break;
+ case "text":
+ t = "TEXT";
+ break;
+ case "bool":
+ case "boolean":
+ t = "BOOLEAN";
+ break;
+ case "float":
+ t = "FLOAT";
+ break;
+ case "double":
+ t = "DOUBLE";
+ break;
+ case "char":
+ t = "CHAR";
+ break;
+ case "date":
+ t = "TIMESTAMP";
+ break;
+ case "uuid":
+ t = "UUID";
+ break;
+ case "bigdecimal":
+ case "decimal":
+ t = "DECIMAL";
+ break;
+ default:
+ throw new SQLException("Not support type:" + f);
+ }
+ sb.append(field.getKey()).append(" ").append(t).append(" ,");
+ }
+ if (pkField != null && !Objects.equals(pkField.trim(), "")) {
+ return sb.append("primary key(").append(pkField.trim()).append(") )").toString();
+ } else {
+ return sb.substring(0, sb.length() - 1) + ")";
+ }
+ }
+
+ @Override
+ public String validationQuery() {
+ return "SELECT 1";
+ }
+
+ @Override
+ public String getDriver() {
+ return "com.mysql.jdbc.Driver";
+ }
+
+ @Override
+ public DialectType getDialectType() {
+ return DialectType.MYSQL;
+ }
+
+}
diff --git a/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dialect/PostgresDialect.java b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dialect/PostgresDialect.java
new file mode 100644
index 00000000..5ff49b24
--- /dev/null
+++ b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dialect/PostgresDialect.java
@@ -0,0 +1,128 @@
+/*
+ * Copyright 2020. 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
+ *
+ * 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.
+ */
+
+package group.idealworld.dew.core.dbutils.dialect;
+
+import java.sql.SQLException;
+import java.util.List;
+import java.util.Map;
+import java.util.Objects;
+
+class PostgresDialect implements Dialect {
+
+ @Override
+ public String paging(String sql, long pageNumber, long pageSize) throws SQLException {
+ return sql + " LIMIT " + pageSize + " OFFSET " + (pageNumber - 1) * pageSize;
+ }
+
+ @Override
+ public String count(String sql) throws SQLException {
+ return "SELECT COUNT(1) FROM ( " + sql + " ) _" + System.currentTimeMillis();
+ }
+
+ @Override
+ public String getTableInfo(String tableName) throws SQLException {
+ return "SELECT * FROM pg_tables t WHERE t.tablename = '" + tableName + "'";
+ }
+
+ @Override
+ public String createTableIfNotExist(String tableName, String tableDesc, Map fields, Map fieldsDesc, List indexFields, List uniqueFields, String pkField) throws SQLException {
+ StringBuilder sb = new StringBuilder("CREATE TABLE IF NOT EXISTS " + tableName + " ( ");
+ for (Map.Entry field : fields.entrySet()) {
+ String f = field.getValue().toLowerCase();
+ String t;
+ switch (f) {
+ case "seq":
+ t = "serial";
+ break;
+ case "int":
+ case "integer":
+ case "short":
+ t = "integer";
+ break;
+ case "long":
+ t = "bigint";
+ break;
+ case "string":
+ t = "character varying(65535)";
+ break;
+ case "text":
+ t = "text";
+ break;
+ case "bool":
+ case "boolean":
+ t = "boolean";
+ break;
+ case "float":
+ case "double":
+ t = "double precision";
+ break;
+ case "char":
+ t = "character";
+ break;
+ case "date":
+ t = "date";
+ break;
+ case "bigdecimal":
+ case "decimal":
+ t = "numeric";
+ break;
+ default:
+ throw new SQLException("Not support type:" + f);
+ }
+ sb.append(field.getKey()).append(" ").append(t).append(" ,");
+ }
+ if (uniqueFields != null && !uniqueFields.isEmpty()) {
+ for (String uField : uniqueFields) {
+ sb.append("CONSTRAINT \"u_").append(tableName).append("_").append(uField).append("\" UNIQUE (\"").append(uField).append("\"),");
+ }
+ }
+ if (pkField != null && !Objects.equals(pkField.trim(), "")) {
+ sb.append("primary key(").append(pkField.trim()).append(") );");
+ } else {
+ sb = new StringBuilder(sb.substring(0, sb.length() - 1) + ");");
+ }
+ if (indexFields != null && !indexFields.isEmpty()) {
+ for (String idxFields : indexFields) {
+ sb.append("CREATE INDEX \"i_").append(tableName).append("_").append(idxFields).append("\" ON \"").append(tableName).append("\" (\"").append(idxFields).append("\");");
+ }
+ }
+ if (tableDesc != null && !tableDesc.isEmpty()) {
+ sb.append("COMMENT ON TABLE \"").append(tableName).append("\" IS '").append(tableDesc).append("';");
+ }
+ if (fieldsDesc != null && !fieldsDesc.isEmpty()) {
+ for (Map.Entry field : fieldsDesc.entrySet()) {
+ sb.append("COMMENT ON COLUMN \"").append(tableName).append("\".\"").append(field.getKey()).append("\" IS '").append(field.getValue()).append("';");
+ }
+ }
+ return sb.toString();
+ }
+
+ @Override
+ public String validationQuery() {
+ return "SELECT 'x'";
+ }
+
+ @Override
+ public String getDriver() {
+ return "org.postgresql.Driver";
+ }
+
+ @Override
+ public DialectType getDialectType() {
+ return DialectType.POSTGRE;
+ }
+}
diff --git a/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dto/DBUtilsConfig.java b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dto/DBUtilsConfig.java
new file mode 100644
index 00000000..76c4cd7f
--- /dev/null
+++ b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dto/DBUtilsConfig.java
@@ -0,0 +1,46 @@
+/*
+ * Copyright 2020. 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
+ *
+ * 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.
+ */
+
+package group.idealworld.dew.core.dbutils.dto;
+
+import lombok.Data;
+import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.stereotype.Component;
+
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * 配置类
+ *
+ * @author gudaoxuri
+ */
+@ConfigurationProperties(prefix = "ds")
+@Data
+public class DBUtilsConfig {
+
+ private List ds = new ArrayList<>();
+ private DynamicDS dynamicDS = new DynamicDS();
+
+ @Data
+ public static class DynamicDS {
+
+ private Boolean enabled = false;
+ private String dsCode;
+ private String fetchSql = "select code,url,username,password,monitor,pool_initialSize,pool_maxActive from multi_ds";
+
+ }
+}
diff --git a/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dto/DSConfig.java b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dto/DSConfig.java
new file mode 100644
index 00000000..b542a946
--- /dev/null
+++ b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dto/DSConfig.java
@@ -0,0 +1,59 @@
+/*
+ * Copyright 2020. 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
+ *
+ * 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.
+ */
+
+package group.idealworld.dew.core.dbutils.dto;
+
+import lombok.Builder;
+import lombok.Data;
+import lombok.experimental.Tolerate;
+import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.stereotype.Component;
+
+/**
+ * @author gudaoxuri
+ */
+@Data
+@Builder
+public class DSConfig {
+
+ private String code;
+ private String url;
+ private String username;
+ private String password;
+ @Builder.Default
+ private Boolean monitor = false;
+ @Builder.Default
+ private PoolConfig pool = new PoolConfig();
+
+ @Data
+ @Builder
+ public static class PoolConfig {
+
+ @Builder.Default
+ private Integer initialSize = 5;
+ @Builder.Default
+ private Integer maxActive = 20;
+
+ @Tolerate
+ public PoolConfig() {
+ }
+
+ }
+
+ @Tolerate
+ public DSConfig() {
+ }
+}
diff --git a/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dto/Meta.java b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dto/Meta.java
new file mode 100644
index 00000000..ccb9de07
--- /dev/null
+++ b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dto/Meta.java
@@ -0,0 +1,33 @@
+/*
+ * Copyright 2020. 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
+ *
+ * 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.
+ */
+
+package group.idealworld.dew.core.dbutils.dto;
+
+import lombok.Data;
+
+@Data
+public class Meta {
+
+ private int type;
+ private String code;
+ private String label;
+
+ public Meta(int type, String code, String label) {
+ this.type = type;
+ this.code = code;
+ this.label = label;
+ }
+}
diff --git a/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dto/Page.java b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dto/Page.java
new file mode 100644
index 00000000..6f890523
--- /dev/null
+++ b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/dto/Page.java
@@ -0,0 +1,36 @@
+/*
+ * Copyright 2020. 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
+ *
+ * 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.
+ */
+
+package group.idealworld.dew.core.dbutils.dto;
+
+import lombok.Data;
+
+import java.util.List;
+
+/**
+ * 分页辅助类
+ */
+@Data
+public class Page {
+
+ //start with 1
+ private long pageNumber;
+ private long pageSize;
+ private long pageTotal;
+ private long recordTotal;
+ private List objects;
+
+}
diff --git a/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/process/DBExecutor.java b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/process/DBExecutor.java
new file mode 100644
index 00000000..e66eca5b
--- /dev/null
+++ b/framework/modules/dbutils-starter/src/main/java/group/idealworld/dew/core/dbutils/process/DBExecutor.java
@@ -0,0 +1,386 @@
+/*
+ * Copyright 2020. 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
+ *
+ * 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.
+ */
+
+package group.idealworld.dew.core.dbutils.process;
+
+import group.idealworld.dew.core.dbutils.dialect.Dialect;
+import group.idealworld.dew.core.dbutils.dialect.DialectType;
+import group.idealworld.dew.core.dbutils.dto.Meta;
+import group.idealworld.dew.core.dbutils.dto.Page;
+import lombok.SneakyThrows;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.dbutils.QueryRunner;
+import org.apache.commons.dbutils.handlers.*;
+
+import java.io.BufferedReader;
+import java.io.Reader;
+import java.math.BigDecimal;
+import java.sql.*;
+import java.util.ArrayList;
+import java.util.LinkedHashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.stream.Collectors;
+
+@Slf4j
+public class DBExecutor {
+
+ static final QueryRunner queryRunner = new QueryRunner();
+
+ public static E get(String sql, Object[] params, Class clazz, Connection conn, boolean isCloseConn) throws SQLException {
+ try {
+ if (params == null) {
+ return (E) queryRunner.query(conn, sql, new BeanHandler(clazz));
+ } else {
+ return (E) queryRunner.query(conn, sql, new BeanHandler(clazz), params);
+ }
+ } catch (SQLException e) {
+ log.error("[DewDBUtils]Get error : " + sql, e);
+ throw e;
+ } finally {
+ if (isCloseConn) {
+ closeConnection(conn);
+ }
+ }
+ }
+
+ public static List find(String sql, Object[] params, Class clazz, Connection conn, boolean isCloseConn) throws SQLException {
+ try {
+ if (null == params) {
+ return (List) queryRunner.query(conn, sql, new BeanListHandler(clazz));
+ } else {
+ return (List) queryRunner.query(conn, sql, new BeanListHandler(clazz), params);
+ }
+ } catch (SQLException e) {
+ log.error("[DewDBUtils]Find error : " + sql, e);
+ throw e;
+ } finally {
+ if (isCloseConn) {
+ closeConnection(conn);
+ }
+ }
+ }
+
+ public static Page page(String sql, Object[] params, long pageNumber, long pageSize, Class clazz, Connection conn, boolean isCloseConn, Dialect dialect) throws SQLException {
+ Page page = new Page<>();
+ String pagedSql = dialect.paging(sql, pageNumber, pageSize);
+ page.setPageNumber(pageNumber);
+ page.setPageSize(pageSize);
+ page.setRecordTotal(count(sql, params, conn, false, dialect));
+ page.setPageTotal((page.getRecordTotal() + pageSize - 1) / pageSize);
+ page.setObjects(find(pagedSql, params, clazz, conn, isCloseConn));
+ return page;
+ }
+
+ public static Map get(String sql, Object[] params, Connection conn, boolean isCloseConn) throws SQLException {
+ try {
+ Map result;
+ if (null == params) {
+ result = queryRunner.query(conn, sql, new MapHandler());
+ } else {
+ result = queryRunner.query(conn, sql, new MapHandler(), params);
+ }
+ if (result != null) {
+ Map lowCaseResult = new LinkedHashMap<>();
+ for (Map.Entry entry : result.entrySet()) {
+ if (entry.getValue() instanceof Clob) {
+ lowCaseResult.put(entry.getKey().toLowerCase(), convertClob((Clob) entry.getValue()));
+ } else {
+ lowCaseResult.put(entry.getKey().toLowerCase(), entry.getValue());
+ }
+ }
+ result = lowCaseResult;
+ }
+ return result;
+ } catch (SQLException e) {
+ log.error("[DewDBUtils]Get error : " + sql, e);
+ throw e;
+ } finally {
+ if (isCloseConn) {
+ closeConnection(conn);
+ }
+ }
+ }
+
+ public static List> find(String sql, Object[] params, Connection conn, boolean isCloseConn) throws SQLException {
+ try {
+ List> result;
+ if (null == params) {
+ result = queryRunner.query(conn, sql, new MapListHandler());
+ } else {
+ result = queryRunner.query(conn, sql, new MapListHandler(), params);
+ }
+ if (result != null && !result.isEmpty()) {
+ List> lowCaseResult = new ArrayList<>();
+ for (Map item : result) {
+ Map lowCaseItem = new LinkedHashMap<>();
+ for (Map.Entry entry : item.entrySet()) {
+ if (entry.getValue() instanceof Clob) {
+ lowCaseItem.put(entry.getKey().toLowerCase(), convertClob((Clob) entry.getValue()));
+ } else {
+ lowCaseItem.put(entry.getKey().toLowerCase(), entry.getValue());
+ }
+ }
+ lowCaseResult.add(lowCaseItem);
+ }
+ result = lowCaseResult;
+ }
+ return result;
+ } catch (SQLException e) {
+ log.error("[DewDBUtils]Find error : " + sql, e);
+ throw e;
+ } finally {
+ if (isCloseConn) {
+ closeConnection(conn);
+ }
+ }
+ }
+
+ public static Page> page(String sql, Object[] params, long pageNumber, long pageSize, Connection conn, boolean isCloseConn, Dialect dialect) throws SQLException {
+ Page> page = new Page<>();
+ String pagedSql = dialect.paging(sql, pageNumber, pageSize);
+ page.setPageNumber(pageNumber);
+ page.setPageSize(pageSize);
+ page.setRecordTotal(count(sql, params, conn, false, dialect));
+ page.setPageTotal((page.getRecordTotal() + pageSize - 1) / pageSize);
+ page.setObjects(find(pagedSql, params, conn, isCloseConn));
+ return page;
+ }
+
+ public static long count(String sql, Connection conn, boolean isCloseConn, Dialect dialect) throws SQLException {
+ return count(sql, null, conn, isCloseConn, dialect);
+ }
+
+ public static long count(String sql, Object[] params, Connection conn, boolean isCloseConn, Dialect dialect) throws SQLException {
+ String countSql = dialect.count(sql);
+ try {
+ if (null == params) {
+ return (Long) queryRunner.query(conn, countSql, scalarHandler);
+ } else {
+ return (Long) queryRunner.query(conn, countSql, scalarHandler, params);
+ }
+
+ } catch (SQLException e) {
+ log.error("[DewDBUtils]Count error : " + countSql, e);
+ throw e;
+ } finally {
+ if (isCloseConn) {
+ closeConnection(conn);
+ }
+ }
+ }
+
+ public static int insert(String tableName, Map values,
+ Connection conn, boolean closeConnection, Dialect dialect) throws SQLException {
+ String fields = String.join(",", values.keySet());
+ String valueArgs = values.keySet().stream().map(f -> "?").collect(Collectors.joining(","));
+ String sql = "INSERT INTO " + tableName + " (" + fields + ") VALUES (" + valueArgs + ")";
+ return update(sql, values.values().toArray(), conn, closeConnection, dialect);
+ }
+
+ public static int modify(String tableName, String pkField, Object pkValue, Map values,
+ Connection conn, boolean closeConnection, Dialect dialect) throws SQLException {
+ String set = values.keySet().stream().map(k -> k + " = ?").collect(Collectors.joining(", "));
+ String sql = "UPDATE " + tableName + " SET " + set + " WHERE " + pkField + " = ? ";
+ List