Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.pugwoo.dbhelper.impl.part.P0_JdbcTemplateOp Maven / Gradle / Ivy
package com.pugwoo.dbhelper.impl.part;
import com.pugwoo.dbhelper.DBHelper;
import com.pugwoo.dbhelper.DBHelperInterceptor;
import com.pugwoo.dbhelper.IDBHelperSlowSqlCallback;
import com.pugwoo.dbhelper.enums.FeatureEnum;
import com.pugwoo.dbhelper.impl.DBHelperContext;
import com.pugwoo.dbhelper.impl.SpringJdbcDBHelper;
import com.pugwoo.dbhelper.utils.InnerCommonUtils;
import com.pugwoo.dbhelper.utils.NamedParameterUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate;
import org.springframework.transaction.interceptor.TransactionAspectSupport;
import org.springframework.transaction.support.TransactionSynchronization;
import org.springframework.transaction.support.TransactionSynchronizationManager;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
/**
* jdbcTemplate原生操作接口封装
* @author NICK
*/
public abstract class P0_JdbcTemplateOp implements DBHelper, ApplicationContextAware {
protected static final Logger LOGGER = LoggerFactory.getLogger(SpringJdbcDBHelper.class);
protected JdbcTemplate jdbcTemplate;
protected NamedParameterJdbcTemplate namedParameterJdbcTemplate;
protected long timeoutWarningValve = 1000;
protected Integer maxPageSize = null; // 每页最大个数，为null表示不限制
protected ApplicationContext applicationContext;
protected List interceptors = new ArrayList<>();
protected Map features = new ConcurrentHashMap() {{
put(FeatureEnum.AUTO_SUM_NULL_TO_ZERO, true);
put(FeatureEnum.LOG_SQL_AT_INFO_LEVEL, false);
put(FeatureEnum.THROW_EXCEPTION_IF_COLUMN_NOT_EXIST, false);
put(FeatureEnum.AUTO_ADD_ORDER_FOR_PAGINATION, true);
}};
private IDBHelperSlowSqlCallback slowSqlCallback;
/**全局的SQL注释*/
private String globalComment;
protected void log(String sql, Object keyValues) {
if (features.get(FeatureEnum.LOG_SQL_AT_INFO_LEVEL)) {
LOGGER.info("ExecSQL:{},params:{}", sql, keyValues);
} else {
LOGGER.debug("ExecSQL:{},params:{}", sql, keyValues);
}
}
protected void logSlow(long cost, String sql, List keyValues) {
if(cost > timeoutWarningValve) {
LOGGER.warn("SlowSQL:{},cost:{}ms,params:{}", sql, cost, keyValues);
try {
if(slowSqlCallback != null) {
slowSqlCallback.callback(cost, sql, keyValues);
}
} catch (Throwable e) {
LOGGER.error("DBHelperSlowSqlCallback fail, SlowSQL:{},cost:{}ms,params:{}",
sql, cost, keyValues, e);
}
}
}
protected void logSlowForParamMap(long cost, String sql, Map paramMap) {
List params = new ArrayList<>();
params.add(paramMap);
logSlow(cost, sql, params);
}
protected void logSlowForBatch(long cost, String sql, int listSize) {
if(cost > timeoutWarningValve) {
LOGGER.warn("SlowSQL:{},cost:{}ms,listSize:{}", sql, cost, listSize);
try {
if(slowSqlCallback != null) {
slowSqlCallback.callback(cost, sql, new ArrayList<>());
}
} catch (Throwable e) {
LOGGER.error("DBHelperSlowSqlCallback fail, SlowSQL:{},cost:{}ms,listSize:{}",
sql, cost, listSize, e);
}
}
}
@Override
public void rollback() {
TransactionAspectSupport.currentTransactionStatus().setRollbackOnly();
}
@Override
public boolean executeAfterCommit(final Runnable runnable) {
if(runnable == null) {
return false;
}
if(!TransactionSynchronizationManager.isActualTransactionActive()) {
return false;
}
TransactionSynchronizationManager.registerSynchronization(new TransactionSynchronization() {
@Override
public void suspend() {
}
@Override
public void resume() {
}
@Override
public void flush() {
}
@Override
public void beforeCompletion() {
}
@Override
public void beforeCommit(boolean readOnly) {
}
@Override
public void afterCompletion(int status) {
}
@Override
public void afterCommit() {
runnable.run();
}
});
return true;
}
/**
* 使用jdbcTemplate模版执行update，不支持in (?)表达式
* @return 实际修改的行数
*/
protected int jdbcExecuteUpdate(String sql, Object... args) {
sql = addComment(sql);
log(sql, args);
long start = System.currentTimeMillis();
int rows = jdbcTemplate.update(sql, args);// 此处可以用jdbcTemplate，因为没有in (?)表达式
long cost = System.currentTimeMillis() - start;
logSlow(cost, sql, args == null ? new ArrayList<>() : Arrays.asList(args));
return rows;
}
/**
* 使用namedParameterJdbcTemplate模版执行update，支持in(?)表达式
*/
protected int namedJdbcExecuteUpdate(String sql, Object... args) {
sql = addComment(sql);
log(sql, args);
long start = System.currentTimeMillis();
List argsList = new ArrayList<>(); // 不要直接用Arrays.asList，它不支持clear方法
if(args != null) {
argsList.addAll(Arrays.asList(args));
}
int rows = namedParameterJdbcTemplate.update(
NamedParameterUtils.trans(sql, argsList),
NamedParameterUtils.transParam(argsList)); // 因为有in (?) 所以使用namedParameterJdbcTemplate
long cost = System.currentTimeMillis() - start;
logSlow(cost, sql, argsList);
return rows;
}
public void setJdbcTemplate(JdbcTemplate jdbcTemplate) {
this.jdbcTemplate = jdbcTemplate;
this.namedParameterJdbcTemplate = new NamedParameterJdbcTemplate(jdbcTemplate);
}
/**
* 废弃，不需要设置了 since 1.2
*/
@Deprecated
public void setNamedParameterJdbcTemplate(NamedParameterJdbcTemplate namedParameterJdbcTemplate) {
this.namedParameterJdbcTemplate = namedParameterJdbcTemplate;
}
public JdbcTemplate getJdbcTemplate() {
return this.jdbcTemplate;
}
public NamedParameterJdbcTemplate getNamedParameterJdbcTemplate() {
return this.namedParameterJdbcTemplate;
}
@Override
public void setTimeoutWarningValve(long timeMS) {
timeoutWarningValve = timeMS;
}
@Override
public void setMaxPageSize(int maxPageSize) {
this.maxPageSize = maxPageSize;
}
@Override
public void setApplicationContext(ApplicationContext applicationContext)
throws BeansException {
this.applicationContext = applicationContext;
}
@Override
public void setInterceptors(List interceptors) {
if(interceptors != null) {
// 移除为null的值
for(DBHelperInterceptor interceptor : interceptors) {
if(interceptor != null) {
this.interceptors.add(interceptor);
}
}
}
}
@Override
public void setTimeoutWarningCallback(IDBHelperSlowSqlCallback callback) {
this.slowSqlCallback = callback;
}
@Override
public void setTableName(Class clazz, String tableName) {
DBHelperContext.setTableName(clazz, tableName);
}
@Override
public void resetTableNames() {
DBHelperContext.resetTableName();
}
@Override
public void turnOffSoftDelete(Class... clazz) {
if (clazz == null) {
return;
}
for (Class c : clazz) {
DBHelperContext.turnOffSoftDelete(c);
}
}
@Override
public void turnOnSoftDelete(Class... clazz) {
if (clazz == null) {
return;
}
for (Class c : clazz) {
DBHelperContext.turnOnSoftDelete(c);
}
}
@Override
public void turnOnFeature(FeatureEnum featureEnum) {
features.put(featureEnum, true);
}
@Override
public void turnOffFeature(FeatureEnum featureEnum) {
features.put(featureEnum, false);
}
public boolean getFeature(FeatureEnum featureEnum) {
Boolean enabled = features.get(featureEnum);
return enabled != null && enabled;
}
@Override
public void setGlobalComment(String comment) {
this.globalComment = comment;
}
@Override
public void setLocalComment(String comment) {
DBHelperContext.setComment(comment);
}
/**
* 给sql加上注释，返回加完注释之后的sql
*/
protected String addComment(String sql) {
if (sql == null) {
sql = "";
}
if (InnerCommonUtils.isNotBlank(globalComment)) {
sql = "/*" + globalComment + "*/" + sql;
}
String comment = DBHelperContext.getComment();
if (InnerCommonUtils.isNotBlank(comment)) {
sql = "/*" + comment + "*/" + sql;
}
return sql;
}
}
