java通過MySQL驅(qū)動攔截器實(shí)現(xiàn)執(zhí)行sql耗時計(jì)算

更新時間：2023年03月20日 14:25:14 作者：Redick01

本文主要介紹了java通過MySQL驅(qū)動攔截器實(shí)現(xiàn)執(zhí)行sql耗時計(jì)算，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

背景

公司的一個需求，公司既有的鏈路追蹤日志組件要支持MySQL的sql執(zhí)行時間打印，要實(shí)現(xiàn)鏈路追蹤常用的手段就是實(shí)現(xiàn)第三方框架或工具提供的攔截器接口或者是過濾器接口，對于MySQL也不例外，實(shí)際上就是實(shí)現(xiàn)了MySQL驅(qū)動的攔截器接口而已。

具體實(shí)現(xiàn)

MySQL的渠道有不同的版本，不同版本的攔截器接口是不同的，所以要針對你所使用的不同版本的MySQL驅(qū)動去實(shí)現(xiàn)響應(yīng)的攔截器，接下來分別介紹下MySQL渠道5,6,8版本的實(shí)現(xiàn)方式。

MySQL5

這里以MySQL渠道5.1.18版本為例實(shí)現(xiàn)，實(shí)現(xiàn)StatementInterceptorV2接口，主要實(shí)現(xiàn)邏輯在preProcess和postProcess方法，這兩個方法是sql執(zhí)行前后要執(zhí)行的方法，我所使用的框架是logback，這里使用MDC來記錄sql執(zhí)行前的一個時間戳，代碼在postProcess方法MDC.put("sql_exec_time", start);，自己也可以使用ThreadLocal等來實(shí)現(xiàn)，然后在postProcess方法中使用MDC.get("sql_exec_time")將記錄的sql執(zhí)行前的時間取出來，最后再用當(dāng)前時間戳減去sql執(zhí)行前的時間，就算出了sql執(zhí)行的時間。

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"), "開始執(zhí)行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)), "結(jié)束執(zhí)行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"), "開始執(zhí)行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)), "結(jié)束執(zhí)行sql");
        return null;
    }
}

MySQL8

MySQL8和MySQL5/6的攔截器接口又不一樣了，MySQL8的攔截器接口是com.mysql.cj.interceptors.QueryInterceptor，統(tǒng)計(jì)sql執(zhí)行時間的方式還是一樣的，代碼如下：

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"), "開始執(zhí)行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)), "結(jié)束執(zhí)行sql");
        return null;
    }
}

使用方法

MySQL5和6的使用方式一樣，在數(shù)據(jù)庫鏈接的url中增加如下statementInterceptors參數(shù)，例如：

 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參數(shù)，例如：

url: jdbc:mysql://127.0.0.1:3316/log-helper?useUnicode=true&characterEncoding=UTF8&queryInterceptors=com.redick.support.mysql.Mysql8QueryInterceptor&serverTimezone=CST

測試結(jié)果

sql執(zhí)行前日志

{"@timestamp":"2023-02-28T17:16:29.234+08:00","@version":"0.0.1","message":"開始執(zhí)行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執(zhí)行后日志，sql_duration標(biāo)識執(zhí)行sql耗時3ms

{"@timestamp":"2023-02-28T17:16:29.237+08:00","@version":"0.0.1","message":"結(jié)束執(zhí)行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}

具體實(shí)現(xiàn)代碼參考基于Spring AOP + logstash-logback-encoder日志鏈路追蹤工具LogHelper

到此這篇關(guān)于java通過MySQL驅(qū)動攔截器實(shí)現(xiàn)執(zhí)行sql耗時計(jì)算的文章就介紹到這了,更多相關(guān)java執(zhí)行sql耗時計(jì)算內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: