mybatis統計每條SQL的執行時間的方法示例
阿新 • • 發佈：2020-01-10
背景
最近面試經常被問到關於資料庫的事務的問題，可能平時我就知道加個註解@Transactional之後就一臉懵逼的。現在發現這一塊真的是常常被忽略了，然而面試官就是最喜歡這種看是不常用，但是非常重要的問題，進而達到出其不意攻其不備。不吹水了，開始正文。
方案一：切面程式設計@Aspect
此方案主要是通過環繞切面的方式將mapper包下的介面方法，然後前後計算時間差即可。這就是典型的AOP知識，不過這種計算比較粗糙，但是也是個辦法。具體方法如下：
@Aspect
@Component
@Slf4j
public class MapperAspect {
@AfterReturning("execution(* cn.xbmchina.mybatissqltime.mapper.*Mapper.*(..))")
public void logServiceAccess(JoinPoint joinPoint) {
log.info("Completed: " + joinPoint);
}
/**
* 監控cn.xbmchina.mybatissqltime.mapper..*Mapper包及其子包的所有public方法
*/
@Pointcut("execution(* cn.xbmchina.mybatissqltime.mapper.*Mapper.*(..))")
private void pointCutMethod() {
}
/**
* 宣告環繞通知
*
* @param pjp
* @return
* @throws Throwable
*/
@Around("pointCutMethod()")
public Object doAround(ProceedingJoinPoint pjp) throws Throwable {
long begin = System.nanoTime();
Object obj = pjp.proceed();
long end = System.nanoTime();
log.info("呼叫Mapper方法：{}，引數：{}，執行耗時：{}納秒，耗時：{}毫秒",pjp.getSignature().toString(),Arrays.toString(pjp.getArgs()),(end - begin),(end - begin) / 1000000);
return obj;
}
}
方案二：mybatis 的外掛
MyBatis在四大物件的建立過程中，都會有外掛進行介入。外掛可以利用動態代理機制一層層的包裝目標物件，而實現在目標物件執行目標方法之前進行攔截的效果。
MyBatis 允許在已對映語句執行過程中的某一點進行攔截呼叫。
預設情況下，MyBatis 允許使用外掛來攔截的方法呼叫包括：
①Executor(update,query,flushStatements,commit,rollback,getTransaction,close,isClosed)
②ParameterHandler(getParameterObject,setParameters)
④StatementHandler(prepare,parameterize,batch,update,query)
下面是程式碼:
import org.apache.ibatis.executor.statement.StatementHandler;
import org.apache.ibatis.mapping.BoundSql;
import org.apache.ibatis.mapping.ParameterMapping;
import org.apache.ibatis.plugin.Interceptor;
import org.apache.ibatis.plugin.Intercepts;
import org.apache.ibatis.plugin.Invocation;
import org.apache.ibatis.plugin.Plugin;
import org.apache.ibatis.plugin.Signature;
import org.apache.ibatis.session.ResultHandler;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;
import java.sql.Statement;
import java.util.List;
import java.util.Properties;
/**
* Sql執行時間記錄攔截器
*
* @author zero
* 2019年12月13日17:05:28
*/
@Intercepts({@Signature(type = StatementHandler.class,method = "query",args = {Statement.class,ResultHandler.class}),@Signature(type = StatementHandler.class,method = "update",args = {Statement.class}),method = "batch",args = {Statement.class})})
@Component
public class SqlExecuteTimeCountInterceptor implements Interceptor {
private static Logger logger = LoggerFactory.getLogger(SqlExecuteTimeCountInterceptor.class);
/**
* 列印的引數字串的最大長度
*/
private final static int MAX_PARAM_LENGTH = 50;
/**
* 記錄的最大SQL長度
*/
private final static int MAX_SQL_LENGTH = 200;
@Override
public Object intercept(Invocation invocation) throws Throwable {
Object target = invocation.getTarget();
long startTime = System.currentTimeMillis();
StatementHandler statementHandler = (StatementHandler) target;
try {
return invocation.proceed();
} finally {
long endTime = System.currentTimeMillis();
long timeCount = endTime - startTime;
BoundSql boundSql = statementHandler.getBoundSql();
String sql = boundSql.getSql();
Object parameterObject = boundSql.getParameterObject();
List<ParameterMapping> parameterMappingList = boundSql.getParameterMappings();
// 格式化Sql語句，去除換行符，替換引數
sql = formatSQL(sql,parameterObject,parameterMappingList);
logger.info("執行 SQL：[,{} ]執行耗時[ {} ms]",sql,timeCount);
}
}
/**
* 格式化/美化 SQL語句
*
* @param sql sql 語句
* @param parameterObject 引數的Map
* @param parameterMappingList 引數的List
* @return 格式化之後的SQL
*/
private String formatSQL(String sql,Object parameterObject,List<ParameterMapping> parameterMappingList) {
// 輸入sql字串空判斷
if (sql == null || sql.length() == 0) {
return "";
}
// 美化sql
sql = beautifySql(sql);
// 不傳引數的場景，直接把sql美化一下返回出去
if (parameterObject == null || parameterMappingList == null || parameterMappingList.size() == 0) {
return sql;
}
return LimitSQLLength(sql);
}
/**
* 返回限制長度之後的SQL語句
*
*
* @param sql 原始SQL語句
*/
private String LimitSQLLength(String sql) {
if (sql == null || sql.length() == 0) {
return "";
}
if (sql.length() > MAX_SQL_LENGTH) {
return sql.substring(0,MAX_SQL_LENGTH);
} else {
return sql;
}
}
@Override
public Object plugin(Object target) {
return Plugin.wrap(target,this);
}
@Override
public void setProperties(Properties properties) {
}
/**
* 替換SQL 中? 所對應的值,只保留前50個字元
*
* @param sql sql語句
* @param valueOf ?對應的值
*/
private String replaceValue(String sql,String valueOf) {
//超過50個字元只取前50個
if (valueOf != null && valueOf.length() > MAX_PARAM_LENGTH) {
valueOf = valueOf.substring(0,MAX_PARAM_LENGTH);
}
sql = sql.replaceFirst("\\?",valueOf);
return sql;
}
/**
* 美化sql
*
* @param sql sql語句
*/
private String beautifySql(String sql) {
sql = sql.replaceAll("[\\s\n ]+"," ");
return sql;
}
}方案三：直接用druid
這種就是我們平時用的最多的，但是面試的話說一下就得了，估計也沒有怎麼好問的了。
Springboot+druid的配置application.yml檔案如下：
spring:
datasource:
url: jdbc:mysql://localhost:3306/testdb1?characterEncoding=utf-8&useUnicode=true&useSSL=false&serverTimezone=UTC
driver-class-name: com.mysql.jdbc.Driver # mysql8.0以前使用com.mysql.jdbc.Driver
username: root
password: root
platform: mysql
#通過這句配置將druid連線池引入到我們的配置中，spring會盡可能判斷型別是什麼，然後根據情況去匹配驅動類。
type: com.alibaba.druid.pool.DruidDataSource
druid:
initial-size: 5 # 初始化大小
min-idle: 5 # 最小
max-active: 100 # 最大
max-wait: 60000 # 配置獲取連線等待超時的時間
time-between-eviction-runs-millis: 60000 # 配置間隔多久才進行一次檢測，檢測需要關閉的空閒連線，單位是毫秒
min-evictable-idle-time-millis: 300000 # 指定一個空閒連線最少空閒多久後可被清除，單位是毫秒
validationQuery: select 'x'
test-while-idle: true # 當連線空閒時，是否執行連線測試
test-on-borrow: false # 當從連線池借用連線時，是否測試該連線
test-on-return: false # 在連線歸還到連線池時是否測試該連線
filters: config,wall,stat # 配置監控統計攔截的filters，去掉後監控介面sql無法統計，'wall'用於防火牆
poolPreparedStatements: true # 開啟PSCache，並且指定每個連線上PSCache的大小
maxPoolPreparedStatementPerConnectionSize: 20
maxOpenPreparedStatements: 20
# 通過connectProperties屬性來開啟mergeSql功能；慢SQL記錄
connectionProperties: druid.stat.slowSqlMillis=200;druid.stat.logSlowSql=true;config.decrypt=false
# 合併多個DruidDataSource的監控資料
#use-global-data-source-stat: true
#WebStatFilter配置，說明請參考Druid Wiki，配置_配置WebStatFilter
web-stat-filter:
enabled: true #是否啟用StatFilter預設值true
url-pattern: /*
exclusions: /druid/*,*.js,*.gif,*.jpg,*.bmp,*.png,*.css,*.ico
session-stat-enable: true
session-stat-max-count: 10
#StatViewServlet配置，說明請參考Druid Wiki，配置_StatViewServlet配置
stat-view-servlet:
enabled: true #是否啟用StatViewServlet預設值true
url-pattern: /druid/*
reset-enable: true
login-username: admin
login-password: admin
總結
以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支援我們。