mybatis 拦截器获取sql语句

1.需求场景

在多数据源业务场景下，获取不同数据源执行的sql语句。

2.项目环境

spring spring mvc mybatis （基于注解）

3.相关配置方法

(1)mybatis拦截器

package com.demo.action;

import java.text.DateFormat;
import java.util.Date;
import java.util.List;
import java.util.Locale;
import java.util.Properties;

import org.apache.commons.dbcp.BasicDataSource;
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.plugin.Interceptor;
import org.apache.ibatis.plugin.Intercepts;
import org.apache.ibatis.plugin.Invocation;
import org.apache.ibatis.plugin.Plugin;
import org.apache.ibatis.plugin.Signature;
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 com.jlc.util.FileUtil;

 @Intercepts({  
    @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 }) })  
public class SqlLogInterceptor implements Interceptor{
	@Override
	public Object intercept(Invocation invocation) throws Throwable {
		 MappedStatement mappedStatement = (MappedStatement) invocation.getArgs()[0];  
	        Object parameter = null;  
	        if (invocation.getArgs().length > 1) {  
	            parameter = invocation.getArgs()[1];  
	        }  

	        BoundSql boundSql = mappedStatement.getBoundSql(parameter);  
	        Configuration configuration = mappedStatement.getConfiguration();  
	       
	        Object returnValue = null;  
	        //获取sql语句
	        String sql =showSql(configuration, boundSql);  
	        //获取数据源
	        BasicDataSource db = (BasicDataSource) configuration.getEnvironment().getDataSource();
	        
	         String pname="";
	         sql=sql.replaceAll("\'","").replace("\"", "");
	        // sql+=sql;
	         String sp = sql.toLowerCase().replace("call", "");
	         if(sp.startsWith("{")){
	        	int s = sp.indexOf("{")+1;
	        	int e = sp.indexOf("(");
	        	pname = sp.substring(s,e);
	         }
	       // System.out.println(sql);
	        String url = db.getUrl();
	        int s = url.indexOf("?");
	        System.out.println(db.getUrl());
	        String updatestr = "update tbl_procedure set updatedate=NOW(),updator='czq',remark='"+sql+"',dbname='"+url.substring(0,s)+"'"+",pname='"+pname+"'"+" where redirect=";
	        FileUtil.writeFile(updatestr,"d:/src/log.txt");
	        //执行结果
	        returnValue = invocation.proceed();  
	        return returnValue;  
	}

	@Override
	public Object plugin(Object target) {
		// TODO Auto-generated method stub
		return Plugin.wrap(target, this);
	}

	@Override
	public void setProperties(Properties properties) {
		// TODO Auto-generated method stub
	}
	
	
 private String getParameterValue(Object obj) {  
        String value = null;  
        if (obj instanceof String) {  
            value = "'" + obj.toString() + "'";  
    } else if (obj instanceof Date) {  
        DateFormat formatter = DateFormat.getDateTimeInstance(DateFormat.DEFAULT, DateFormat.DEFAULT, Locale.CHINA);  
        value = "'" + formatter.format(obj) + "'";  
//	            System.out.println(value);  
    } else {  
        if (obj != null) {  
            value = obj.toString();  
        } else {  
            value = "";  
            }  
   
        }  
        return value;  
    }  
	   
public String showSql(Configuration configuration, BoundSql boundSql) {  
    Object parameterObject = boundSql.getParameterObject();  
    List<ParameterMapping> parameterMappings = boundSql.getParameterMappings();  
    String sql = boundSql.getSql().replaceAll("[\\s]+", " ");  
    if (parameterMappings.size() > 0 && parameterObject != null) {  
        TypeHandlerRegistry typeHandlerRegistry = configuration.getTypeHandlerRegistry();  
        if (typeHandlerRegistry.hasTypeHandler(parameterObject.getClass())) {  
            sql = sql.replaceFirst("\\?", getParameterValue(parameterObject));  
   
            } else {  
                MetaObject metaObject = configuration.newMetaObject(parameterObject);  
                for (ParameterMapping parameterMapping : parameterMappings) {  
                    String propertyName = parameterMapping.getProperty();  
                    if (metaObject.hasGetter(propertyName)) {  
                        Object obj = metaObject.getValue(propertyName);  
                        sql = sql.replaceFirst("\\?", getParameterValue(obj));  
                } else if (boundSql.hasAdditionalParameter(propertyName)) {  
                    Object obj = boundSql.getAdditionalParameter(propertyName);  
                    sql = sql.replaceFirst("\\?", getParameterValue(obj));  
                }  
            }  
        }  
    }  
    return sql;  
}
}

/***
	 * 
	* @Title: writeFile 
	* @Description: TODO 将内容写入文件
	* @param filepath
	* @param data
	 */
	public static void writeFile(String data,String resultfilepath){
		try {
		    File file = new File(resultfilepath);
		    if(!file.exists()){
			    file.createNewFile();
			    }
		    FileWriter fw = null;
		    //true:表示是追加的标志
		    fw = new FileWriter(file, true);
		    fw.write(data);
		    fw.close();
		} catch (Exception e) {
			e.printStackTrace();
		}finally{
			
		}
	}

(2)mybaits-config.xml配置

<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE configuration PUBLIC "-//mybatis.org//DTD Config 3.0//EN"  
"http://mybatis.org/dtd/mybatis-3-config.dtd">
<configuration>
   <settings>
   <!-- 开发配置：输出sql日志到控制台 -->
   <setting name="logImpl" value="STDOUT_LOGGING"/> 
   </settings>
	<typeAliases>
		<package name="com.jlc.bean" />
	</typeAliases>
	<plugins>
	<plugin interceptor="com.jlc.action.SqlLogInterceptor"></plugin>
	</plugins>
	<mappers>
		<mapper resource="xxx/xx/demo.xml" />
</mappers>

</configuration>

扫描二维码关注公众号，回复： 945725 查看本文章