优化mybatis插件，解决线上cpu负载高的问题

integration/mybatis/interceptor/CatMybatisInterceptor.java

@@ -0,0 +1,141 @@
+package com.dianping.cat.plugins;
+
+import com.dianping.cat.Cat;
+import com.dianping.cat.CatConstants;
+import com.dianping.cat.message.Message;
+import com.dianping.cat.message.Transaction;
+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 java.text.DateFormat;
+import java.util.Date;
+import java.util.List;
+import java.util.Locale;
+import java.util.Properties;
+
+@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 CatMybatisInterceptor implements Interceptor {
+
+    private Properties properties;
+
+    private String datasourceUrl;
+
+    public CatMybatisInterceptor(String datasourceUrl) {
+        this.datasourceUrl = datasourceUrl;
+    }
+
+    public Object intercept(Invocation invocation) throws Throwable {
+        //begin cat transaction
+        MappedStatement mappedStatement = (MappedStatement) invocation.getArgs()[0];
+        String methodName = this.getMethodName(mappedStatement);
+        Transaction t = Cat.newTransaction(CatConstants.TYPE_SQL, methodName);
+
+        Cat.logEvent("SQL.Database", datasourceUrl);
+        Cat.logEvent("SQL.Method", mappedStatement.getSqlCommandType().name().toLowerCase(), Message.SUCCESS, getSql(invocation, mappedStatement));
+        try {
+            Object returnValue = invocation.proceed();
+            t.setStatus(Transaction.SUCCESS);
+            return returnValue;
+        } catch (Throwable e) {
+            Cat.logError(e);
+            t.setStatus(e);
+            throw e;
+        } finally {
+            t.complete();
+        }
+
+    }
+
+    private String getMethodName(MappedStatement mappedStatement) {
+        String[] strArr = mappedStatement.getId().split("\\.");
+        String methodName = strArr[strArr.length - 2] + "." + strArr[strArr.length - 1];
+        return methodName;
+    }
+
+    private String getSql(Invocation invocation, MappedStatement mappedStatement) {
+        Object parameter = null;
+        if (invocation.getArgs().length > 1) {
+            parameter = invocation.getArgs()[1];
+        }
+        BoundSql boundSql = mappedStatement.getBoundSql(parameter);
+        Configuration configuration = mappedStatement.getConfiguration();
+        return showSql(configuration, boundSql);
+    }
+
+    private static String getParameterValue(Object obj) {
+        StringBuilder retStringBuilder = new StringBuilder();
+        if (obj instanceof String) {
+            retStringBuilder.append("'").append(obj).append("'");
+        } else if (obj instanceof Date) {
+            DateFormat formatter = DateFormat.getDateTimeInstance(DateFormat.DEFAULT, DateFormat.DEFAULT, Locale.CHINA);
+            retStringBuilder.append("'").append(formatter.format(new Date())).append("'");
+        } else {
+            retStringBuilder.append("'").append(obj == null ? "" : obj).append("'");
+        }
+        return retStringBuilder.toString();
+    }
+
+    public static String showSql(Configuration configuration, BoundSql boundSql) {
+        Object parameterObject = boundSql.getParameterObject();
+        List<ParameterMapping> parameterMappings = boundSql.getParameterMappings();
+        String sql = boundSql.getSql().replaceAll("[\\s]+", " ");
+        StringBuilder sqlBuilder = new StringBuilder(sql);
+        if (parameterMappings.size() > 0 && parameterObject != null) {
+            int start = sqlBuilder.indexOf("?");
+            int end = start + 1;
+
+            TypeHandlerRegistry typeHandlerRegistry = configuration.getTypeHandlerRegistry();
+            if (typeHandlerRegistry.hasTypeHandler(parameterObject.getClass())) {
+                sqlBuilder.replace(start, end, 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);
+                        sqlBuilder.replace(start, end, getParameterValue(obj));
+                    } else if (boundSql.hasAdditionalParameter(propertyName)) {
+                        Object obj = boundSql.getAdditionalParameter(propertyName);
+                        sqlBuilder.replace(start, end, getParameterValue(obj));
+                    }
+
+                    start = sqlBuilder.indexOf("?");
+                    end = start + 1;
+                }
+            }
+        }
+        return sqlBuilder.toString();
+    }
+
+    public Object plugin(Object target) {
+        return Plugin.wrap(target, this);
+    }
+
+    @Override
+    public void setProperties(Properties properties) {
+        this.properties = properties;
+    }
+
+    public String getDatasourceUrl() {
+        return datasourceUrl;
+    }
+
+    public void setDatasourceUrl(String datasourceUrl) {
+        this.datasourceUrl = datasourceUrl;
+    }
+}

integration/mybatis/interceptor/README.md

@@ -0,0 +1,17 @@
+
+# mybatis 配置插件 
+
+```
+     @Value("${jdbc-url}")
+     private String jdbcUrl;
+         
+     @Bean
+     public SqlSessionFactory mysqlSessionFactory(DataSource mysqlDataSource) throws Exception {
+         SqlSessionFactoryBean sqlSessionFactoryBean = new SqlSessionFactoryBean();
+         sqlSessionFactoryBean.setDataSource(mysqlDataSource);
+         PathMatchingResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
+         sqlSessionFactoryBean.setMapperLocations(resolver.getResources(mapperLocations));
+         sqlSessionFactoryBean.setPlugins(new Interceptor[]{new CatMybatisInterceptor(jdbcUrl)});
+         return sqlSessionFactoryBean.getObject();
+     }
+```