快捷導(dǎo)航

mybatis統(tǒng)計(jì)每條SQL的執(zhí)行時(shí)間的方法示例

更新時(shí)間：2020年01月02日 13:04:42 作者：zero

這篇文章主要介紹了mybatis統(tǒng)計(jì)每條SQL的執(zhí)行時(shí)間的方法示例，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

背景

最近面試經(jīng)常被問到關(guān)于數(shù)據(jù)庫的事務(wù)的問題，可能平時(shí)我就知道加個(gè)注解@Transactional之后就一臉懵逼的?，F(xiàn)在發(fā)現(xiàn)這一塊真的是常常被忽略了，然而面試官就是最喜歡這種看是不常用，但是非常重要的問題，進(jìn)而達(dá)到出其不意攻其不備。不吹水了，開始正文。

方案一：切面編程@Aspect

此方案主要是通過環(huán)繞切面的方式將mapper包下的接口方法，然后前后計(jì)算時(shí)間差即可。這就是典型的AOP知識(shí)，不過這種計(jì)算比較粗糙，但是也是個(gè)辦法。具體方法如下：

@Aspect
@Component
@Slf4j
public class MapperAspect {

  @AfterReturning("execution(* cn.xbmchina.mybatissqltime.mapper.*Mapper.*(..))")
  public void logServiceAccess(JoinPoint joinPoint) {
    log.info("Completed: " + joinPoint);
  }


  /**
   * 監(jiān)控cn.xbmchina.mybatissqltime.mapper..*Mapper包及其子包的所有public方法
   */
  @Pointcut("execution(* cn.xbmchina.mybatissqltime.mapper.*Mapper.*(..))")
  private void pointCutMethod() {
  }

  /**
   * 聲明環(huán)繞通知
   *
   * @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("調(diào)用Mapper方法：{}，參數(shù)：{}，執(zhí)行耗時(shí)：{}納秒，耗時(shí)：{}毫秒",
        pjp.getSignature().toString(), Arrays.toString(pjp.getArgs()),
        (end - begin), (end - begin) / 1000000);
    return obj;
  }
}

方案二：mybatis 的插件

MyBatis在四大對(duì)象的創(chuàng)建過程中，都會(huì)有插件進(jìn)行介入。插件可以利用動(dòng)態(tài)代理機(jī)制一層層的包裝目標(biāo)對(duì)象，而實(shí)現(xiàn)在目標(biāo)對(duì)象執(zhí)行目標(biāo)方法之前進(jìn)行攔截的效果。

MyBatis 允許在已映射語句執(zhí)行過程中的某一點(diǎn)進(jìn)行攔截調(diào)用。

默認(rèn)情況下，MyBatis 允許使用插件來攔截的方法調(diào)用包括：

①Executor(update, query, flushStatements, commit, rollback, getTransaction, close, isClosed)
②ParameterHandler(getParameterObject, setParameters)
③ResultSetHandler(handleResultSets, handleOutputParameters)
④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執(zhí)行時(shí)間記錄攔截器
 *
 * @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}),
    @Signature(type = StatementHandler.class, method = "batch", args = {Statement.class})})
@Component
public class SqlExecuteTimeCountInterceptor implements Interceptor {

  private static Logger logger = LoggerFactory.getLogger(SqlExecuteTimeCountInterceptor.class);

  /**
   * 打印的參數(shù)字符串的最大長(zhǎng)度
   */
  private final static int MAX_PARAM_LENGTH = 50;

  /**
   * 記錄的最大SQL長(zhǎng)度
   */
  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語句，去除換行符，替換參數(shù)
      sql = formatSQL(sql, parameterObject, parameterMappingList);

      logger.info("執(zhí)行 SQL：[ , {} ]執(zhí)行耗時(shí)[ {} ms]", sql, timeCount);
    }
  }


  /**
   * 格式化/美化 SQL語句
   *
   * @param sql         sql 語句
   * @param parameterObject   參數(shù)的Map
   * @param parameterMappingList 參數(shù)的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);
    // 不傳參數(shù)的場(chǎng)景，直接把sql美化一下返回出去
    if (parameterObject == null || parameterMappingList == null || parameterMappingList.size() == 0) {
      return sql;
    }
    return LimitSQLLength(sql);
  }


  /**
   * 返回限制長(zhǎng)度之后的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 中? 所對(duì)應(yīng)的值, 只保留前50個(gè)字符
   *
   * @param sql   sql語句
   * @param valueOf ?對(duì)應(yīng)的值
   */
  private String replaceValue(String sql, String valueOf) {
    //超過50個(gè)字符只取前50個(gè)
    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

這種就是我們平時(shí)用的最多的，但是面試的話說一下就得了，估計(jì)也沒有怎么好問的了。

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會(huì)盡可能判斷類型是什么，然后根據(jù)情況去匹配驅(qū)動(dòng)類。
    type: com.alibaba.druid.pool.DruidDataSource
    druid:
      initial-size: 5 # 初始化大小
      min-idle: 5 # 最小
      max-active: 100 # 最大
      max-wait: 60000 # 配置獲取連接等待超時(shí)的時(shí)間
      time-between-eviction-runs-millis: 60000 # 配置間隔多久才進(jìn)行一次檢測(cè)，檢測(cè)需要關(guān)閉的空閑連接，單位是毫秒
      min-evictable-idle-time-millis: 300000 # 指定一個(gè)空閑連接最少空閑多久后可被清除，單位是毫秒
      validationQuery: select 'x'
      test-while-idle: true # 當(dāng)連接空閑時(shí)，是否執(zhí)行連接測(cè)試
      test-on-borrow: false # 當(dāng)從連接池借用連接時(shí)，是否測(cè)試該連接
      test-on-return: false # 在連接歸還到連接池時(shí)是否測(cè)試該連接
      filters: config,wall,stat # 配置監(jiān)控統(tǒng)計(jì)攔截的filters，去掉后監(jiān)控界面sql無法統(tǒng)計(jì)，'wall'用于防火墻
      poolPreparedStatements: true # 打開PSCache，并且指定每個(gè)連接上PSCache的大小
      maxPoolPreparedStatementPerConnectionSize: 20
      maxOpenPreparedStatements: 20
      # 通過connectProperties屬性來打開mergeSql功能；慢SQL記錄
      connectionProperties: druid.stat.slowSqlMillis=200;druid.stat.logSlowSql=true;config.decrypt=false
       # 合并多個(gè)DruidDataSource的監(jiān)控?cái)?shù)據(jù)
      #use-global-data-source-stat: true
      #WebStatFilter配置，說明請(qǐng)參考Druid Wiki，配置_配置WebStatFilter
      web-stat-filter:
        enabled: true #是否啟用StatFilter默認(rèn)值true
        url-pattern: /*
        exclusions: /druid/*,*.js,*.gif,*.jpg,*.bmp,*.png,*.css,*.ico
        session-stat-enable: true
        session-stat-max-count: 10
      #StatViewServlet配置，說明請(qǐng)參考Druid Wiki，配置_StatViewServlet配置
      stat-view-servlet:
        enabled: true #是否啟用StatViewServlet默認(rèn)值true
        url-pattern: /druid/*
        reset-enable: true
        login-username: admin
        login-password: admin

總結(jié)

以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: