Bagaimanakah java menggunakan pemintas pemacu MySQL untuk melaksanakan pengiraan sql yang memakan masa?

Latar Belakang

Satu keperluan syarikat ialah komponen log penjejakan pautan sedia ada syarikat harus menyokong pencetakan masa pelaksanaan SQL MySQL Kaedah biasa untuk melaksanakan penjejakan pautan adalah dengan melaksanakan rangka kerja atau alatan pihak ketiga antara muka pemintas atau antara muka penapis tidak terkecuali untuk MySQL Malah, ia hanya melaksanakan antara muka pemintas yang didorong oleh MySQL.

Pelaksanaan khusus

Saluran MySQL mempunyai versi yang berbeza, dan antara muka pemintas versi berbeza adalah berbeza, jadi anda perlu melaksanakan pemintas tindak balas mengikut versi berbeza pemacu MySQL yang anda gunakan , Seterusnya, kami akan memperkenalkan kaedah pelaksanaan MySQL saluran 5, 6, dan 8 masing-masing.

MySQL5

Di sini kami mengambil versi MySQL 5.1.18 sebagai contoh untuk melaksanakan antara muka StatementInterceptorV2 Logik pelaksanaan utama adalah dalam kaedah preProcess dan postProcess ini dua kaedah ialah sql Kaedah yang akan dilaksanakan sebelum dan selepas pelaksanaan Rangka kerja yang saya gunakan ialah log balik digunakan di sini untuk merekodkan cap waktu sebelum pelaksanaan SQL untuk melaksanakannya, dan kemudian dalam postProcess Gunakan MDC.put("sql_exec_time", start); dalam kaedah untuk mengambil masa yang direkodkan sebelum pelaksanaan sql, dan akhirnya menolak masa sebelum pelaksanaan sql daripada cap waktu semasa untuk mengira masa pelaksanaan sql. postProcess

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;
    }
}

MDC.get("sql_exec_time")MySQL6

MySQL6 dan MySQL5 pada asasnya sama, tetapi antara muka tidak sama, letak kod terus

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

Pintasan daripada MySQL8 dan MySQL5/6 Antara muka pemintas adalah berbeza lagi Antara muka pemintas MySQL8 ialah

Kaedah mengira masa pelaksanaan SQL adalah seperti berikut:

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;
    }
}

com.mysql.cj.interceptors.QueryInterceptorKaedah penggunaan

Penggunaan MySQL5 dan 6 Dengan cara yang sama, tambahkan parameter

berikut pada url pautan pangkalan data, contohnya:

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

statementInterceptorsMySQL8 menambah

parameter pada url , sebagai contoh:

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

queryInterceptorsHasil ujian

Log sebelum pelaksanaan 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"}

Log selepas pelaksanaan sql, bendera sql_duration mengambil masa 3ms untuk melaksanakan sql

Atas ialah kandungan terperinci Bagaimanakah java menggunakan pemintas pemacu MySQL untuk melaksanakan pengiraan sql yang memakan masa?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!