package com.dy.common.mybatis;
|
|
import java.lang.reflect.InvocationTargetException;
|
import java.util.Properties;
|
|
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.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.Configuration;
|
import org.apache.ibatis.session.ResultHandler;
|
import org.apache.ibatis.session.RowBounds;
|
|
/**
|
* 拦截执行SQL发生异常的场景，并将执行错误，执行SQL和参数打印出来
|
* 拦截Executor里面的query和update方法
|
*/
|
@Intercepts({
|
@Signature(
|
method = "query",
|
type = Executor.class,
|
args = {
|
MappedStatement.class,
|
Object.class,
|
RowBounds.class,
|
ResultHandler.class
|
}
|
),
|
@Signature(
|
type = Executor.class,
|
method = "update",
|
args = {
|
MappedStatement.class,
|
Object.class
|
}
|
)
|
})
|
@Slf4j
|
public class PrintExceptionSqlInterceptor implements Interceptor {
|
|
@Override
|
public Object intercept(Invocation invocation) throws Throwable {
|
// 获取执行方法的MappedStatement参数
|
Object[] args = invocation.getArgs();
|
if(args != null && args.length > 1){
|
MappedStatement mappedStatement = (MappedStatement) args[0];
|
Object parameter = args[1];
|
|
String sqlId = mappedStatement.getId();
|
BoundSql boundSql = mappedStatement.getBoundSql(parameter);
|
Configuration configuration = mappedStatement.getConfiguration();
|
Object response;
|
try {
|
response = invocation.proceed();
|
} catch (Exception e) {
|
// 输出SQL异常信息
|
log.error("SQL ErrorException:", e);
|
log.error("SQL Id: {}", sqlId);
|
log.error("SQL Parameters: {}", boundSql.getParameterObject());
|
log.error("SQL: {}", PrintSqlHelp.getFullSql(configuration, boundSql));
|
// 根据源异常类型进行返回
|
if (e instanceof InvocationTargetException) {
|
throw new InvocationTargetException(e);
|
} else if (e instanceof IllegalAccessException) {
|
throw new IllegalAccessException(e.getMessage());
|
} else {
|
throw new RuntimeException(e);
|
}
|
}
|
return response;
|
}else{
|
return invocation.proceed();
|
}
|
}
|
|
/**
|
* 通过该方法决定要返回的对象是目标对象还是对应的代理
|
* 不要想的太复杂，一般就两种情况：
|
* <p>
|
* 1. return target; 直接返回目标对象，相当于当前Interceptor没起作用，不会调用上面的intercept()方法
|
* 2. return Plugin.wrap(target, this); 返回代理对象，会调用上面的intercept()方法
|
*
|
* @param target 目标对象
|
* @return 目标对象或者代理对象
|
*/
|
@Override
|
public Object plugin(Object target) {
|
return Plugin.wrap(target, this);
|
}
|
|
/**
|
* 用于获取在Configuration初始化当前的Interceptor时时候设置的一些参数
|
*
|
* @param properties Properties参数
|
*/
|
@Override
|
public void setProperties(Properties properties) {
|
}
|
|
|
}
|
