mybatis拦截器打印完整带参数sql,可直接放入数据库工具执行

本文主要是介绍mybatis拦截器打印完整带参数sql,可直接放入数据库工具执行,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

前言

如果你使用的是mybatisplus,那至少有3种方法打印sql, 如下博客

mybatisplus开启sql打印的三种方式_mybatisplus打印sql语句配置-CSDN博客

但是其中2种方法参数会打印问号,不能直接放入数据库工具执行

而使用p6spy插件需要引入新依赖,有未知的性能消耗。

因此,这里分享一种通过mybatis拦截器,实现sql打印

代码

package com.gree;import org.apache.ibatis.cache.CacheKey;
import org.apache.ibatis.executor.Executor;
import org.apache.ibatis.mapping.BoundSql;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.ParameterMapping;
import org.apache.ibatis.mapping.ParameterMode;
import org.apache.ibatis.plugin.*;
import org.apache.ibatis.reflection.MetaObject;
import org.apache.ibatis.session.Configuration;
import org.apache.ibatis.session.ResultHandler;
import org.apache.ibatis.session.RowBounds;
import org.apache.ibatis.type.TypeHandlerRegistry;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;import java.sql.Timestamp;
import java.text.SimpleDateFormat;
import java.util.*;/*** 基于Mybatis Plus的SQL输出拦截器。* 完美的输出打印 SQL 及执行时长、statement。* 注意:该插件有性能损耗,不建议生产环境使用。*/@Intercepts(value = {@Signature(type = Executor.class, method = "update", args = {MappedStatement.class, Object.class}),@Signature(type = Executor.class, method = "query", args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class, CacheKey.class, BoundSql.class}),
//        @Signature(type = Executor.class, method = "query", args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class})
})
//@Component用于注入到spring中,后续会被mybatis-plus-boot-starter中的代码从spring中取出并注入到mybatis中
@Component
public class MybatisPlusPrintSqlInterceptor implements Interceptor {Logger logger = LoggerFactory.getLogger(MybatisPlusPrintSqlInterceptor.class);@Overridepublic Object intercept(Invocation invocation) throws Throwable {//1. 执行sqllong proceedStart = System.currentTimeMillis();Object returnValue = null;Exception proceedException = null;//执行sql时catch下异常,即使sql语法报错,也要打印完整sqltry {returnValue = invocation.proceed();} catch (Exception e) {proceedException = e;}long proceedCost = System.currentTimeMillis() - proceedStart;//2.打印sqllong printBegin = System.currentTimeMillis();MappedStatement mappedStatement = (MappedStatement) invocation.getArgs()[0];//部分mybatisplus拦截器内部可能会对Args中的sql进行修改,因此从Args中获取boundSql更接近与真实执行sqlBoundSql boundSql = null;for (int i = invocation.getArgs().length - 1; i >= 0; i--) {if (invocation.getArgs()[i] instanceof BoundSql) {boundSql = (BoundSql) invocation.getArgs()[i];}}if (boundSql == null) {Object parameter = null;if (invocation.getArgs().length > 1) {parameter = invocation.getArgs()[1];}boundSql = mappedStatement.getBoundSql(parameter);}String statement = mappedStatement.getId();Configuration configuration = mappedStatement.getConfiguration();showSql(configuration, boundSql, proceedCost, statement);long printEnd = System.currentTimeMillis();System.out.println("本次打印sql消耗时间:" + (printEnd - printBegin));//3. sql执行异常的报错扔出去if (proceedException != null) {throw proceedException;}return returnValue;}private void showSql(Configuration configuration, BoundSql boundSql, long elapsed, String statement) {String logText = formatMessage(elapsed, getSqlWithValues(boundSql.getSql(), buildParameterValues(configuration, boundSql)), statement);if (Boolean.TRUE == false) {// 打印红色 SQL 日志System.err.println(logText);} else {logger.info("\n{}", logText);}}// com.baomidou.mybatisplus.core.MybatisParameterHandler#setParametersprivate static Map<Integer, Object> buildParameterValues(Configuration configuration, BoundSql boundSql) {Object parameterObject = boundSql.getParameterObject();// ParameterMapping描述参数,包括属性、名称、表达式、javaType、jdbcType、typeHandler等信息List<ParameterMapping> parameterMappings = boundSql.getParameterMappings();if (parameterMappings != null) {Map<Integer, Object> parameterValues = new HashMap<>();//类型处理器用于注册所有的 TypeHandler,并建立 Jdbc 类型、JDBC 类型与 TypeHandler 之间的对应关系TypeHandlerRegistry typeHandlerRegistry = configuration.getTypeHandlerRegistry();for (int i = 0; i < parameterMappings.size(); i++) {ParameterMapping parameterMapping = parameterMappings.get(i);if (parameterMapping.getMode() != ParameterMode.OUT) {Object value;String propertyName = parameterMapping.getProperty();if (boundSql.hasAdditionalParameter(propertyName)) { // issue #448 ask first for additional paramsvalue = boundSql.getAdditionalParameter(propertyName);} else if (parameterObject == null) {value = null;} else if (typeHandlerRegistry.hasTypeHandler(parameterObject.getClass())) {value = parameterObject;} else {MetaObject metaObject = configuration.newMetaObject(parameterObject);value = metaObject.getValue(propertyName);}parameterValues.put(i, new Value(value));}}return parameterValues;}return Collections.emptyMap();}public static String formatMessage( long elapsed, String sql, String statement) {return (sql != null && sql != "") ?
//                " Consume Time:" + elapsed + " ms "  + " (" + statement + ")" + "  Execute SQL:" + sql.replaceAll("[\\s]+", " ")//" Consume Time:" + elapsed + " ms ,  Execute SQL:" + sql.replaceAll("[\\s]+", " ")
//                " Consume Time:" + elapsed + " ms ,  Execute SQL:" + sql" 本次执行sql耗时:" + elapsed + " ms "  + " (" + statement + ")" + "  执行的sql打印:" + sql: "";}@Overridepublic Object plugin(Object target) {return Plugin.wrap(target, this);}@Overridepublic void setProperties(Properties properties0) {}public static String getSqlWithValues(String statementQuery, Map<Integer, Object> parameterValues) {final StringBuilder sb = new StringBuilder();// iterate over the characters in the query replacing the parameter placeholders// with the actual valuesint currentParameter = 0;for (int pos = 0; pos < statementQuery.length(); pos++) {char character = statementQuery.charAt(pos);if (statementQuery.charAt(pos) == '?' && currentParameter <= parameterValues.size()) {// replace with parameter valueObject value = parameterValues.get(currentParameter);sb.append(value != null ? value.toString() : new Value().toString());currentParameter++;} else {sb.append(character);}}return sb.toString();}/*** 基于p6spy的简易数据类型转换类。** @author laiqi* @date 2023-4-4*/public static class Value {public static final String NORM_DATETIME_PATTERN = "yyyy-MM-dd HH:mm:ss";public static final String databaseDialectDateFormat = NORM_DATETIME_PATTERN;public static final String databaseDialectTimestampFormat = NORM_DATETIME_PATTERN;public static final String databaseDialectBooleanFormat = "numeric";private Object value;public Value(Object valueToSet) {this();this.value = valueToSet;}public Value() {}public Object getValue() {return value;}public void setValue(Object value) {this.value = value;}@Overridepublic String toString() {return convertToString(this.value);}public String convertToString(Object value) {String result;if (value == null) {result = "NULL";} else {if (value instanceof byte[]) {result = new String((byte[]) value);} else if (value instanceof Timestamp) {result = new SimpleDateFormat(databaseDialectTimestampFormat).format(value);} else if (value instanceof Date) {result = new SimpleDateFormat(databaseDialectDateFormat).format(value);} else if (value instanceof Boolean) {if ("numeric".equals(databaseDialectBooleanFormat)) {result = Boolean.FALSE.equals(value) ? "0" : "1";} else {result = value.toString();}} else {result = value.toString();}result = quoteIfNeeded(result, value);}return result;}private String quoteIfNeeded(String stringValue, Object obj) {if (stringValue == null) {return null;}if (Number.class.isAssignableFrom(obj.getClass()) || Boolean.class.isAssignableFrom(obj.getClass())) {return stringValue;} else {return "'" + escape(stringValue) + "'";}}private String escape(String stringValue) {return stringValue.replaceAll("'", "''");}}
}

注意

1. 如果你的项目里有mybatis-plus-boot-starter,那只需要把这个类复制到springboot能被扫描到的地方就行,如果项目中没有mybatis-plus-boot-starter,那你可能还需要手动将这个拦截器注入到mybatis里

2. 如果你的项目有有mybatis-plus-boot-starter,并且存在其他mybatis拦截器或者mybatisplus的拦截器,那你需要调整bean的创建顺序,使得该打印sql的拦截器最早注入spring中,这样mybatisplus从spring中获取拦截器的时候,该打印sql的拦截器排在最前面,注入mybatis后,该打印sql的拦截器就会最后执行,你能获得更加准确的sql打印结果

其他参考博客

本文主要代码参考基于以下博客

基于Mybatis Plus的SQL输出拦截器。完美的输出打印 SQL 及执行时长、statement。_mybatissqlprintlnterceptorjava-CSDN博客

这篇关于mybatis拦截器打印完整带参数sql,可直接放入数据库工具执行的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/1106264

相关文章

Python调用Orator ORM进行数据库操作

《Python调用OratorORM进行数据库操作》OratorORM是一个功能丰富且灵活的PythonORM库,旨在简化数据库操作,它支持多种数据库并提供了简洁且直观的API,下面我们就... 目录Orator ORM 主要特点安装使用示例总结Orator ORM 是一个功能丰富且灵活的 python O

JavaScript中的reduce方法执行过程、使用场景及进阶用法

《JavaScript中的reduce方法执行过程、使用场景及进阶用法》:本文主要介绍JavaScript中的reduce方法执行过程、使用场景及进阶用法的相关资料,reduce是JavaScri... 目录1. 什么是reduce2. reduce语法2.1 语法2.2 参数说明3. reduce执行过程

mybatis和mybatis-plus设置值为null不起作用问题及解决

《mybatis和mybatis-plus设置值为null不起作用问题及解决》Mybatis-Plus的FieldStrategy主要用于控制新增、更新和查询时对空值的处理策略,通过配置不同的策略类型... 目录MyBATis-plusFieldStrategy作用FieldStrategy类型每种策略的作

Springboot中分析SQL性能的两种方式详解

《Springboot中分析SQL性能的两种方式详解》文章介绍了SQL性能分析的两种方式:MyBatis-Plus性能分析插件和p6spy框架,MyBatis-Plus插件配置简单,适用于开发和测试环... 目录SQL性能分析的两种方式:功能介绍实现方式:实现步骤:SQL性能分析的两种方式:功能介绍记录

使用 sql-research-assistant进行 SQL 数据库研究的实战指南(代码实现演示)

《使用sql-research-assistant进行SQL数据库研究的实战指南(代码实现演示)》本文介绍了sql-research-assistant工具,该工具基于LangChain框架,集... 目录技术背景介绍核心原理解析代码实现演示安装和配置项目集成LangSmith 配置(可选)启动服务应用场景

oracle DBMS_SQL.PARSE的使用方法和示例

《oracleDBMS_SQL.PARSE的使用方法和示例》DBMS_SQL是Oracle数据库中的一个强大包,用于动态构建和执行SQL语句,DBMS_SQL.PARSE过程解析SQL语句或PL/S... 目录语法示例注意事项DBMS_SQL 是 oracle 数据库中的一个强大包,它允许动态地构建和执行

SQL 中多表查询的常见连接方式详解

《SQL中多表查询的常见连接方式详解》本文介绍SQL中多表查询的常见连接方式,包括内连接(INNERJOIN)、左连接(LEFTJOIN)、右连接(RIGHTJOIN)、全外连接(FULLOUTER... 目录一、连接类型图表(ASCII 形式)二、前置代码(创建示例表)三、连接方式代码示例1. 内连接(I

在MySQL执行UPDATE语句时遇到的错误1175的解决方案

《在MySQL执行UPDATE语句时遇到的错误1175的解决方案》MySQL安全更新模式(SafeUpdateMode)限制了UPDATE和DELETE操作,要求使用WHERE子句时必须基于主键或索引... mysql 中遇到的 Error Code: 1175 是由于启用了 安全更新模式(Safe Upd

PyCharm 接入 DeepSeek最新完整教程

《PyCharm接入DeepSeek最新完整教程》文章介绍了DeepSeek-V3模型的性能提升以及如何在PyCharm中接入和使用DeepSeek进行代码开发,本文通过图文并茂的形式给大家介绍的... 目录DeepSeek-V3效果演示创建API Key在PyCharm中下载Continue插件配置Con

SpringBoot+MyBatis-Flex配置ProxySQL的实现步骤

《SpringBoot+MyBatis-Flex配置ProxySQL的实现步骤》本文主要介绍了SpringBoot+MyBatis-Flex配置ProxySQL的实现步骤,文中通过示例代码介绍的非常详... 目录 目标 步骤 1:确保 ProxySQL 和 mysql 主从同步已正确配置ProxySQL 的