xin.xihc.jba.core.JbaTemplate Maven / Gradle / Ivy
package xin.xihc.jba.core;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate;
import org.springframework.stereotype.Component;
import org.springframework.transaction.annotation.EnableTransactionManagement;
import xin.xihc.jba.core.bean.JbaBeanProperty;
import xin.xihc.jba.core.bean.JbaBeanPropertyRowMapper;
import xin.xihc.jba.core.bean.JbaMapSqlSource;
import xin.xihc.jba.core.utils.JbaLog;
import xin.xihc.jba.core.utils.SQLUtils;
import xin.xihc.jba.scan.TableManager;
import xin.xihc.jba.scan.tables.properties.ColumnProperties;
import xin.xihc.jba.scan.tables.properties.TableProperties;
import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.stream.Collectors;
/**
* 对JDBCTemplate进行封装
*
* @author Leo.Xi
* @date 2017年7月16日
* @code 对NamedParameterJdbcTemplate和JdbcTemplate进行封装
* @since 1.1.7
*/
@Component
@EnableTransactionManagement
public class JbaTemplate {
@Autowired
private NamedParameterJdbcTemplate namedParameterJdbcTemplate;
/** 数据库产品名 */
private String databaseName;
/**
* 开放出去,可以自己new实例,同时管理多个数据源
*
* @param dataSource 数据源
*/
@Autowired
public JbaTemplate(DataSource dataSource) {
super();
this.namedParameterJdbcTemplate = new NamedParameterJdbcTemplate(dataSource);
try {
Connection connection = dataSource.getConnection();
this.databaseName = connection.getMetaData().getDatabaseProductName();
} catch (SQLException e) {
// ignore
// e.printStackTrace();
}
}
/**
* 获取当前的JdbcTemplate
*
* @return JdbcTemplate
* @since 1.1.7
*/
public JdbcTemplate getJdbcTemplate() {
if (null != this.namedParameterJdbcTemplate) {
return (JdbcTemplate) this.namedParameterJdbcTemplate.getJdbcOperations();
}
return null;
}
/**
* 获取当前数据源的数据库品牌名称
*
* @return "Apache Derby",
* "DB2",
* "MySQL",
* "Microsoft SQL Server",
* "Oracle",
* "PostgreSQL"
* @since 1.8.2
*/
public String getDatabaseName() {
return this.databaseName;
}
/*--------------------------------------------------------------------------------
*
* 以下为【NamedParameterJdbcTemplate】对数据库操作
* 可以使用注解事务 - @Transactional
*
* ------------------------------------------------------------------------------*/
// ===================对表对象进行操作==========================
/**
* 插入数据库 INSERT INTO tblName (id,name) VALUES (:id,:name);
*
* @param model 对象模型
* @return 是否成功
* @code :id后面的必须和model对象的属性名一致
*/
public boolean insertModel(Object model) {
Objects.requireNonNull(model, "表对象model不允许为空");
boolean ret = false;
String sql = SQLUtils.getInsertSql(model);
long start = System.currentTimeMillis();// 记录开始时间戳
int t = namedParameterJdbcTemplate.update(sql, new JbaBeanProperty(model));
JbaLog.infoSql(sql, model, start);
if (t > 0) {
ret = true;
}
return ret;
}
/**
* 批量插入数据库
* 以对象值最多的生成sql
*
* @param models 表对象数组
*/
public void insertModels(Object[] models) {
Objects.requireNonNull(models, "表对象models不允许为空");
if (models.length < 1) {// 为空暂时不处理
return;
}
String sql = "";
String temp = "";
for (int i = 0, l = models.length; i < l; i++) {
temp = SQLUtils.getInsertSql(models[i]);
if (sql.length() < temp.length()) { // 以参数最多的那个为准
sql = temp;
}
}
batchUpdate(sql, models);
}
/**
* 根据字段更新数据
*
* @param model 对象模型
* @param fieldNames where子句条件字段数组,对象属性名（注意大小写）
* @return 是否成功
*/
public boolean updateModel(Object model, final String... fieldNames) throws RuntimeException {
Objects.requireNonNull(model, "表对象model不允许为空");
boolean ret = false;
String sql = SQLUtils.getUpdateSql(model, fieldNames);
long start = System.currentTimeMillis();// 记录开始时间戳
int t = namedParameterJdbcTemplate.update(sql, new JbaBeanProperty(model));
JbaLog.infoSql(sql, model, start);
if (t > 0) {
ret = true;
}
return ret;
}
/**
* 批量执行sql，插入INSERT、UPDATE都可以
*
* @param sql sql语句
* @param params SQL语句中对应的参数
*/
public void batchUpdate(final String sql, Map[] params) {
Objects.requireNonNull(params, "params不允许为空");
if (params.length < 1) {// 为空暂时不处理
return;
}
long start = System.currentTimeMillis();// 记录开始时间戳
namedParameterJdbcTemplate.batchUpdate(sql, params);
JbaLog.infoSql(sql, params, start);
}
/**
* 批量执行sql，插入INSERT、UPDATE都可以
*
* @param sql sql语句
* @param models SQL语句中对应的参数,非Map对象
*/
public void batchUpdate(final String sql, Object[] models) {
Objects.requireNonNull(models, "models不允许为空");
if (models.length < 1) {// 为空暂时不处理
return;
}
long start = System.currentTimeMillis();// 记录开始时间戳
JbaBeanProperty[] params = new JbaBeanProperty[models.length];
for (int i = 0, l = models.length; i < l; i++) {
params[i] = new JbaBeanProperty(models[i]);
}
namedParameterJdbcTemplate.batchUpdate(sql, params);
JbaLog.infoSql(sql, models, start);
}
/**
* 根据字段删除数据
*
* @param model 对象模型
* @return 是否成功
*/
public boolean deleteModel(Object model) throws RuntimeException {
Objects.requireNonNull(model, "表对象model不允许为空");
boolean ret = false;
String sql = SQLUtils.getDeleteSql(model);
// 不允许通过这里清空表
if (sql.toLowerCase().indexOf("where") < 1) {
// JbaLog.error("删除数据sql：" + sql + "\r\n不允许清空表数据,请使用sql清空");
// return false;
throw new RuntimeException("删除数据sql：" + sql + "\r\n不允许清空表数据,请使用sql清空");
}
long start = System.currentTimeMillis();// 记录开始时间戳
int t = namedParameterJdbcTemplate.update(sql, new JbaBeanProperty(model));
JbaLog.infoSql(sql, model, start);
if (t > 0) {
ret = true;
}
return ret;
}
/**
* 执行sql语句
*
* @param sql SQL
* @return 是否执行成功
*/
public boolean executeSQL(final String sql) {
boolean res = false;
long start = System.currentTimeMillis();// 记录开始时间戳
int update = namedParameterJdbcTemplate.update(sql, new JbaBeanProperty(new Object()));
JbaLog.infoSql(sql, null, start);
if (update > 0) {
res = true;
}
return res;
}
/**
* 执行sql语句
*
* @param sql sql带有参数
* @param params 参数对象 支持map
* @return 是否执行成功
*/
public boolean executeSQL(final String sql, Object params) {
boolean res = false;
if (null == params) {
params = new Object();
}
long start = System.currentTimeMillis();// 记录开始时间戳
int update;
if (params instanceof Map) {
update = namedParameterJdbcTemplate.update(sql, new JbaMapSqlSource((Map) params));
} else {
update = namedParameterJdbcTemplate.update(sql, new JbaBeanProperty(params));
}
JbaLog.infoSql(sql, params, start);
if (update > 0) {
res = true;
}
return res;
}
/**
* 查询数据库 - 单个列值
*
* @param sql 需要查询的sql
* @param params sql中参数值 支持map
* @param clazz 需要返回的类型
* @return 返回查询某个单列的值
*/
public T queryColumn(final String sql, Object params, Class clazz) {
T ret = null;
try {
long start = System.currentTimeMillis();// 记录开始时间戳
if (params != null) {
if (params instanceof Map) {
ret = namedParameterJdbcTemplate
.queryForObject(sql, new JbaMapSqlSource((Map) params), clazz);
} else {
ret = namedParameterJdbcTemplate.queryForObject(sql, new JbaBeanProperty(params), clazz);
}
} else {
ret = namedParameterJdbcTemplate.queryForObject(sql, new JbaBeanProperty(new Object()), clazz);
}
JbaLog.infoSql(sql, params, start);
} catch (Exception e) {
JbaLog.error(e);
ret = null;
}
return ret;
}
/**
* 查询数量,查询混合对象时用
*
* @param sql sql语句
* @param params 参数 支持map
* @return 数量
*/
public int queryCount(final String sql, Object params) {
int ret;
Integer count = queryColumn(sql, params, Integer.class);
if (null == count) {
ret = 0;
} else {
ret = count;
}
return ret;
}
/**
* 查询单表的数量
*
* @param model 表对象
* @return 数量
*/
public int queryCount(Object model) {
Objects.requireNonNull(model, "表对象model不允许为空");
int ret;
String sql = "SELECT COUNT(1) FROM " + TableManager.getTable(model.getClass()).getTableName();
String where = SQLUtils.getWhereSql(model);
if (where.length() > 0) {
sql = sql + " WHERE " + where;
}
Integer count = queryColumn(sql, model, Integer.class);
if (null == count) {
ret = 0;
} else {
ret = count;
}
return ret;
}
/**
* 查询数据库 - 单个对象
*
* @param model 对象对应的表名
* @param clazz 返回的类型
* @param orderBy 排序字段，model的属性名注意大小写
* @return 单个对象
*/
public T queryModelOne(Object model, Class clazz, final String... orderBy) {
List list = queryModelList(model, clazz, new PageInfo(1, 1, false), orderBy);
if (list == null || list.size() < 1) {
return null;
}
return list.get(0);
}
/**
* 查询数据库 - 单个混合对象
*
* @param sql sql语句
* @param params 对象模型-创建查询条件 支持map
* @param clazz 返回的类型
* @return 单个对象
*/
public T queryMixModelOne(final String sql, Object params, Class clazz) {
List list = queryMixModelList(sql, params, clazz, new PageInfo(1, 1, false));
if (list == null || list.size() < 1) {
return null;
}
return list.get(0);
}
/**
* 查询单表--列表
*
* @param model 对象对应的表名
* @param clazz 返回的类型
* @param pageInfo 分页信息
* @param orderBy 排序字段，model的属性名注意大小写
* @return 列表
*/
public List queryModelList(Object model, Class clazz, PageInfo pageInfo, final String... orderBy) {
Objects.requireNonNull(model, "表对象model不允许为空");
List ret;
// 获取字段列表
TableProperties table = TableManager.getTable(model.getClass());
String fields = table.getColumns().values().stream()
.map(ColumnProperties::colName).collect(Collectors.joining(","));
String sql = "SELECT " + fields + " FROM " + table.getTableName();
// 存在where子句
String where = SQLUtils.getWhereSql(model);
if (where.length() > 0) {
sql = sql + " WHERE " + where;
}
String by = SQLUtils.getOrderBy(table.getColumns(), orderBy);
if (by.length() > 0) {
sql = sql + " ORDER BY " + by;
}
sql = getNamedPageSql(sql, model, pageInfo);
if (null != pageInfo && pageInfo.getNeedTotalCount() && pageInfo.getTotalCount() < 1) {
return new ArrayList<>(0);
}
long start = System.currentTimeMillis();// 记录开始时间戳
ret = namedParameterJdbcTemplate.query(sql, new JbaBeanProperty(model), new JbaBeanPropertyRowMapper<>(clazz));
JbaLog.infoSql(sql, model, start);
return ret;
}
/**
* 查询数据库--混合对象列表
*
* @param sql sql语句
* @param params 参数对应的对象 OR map
* @param clazz 返回的类型
* @param pageInfo 分页信息
* @return 列表
*/
public List queryMixModelList(final String sql, Object params, Class clazz, PageInfo pageInfo) {
List ret;
String sql_final = getNamedPageSql(sql, params, pageInfo);
if (null != pageInfo && pageInfo.getNeedTotalCount() && pageInfo.getTotalCount() < 1) {
return new ArrayList<>(0);
}
long start = System.currentTimeMillis();// 记录开始时间戳
if (params != null) {
if (params instanceof Map) {
ret = namedParameterJdbcTemplate.query(sql_final, new JbaMapSqlSource((Map) params),
new JbaBeanPropertyRowMapper<>(clazz));
} else {
ret = namedParameterJdbcTemplate
.query(sql_final, new JbaBeanProperty(params), new JbaBeanPropertyRowMapper<>(clazz));
}
} else {
ret = namedParameterJdbcTemplate.query(sql_final, new JbaBeanPropertyRowMapper<>(clazz));
}
JbaLog.infoSql(sql_final, params, start);
return ret;
}
/**
* 获取分页sql
*
* @param sql 原始sql
* @param params 参数对象
* @param pageInfo 分页信息
* @param 泛型
* @return 返回带分页的sql
*/
private String getNamedPageSql(final String sql, T params, PageInfo pageInfo) {
if (null != pageInfo) {
if (pageInfo.getNeedTotalCount()) {
// 先查询总数
String temp = "SELECT count(1) FROM(" + sql + ") t_temp_t";
Integer totalCount = queryCount(temp, params);
if (totalCount < 1) { // 总数=0
pageInfo.setTotalCount(totalCount);
return sql;
}
// 计算总页数
pageInfo.setTotalCount(totalCount);
}
return SQLUtils.getPageSql(sql, pageInfo, getDatabaseName());
}
return sql;
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy