springboot中对mybatis的sql执行进行耗时统计

实现上有两个简单的方法

1、对Mapper的接口做拦截器，通过拦截器统计耗时

2、对mybatis执行sql统计

理论上，业务开发过程中，要关注的是对mapper的接口方法耗时统计，mybatis执行sql的耗时统计应该是没有管sql编译过程。

第一种实现方式，纯方法级别拦截，最好理解

1、引入aop包，使用aspect切面编程

    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-aop</artifactId>
    </dependency>

2、定义一个注解，到时候对需要的方法进行注解

import java.lang.annotation.*;

@Target(value={ElementType.PARAMETER, ElementType.METHOD})
@Retention(value= RetentionPolicy.RUNTIME)
@Documented
public @interface Cost {
    String description()  default "";
}

3、编写aspect切面代码

import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;


@Aspect
@Component
public class MethodHandlerInterceptor {
    Logger logger = LoggerFactory.getLogger(MethodHandlerInterceptor.class);
    @Around("@annotation(com.xucf.aop.Cost)")
    public Object interceptor(ProceedingJoinPoint pjp) throws Throwable {
        long startTime = System.currentTimeMillis();
        Object object = pjp.proceed();
        long endTime = System.currentTimeMillis();
        logger.info( pjp.getSignature().getDeclaringType().getName() + "." + pjp.getSignature().getName() + " time consume " +(endTime - startTime) + " ms");
        return object;
    }

}

4、然后在mapper方法上加上这个注解

@Repository
public interface TestDao {
    @Cost
    List<Map> selectAll ();
}

加个contral测试可以得到效果，每次调用mapper的方法，都可以做统计。

 第二种，借用mybatis拦截器，对所有sql执行进行拦截

写个拦截器就好了

import org.apache.ibatis.executor.statement.StatementHandler;
import org.apache.ibatis.plugin.*;
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.Properties;

@Component
@Intercepts({
        @Signature(
                type = StatementHandler.class,
                method = "query",
                args = {Statement.class, ResultHandler.class})
})
public class MyAop implements Interceptor {
    public static final Logger log = LoggerFactory.getLogger(MyAop.class);
    @Override
    public Object intercept(Invocation invocation) throws Throwable {
        Long b = System.currentTimeMillis();
        String sql = ((StatementHandler)invocation.getTarget()).getBoundSql().getSql();
        Object obj = invocation.proceed();
        log.info(sql + " cost " + (System.currentTimeMillis() - b) + "ms");
        return obj;
    }
    @Override
    public Object plugin(Object target) {
        return Plugin.wrap(target, this);
    }
    @Override
    public void setProperties(Properties properties) {
    }
}

启动后实现效果

 可以看到执行的sql。也可以对执行超时的、异常的语句进行补偿处理等。