package com.cloudroam.common.interceptor;
|
|
import com.baomidou.mybatisplus.core.toolkit.StringPool;
|
|
import java.text.DateFormat;
|
|
import java.util.Date;
|
|
import java.util.List;
|
|
import java.util.Locale;
|
|
import java.util.Properties;
|
|
import com.cloudroam.common.LocalLogger;
|
import com.cloudroam.model.LogDO;
|
import lombok.extern.slf4j.Slf4j;
|
|
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;
|
|
@Slf4j
|
|
@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 PrintSqlInterceptor 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];
|
|
}
|
|
String sqlId = mappedStatement.getId();
|
|
BoundSql boundSql = mappedStatement.getBoundSql(parameter);
|
|
Configuration configuration = mappedStatement.getConfiguration();
|
|
long start = System.currentTimeMillis();
|
|
Object returnValue = invocation.proceed();
|
|
long time = System.currentTimeMillis() - start;
|
|
showSql(configuration, boundSql, time, sqlId);
|
|
return returnValue;
|
|
}
|
|
private static void showSql(Configuration configuration, BoundSql boundSql, long time, String sqlId){
|
|
Object parameterObject = boundSql.getParameterObject();
|
|
List<ParameterMapping> parameterMappings = boundSql.getParameterMappings();
|
|
//替换空格、换行、tab缩进等
|
|
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));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
final LogDO localLogDO = LocalLogger.getLocalLogDO();
|
if(null!=localLogDO){
|
localLogDO.setQueryParams(sql);
|
}
|
|
logs(time, sql, sqlId);
|
|
}
|
|
private static String getParameterValue(Object obj){
|
|
String value;
|
|
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(new Date()) + "'";
|
|
} else {
|
|
if (obj != null) {
|
|
value = obj.toString();
|
|
} else {
|
|
value = "";
|
|
}
|
|
}
|
|
|
return value.replace("$", "\\$");
|
|
}
|
|
private static void logs(long time, String sql, String sqlId){
|
|
StringBuilder sb = new StringBuilder()
|
|
.append(" Time：").append(time)
|
|
.append(" ms - ID：").append(sqlId)
|
|
.append(StringPool.NEWLINE).append("Execute SQL：")
|
|
.append(sql).append(StringPool.NEWLINE);
|
|
// log.info(sb.toString());
|
|
}
|
|
@Override
|
|
public Object plugin(Object target){
|
|
return Plugin.wrap(target, this);
|
|
}
|
|
@Override
|
|
public void setProperties(Properties properties0){
|
|
}
|
|
}
|
