java通过MySQL驱动拦截器实现执行sql耗时计算
更新时间：2023年03月20日 14:25:14 作者：Redick01
本文主要介绍了java通过MySQL驱动拦截器实现执行sql耗时计算，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
背景
公司的一个需求，公司既有的链路追踪日志组件要支持MySQL的sql执行时间打印，要实现链路追踪常用的手段就是实现第三方框架或工具提供的拦截器接口或者是过滤器接口，对于MySQL也不例外，实际上就是实现了MySQL驱动的拦截器接口而已。
具体实现
MySQL的渠道有不同的版本，不同版本的拦截器接口是不同的，所以要针对你所使用的不同版本的MySQL驱动去实现响应的拦截器，接下来分别介绍下MySQL渠道5,6,8版本的实现方式。
MySQL5
这里以MySQL渠道5.1.18版本为例实现，实现StatementInterceptorV2接口，主要实现逻辑在preProcess和postProcess方法，这两个方法是sql执行前后要执行的方法，我所使用的框架是logback，这里使用MDC来记录sql执行前的一个时间戳，代码在postProcess方法MDC.put("sql_exec_time", start);，自己也可以使用ThreadLocal等来实现，然后在postProcess方法中使用MDC.get("sql_exec_time")将记录的sql执行前的时间取出来，最后再用当前时间戳减去sql执行前的时间，就算出了sql执行的时间。
import static net.logstash.logback.marker.Markers.append;
import com.mysql.jdbc.Connection;
import com.mysql.jdbc.ResultSetInternalMethods;
import com.mysql.jdbc.Statement;
import com.mysql.jdbc.StatementInterceptorV2;
import com.redick.util.LogUtil;
import java.sql.SQLException;
import java.util.Properties;
import lombok.extern.slf4j.Slf4j;
import org.slf4j.MDC;
/**
* @author Redick01
*/
@Slf4j
public class Mysql5StatementInterceptor implements StatementInterceptorV2 {
@Override
public void init(Connection connection, Properties properties) throws SQLException {
}
@Override
public ResultSetInternalMethods preProcess(String s, Statement statement, Connection connection)
throws SQLException {
String start = String.valueOf(System.currentTimeMillis());
MDC.put("sql_exec_time", start);
log.info(LogUtil.customizeMarker(LogUtil.kLOG_KEY_TRACE_TAG, "sql_exec_before"), "开始执行sql");
return null;
}
@Override
public boolean executeTopLevelOnly() {
return false;
}
@Override
public void destroy() {
}
@Override
public ResultSetInternalMethods postProcess(String s, Statement statement,
ResultSetInternalMethods resultSetInternalMethods, Connection connection, int i,
boolean b, boolean b1, SQLException e) throws SQLException {
long start = Long.parseLong(MDC.get("sql_exec_time"));
long end = System.currentTimeMillis();
log.info(LogUtil.customizeMarker(LogUtil.kLOG_KEY_TRACE_TAG, "sql_exec_after")
.and(append(LogUtil.kLOG_KEY_SQL_EXEC_DURATION, end - start)), "结束执行sql");
return null;
}
}
MySQL6
MySQL6和MySQL5基本一样，只是接口不是同一个，直接放代码
import static net.logstash.logback.marker.Markers.append;
import com.mysql.cj.api.MysqlConnection;
import com.mysql.cj.api.jdbc.Statement;
import com.mysql.cj.api.jdbc.interceptors.StatementInterceptor;
import com.mysql.cj.api.log.Log;
import com.mysql.cj.api.mysqla.result.Resultset;
import com.redick.util.LogUtil;
import java.sql.SQLException;
import java.util.Properties;
import lombok.extern.slf4j.Slf4j;
import org.slf4j.MDC;
/**
* @author Redick01
*/
@Slf4j
public class Mysql6StatementInterceptor implements StatementInterceptor {
@Override
public StatementInterceptor init(MysqlConnection mysqlConnection, Properties properties,
Log log) {
return null;
}
@Override
public <T extends Resultset> T preProcess(String s, Statement statement) throws SQLException {
String start = String.valueOf(System.currentTimeMillis());
MDC.put("sql_exec_time", start);
log.info(LogUtil.customizeMarker(LogUtil.kLOG_KEY_TRACE_TAG, "sql_exec_before"), "开始执行sql");
return null;
}
@Override
public boolean executeTopLevelOnly() {
return false;
}
@Override
public void destroy() {
}
@Override
public <T extends Resultset> T postProcess(String s, Statement statement, T t, int i, boolean b,
boolean b1, Exception e) throws SQLException {
long start = Long.parseLong(MDC.get("sql_exec_time"));
long end = System.currentTimeMillis();
log.info(LogUtil.customizeMarker(LogUtil.kLOG_KEY_TRACE_TAG, "sql_exec_after")
.and(append(LogUtil.kLOG_KEY_SQL_EXEC_DURATION, end - start)), "结束执行sql");
return null;
}
}
MySQL8
MySQL8和MySQL5/6的拦截器接口又不一样了，MySQL8的拦截器接口是com.mysql.cj.interceptors.QueryInterceptor，统计sql执行时间的方式还是一样的，代码如下：
import static net.logstash.logback.marker.Markers.append;
import com.mysql.cj.MysqlConnection;
import com.mysql.cj.Query;
import com.mysql.cj.interceptors.QueryInterceptor;
import com.mysql.cj.log.Log;
import com.mysql.cj.protocol.Resultset;
import com.mysql.cj.protocol.ServerSession;
import com.redick.util.LogUtil;
import java.util.Properties;
import java.util.function.Supplier;
import lombok.extern.slf4j.Slf4j;
import org.slf4j.MDC;
/**
* @author Redick01
*/
@Slf4j
public class Mysql8QueryInterceptor implements QueryInterceptor {
@Override
public QueryInterceptor init(MysqlConnection mysqlConnection, Properties properties, Log log) {
return null;
}
@Override
public <T extends Resultset> T preProcess(Supplier<String> supplier, Query query) {
String start = String.valueOf(System.currentTimeMillis());
MDC.put("sql_exec_time", start);
log.info(LogUtil.customizeMarker(LogUtil.kLOG_KEY_TRACE_TAG, "sql_exec_before"), "开始执行sql");
return null;
}
@Override
public boolean executeTopLevelOnly() {
return false;
}
@Override
public void destroy() {
}
@Override
public <T extends Resultset> T postProcess(Supplier<String> supplier, Query query, T t,
ServerSession serverSession) {
long start = Long.parseLong(MDC.get("sql_exec_time"));
long end = System.currentTimeMillis();
log.info(LogUtil.customizeMarker(LogUtil.kLOG_KEY_TRACE_TAG, "sql_exec_after")
.and(append(LogUtil.kLOG_KEY_SQL_EXEC_DURATION, end - start)), "结束执行sql");
return null;
}
}
使用方法
MySQL5和6的使用方式一样，在数据库链接的url中增加如下statementInterceptors参数，例如：
url: jdbc:mysql://127.0.0.1:3316/log-helper?useUnicode=true&characterEncoding=UTF8&statementInterceptors=com.redick.support.mysql.Mysql5StatementInterceptor&serverTimezone=CST
MySQL8则是在url中增加queryInterceptors参数，例如：
url: jdbc:mysql://127.0.0.1:3316/log-helper?useUnicode=true&characterEncoding=UTF8&queryInterceptors=com.redick.support.mysql.Mysql8QueryInterceptor&serverTimezone=CST
测试结果
sql执行前日志
{"@timestamp":"2023-02-28T17:16:29.234+08:00","@version":"0.0.1","message":"开始执行sql","logger_name":"com.redick.support.mysql.Mysql5StatementInterceptor","thread_name":"http-nio-3321-exec-4","level":"INFO","level_value":20000,"traceId":"9ed930dc-4cc6-4719-bf33-9fcb618fd65b","spanId":"1","request_type":"getName","parentId":"0","trace_tag":"sql_exec_before"}
sql执行后日志，sql_duration标识执行sql耗时3ms
{"@timestamp":"2023-02-28T17:16:29.237+08:00","@version":"0.0.1","message":"结束执行sql","logger_name":"com.redick.support.mysql.Mysql5StatementInterceptor","thread_name":"http-nio-3321-exec-4","level":"INFO","level_value":20000,"traceId":"9ed930dc-4cc6-4719-bf33-9fcb618fd65b","spanId":"1","request_type":"getName","parentId":"0","trace_tag":"sql_exec_after","sql_duration":3}
具体实现代码参考基于Spring AOP + logstash-logback-encoder日志链路追踪工具LogHelper
到此这篇关于java通过MySQL驱动拦截器实现执行sql耗时计算的文章就介绍到这了,更多相关java执行sql耗时计算内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要介绍了Java 多个异常共享同一个异常处理器的方法,Java 的异常处理机制，在 Java 7 中有了非常大的改进。其中一个特性就是，支持多个异常共享同一个异常处理器。,需要的朋友可以参考下2019-06-06
在高并发系统中，出于系统保护角度考虑，通常会对流量进行限流；不但在工作中要频繁使用，而且也是面试中的高频考点。本文就为大家整理了5种Java中常见限流算法，需要的可以参考一下2023-04-04
这篇文章主要介绍了使用springboot对linux进行操控的方法示例，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-11-11
这篇文章主要介绍了JAVA Spring 中的事件机制的相关资料，文中示例代码非常细致，帮助大家更好的理解和学习，感兴趣的朋友可以了解下2020-07-07
这篇文章主要介绍了Springboot打包部署项目的过程,通过jar包方式打包和war方式打包,本文结合示例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友参考下吧2023-12-12
这篇文章主要介绍了java中struts2实现文件上传下载功能的方法,以实例形式分析了struts2文件上传下载功能的实现技巧与相关问题,具有一定的参考借鉴价值,需要的朋友可以参考下2016-05-05
这篇文章主要为大家详细介绍了Java编写实现登陆窗口，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2022-04-04
Spring Boot Actuator提供了生产上经常用到的功能（如健康检查，审计，指标收集，HTTP跟踪等），帮助我们监控和管理Spring Boot应用程序。这些功能都可以通过JMX或HTTP端点访问2023-01-01
Java AbstractMethodError原因案例详解
这篇文章主要介绍了Java AbstractMethodError原因案例详解,本篇文章通过简要的案例,讲解了该项技术的了解与使用,以下就是详细内容,需要的朋友可以参考下2021-08-08
这篇文章主要介绍了java发送heartbeat心跳包(byte转16进制),需要的朋友可以参考下2014-05-05
最新评论