mybatis 拦截器处理日志

秒速五厘米 2021-09-28 13:08 728阅读 0赞

参考文章：  
[https://www.mybatis.org/mybatis-3/zh/configuration.html\#plugins][https_www.mybatis.org_mybatis-3_zh_configuration.html_plugins]  
[https://www.cnblogs.com/fangjian0423/p/mybatis-interceptor.html][https_www.cnblogs.com_fangjian0423_p_mybatis-interceptor.html]  
[https://www.cnblogs.com/Xrq730/P/6972268.Html][https_www.cnblogs.com_Xrq730_P_6972268.Html]

目前只支持map参数、字符串参数类型、oracle语法。

配置文件中加入

package connTest.conn;
    
    import java.sql.Statement;
    import java.util.List;
    import java.util.Map;
    import java.util.Properties;
    
    import org.apache.ibatis.executor.resultset.FastResultSetHandler;
    import org.apache.ibatis.executor.resultset.ResultSetHandler;
    import org.apache.ibatis.executor.statement.StatementHandler;
    import org.apache.ibatis.mapping.BoundSql;
    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.session.ResultHandler;
    
    @Intercepts({ @Signature(type = StatementHandler.class, method = "query", args = { Statement.class, ResultHandler.class}),
        @Signature(type = StatementHandler.class, method = "update", args = { Statement.class}),
        @Signature(type = StatementHandler.class, method = "batch", args = {  Statement.class })})
    public class LogInterceptor implements Interceptor { 
    
        @Override
        public Object intercept(Invocation invocation) throws Throwable { 
            Object target = invocation.getTarget();
            
            long startTime = System.currentTimeMillis();
            
            StatementHandler statementHandler = (StatementHandler)target;
            
            try { 
                return invocation.proceed();
            } finally { 
                long endTime = System.currentTimeMillis();
                long sqlCost = endTime - startTime;
                
                BoundSql boundSql = statementHandler.getBoundSql();
                String sql = boundSql.getSql();
                Object parameterObject = boundSql.getParameterObject();
                List<ParameterMapping> parameterMappingList = boundSql.getParameterMappings();
                
                // 格式化Sql语句，去除换行符，替换参数
                sql = formatSql(sql, parameterObject, parameterMappingList);
                
                System.out.println("执行耗时[" + sqlCost + "ms] " + "SQL：" + sql + ";");
                
            }
        }
    
        @Override
        public Object plugin(Object target) { 
            return Plugin.wrap(target, this);
        }
    
        @Override
        public void setProperties(Properties properties) { 
            
        }
        
        @SuppressWarnings("unchecked")
        private String formatSql(String sql, Object parameterObject, List<ParameterMapping> parameterMappingList) { 
            
    // if(sql == null || sql.length() == 0 || parameterObject == null 
    // || parameterMappingList == null || parameterMappingList.size() == 0){ 
    // return sql;
    // }
        	
        	if(parameterObject == null){ 
        		sql = sql.replaceAll(" \\?", " null");
        		sql = sql.replaceAll("\r|\n|\t", " ").replaceAll("\\s+", " ");
        		return sql;
        	}
            
            //System.out.println(" class : " + parameterObject.getClass());
        	//bean需要反射，有时间再写
            if(Map.class.isAssignableFrom(parameterObject.getClass())){ 
            	Map pb = (Map)parameterObject;
            	for(ParameterMapping temp: parameterMappingList){ 
            		
            		//System.out.println(temp.getJavaType());
            		sql = sql.replaceFirst(" \\?", pb.get(temp.getProperty()) == null ? " null" : " '" +(String)pb.get(temp.getProperty()) + "'");
            		sql = sql.replaceAll("\r|\n|\t", " ").replaceAll("\\s+", " ");
            	}
            }
            
            return sql;
        }
        
    }

效果

执行耗时[37ms] SQL：select 'a1' "p1", null p2, 'a3' from dual where 1 = 1 and 'a1' = '1';

[https_www.mybatis.org_mybatis-3_zh_configuration.html_plugins]: https://www.mybatis.org/mybatis-3/zh/configuration.html#plugins
[https_www.cnblogs.com_fangjian0423_p_mybatis-interceptor.html]: https://www.cnblogs.com/fangjian0423/p/mybatis-interceptor.html
[https_www.cnblogs.com_Xrq730_P_6972268.Html]: https://www.cnblogs.com/Xrq730/P/6972268.Html