com.github.wzc789376152.springboot.config.mybatisplus.MybatisPlusInterceptor Maven / Gradle / Ivy
The newest version!
package com.github.wzc789376152.springboot.config.mybatisplus;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.annotation.TableName;
import com.baomidou.mybatisplus.core.conditions.AbstractWrapper;
import com.github.wzc789376152.springboot.utils.SqlUtils;
import com.github.wzc789376152.utils.DateUtils;
import lombok.SneakyThrows;
import org.apache.commons.lang3.StringUtils;
import org.apache.ibatis.binding.MapperMethod;
import org.apache.ibatis.cache.CacheKey;
import org.apache.ibatis.executor.Executor;
import org.apache.ibatis.executor.parameter.ParameterHandler;
import org.apache.ibatis.executor.statement.RoutingStatementHandler;
import org.apache.ibatis.executor.statement.StatementHandler;
import org.apache.ibatis.mapping.BoundSql;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.ParameterMapping;
import org.apache.ibatis.mapping.SqlCommandType;
import org.apache.ibatis.plugin.Intercepts;
import org.apache.ibatis.plugin.Invocation;
import org.apache.ibatis.plugin.Signature;
import org.apache.ibatis.session.ResultHandler;
import org.apache.ibatis.session.RowBounds;
import org.apache.shardingsphere.sharding.yaml.config.YamlShardingRuleConfiguration;
import org.apache.shardingsphere.sharding.yaml.config.rule.YamlTableRuleConfiguration;
import java.lang.reflect.Field;
import java.sql.Connection;
import java.sql.Statement;
import java.util.*;
@Intercepts({@Signature(
type = StatementHandler.class,
method = "parameterize",
args = {Statement.class}
)
, @Signature(
type = StatementHandler.class,
method = "prepare",
args = {Connection.class, Integer.class}
), @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}
), @Signature(
type = Executor.class,
method = "query",
args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class, CacheKey.class, BoundSql.class}
)
})
public class MybatisPlusInterceptor extends com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor {
private final YamlShardingRuleConfiguration yamlShardingRuleConfiguration;
private Map shardTableMap;
public MybatisPlusInterceptor(YamlShardingRuleConfiguration yamlShardingRuleConfiguration) {
this.yamlShardingRuleConfiguration = yamlShardingRuleConfiguration;
getShardingTables();
}
@Override
public Object intercept(Invocation invocation) throws Throwable {
String method = invocation.getMethod().getName();
Object obj;
String sql = null;
MapperMethod.ParamMap paramMap = null;
if (method.equals("parameterize")) {
obj = parameterize(invocation);
BoundSql boundSql = getBoundSql(invocation);
sql = boundSql.getSql();
if (boundSql.getParameterObject() instanceof MapperMethod.ParamMap) {
paramMap = ((MapperMethod.ParamMap) boundSql.getParameterObject());
}
} else {
obj = super.intercept(invocation);
Object target = invocation.getTarget();
Object[] args = invocation.getArgs();
if (target instanceof Executor) {
boolean isUpdate = args.length == 2;
MappedStatement ms = (MappedStatement) args[0];
if (!isUpdate && ms.getSqlCommandType() == SqlCommandType.SELECT) {
BoundSql boundSql;
if (args.length == 4) {
Object parameter = args[1];
boundSql = ms.getBoundSql(parameter);
} else {
boundSql = (BoundSql) args[5];
}
sql = boundSql.getSql();
if (boundSql.getParameterObject() instanceof MapperMethod.ParamMap) {
paramMap = ((MapperMethod.ParamMap) boundSql.getParameterObject());
}
} else {
Object parameter = args[1];
BoundSql boundSql = ms.getBoundSql(parameter);
sql = boundSql.getSql();
if (boundSql.getParameterObject() instanceof MapperMethod.ParamMap) {
paramMap = ((MapperMethod.ParamMap) boundSql.getParameterObject());
}
}
}
}
if (StringUtils.isNotEmpty(sql)) {
SqlUtils.getInstance().setSql(sql);
if (paramMap != null) {
Object ew = paramMap.containsKey("param1") ? paramMap.get("param1") : null;
if (ew instanceof AbstractWrapper) {
AbstractWrapper wrapper = (AbstractWrapper) ew;
String expressStr = wrapper.getCustomSqlSegment();
for (Object key : wrapper.getParamNameValuePairs().keySet()) {
String reset = "#\\{ew.paramNameValuePairs." + key + "}";
Object value = wrapper.getParamNameValuePairs().get(key);
if (value instanceof Date) {
value = DateUtils.format((Date) value);
}
expressStr = expressStr.replaceAll(reset, value.toString());
}
SqlUtils.getInstance().setParams(expressStr);
}
}
}
return obj;
}
@SneakyThrows
private Object parameterize(Invocation invocation) {
BoundSql boundSql = getBoundSql(invocation);
MappedStatement mappedStatement = getMappedStatement(invocation);
// 判断是否为更新方法
if (SqlCommandType.UPDATE != mappedStatement.getSqlCommandType()) {
return invocation.proceed();
}
// 由于拦截器的本质是一个拦截器列表循环执行,所以需要重新获取相关数据
Class pojoClazz = mappedStatement.getParameterMap().getType();
if (pojoClazz == null || !pojoClazz.isAnnotationPresent(TableName.class)) {
return invocation.proceed();
}
TableName annotation = pojoClazz.getAnnotation(TableName.class);
if (shardTableMap.containsKey(annotation.value())) {
YamlTableRuleConfiguration shardConfig = shardTableMap.get(annotation.value());
List shardingColumns = new ArrayList<>();
if (shardConfig.getTableStrategy().getStandard() != null) {
shardingColumns.add(shardConfig.getTableStrategy().getStandard().getShardingColumn());
}
if (shardConfig.getTableStrategy().getComplex() != null) {
shardingColumns.addAll(Arrays.asList(shardConfig.getTableStrategy().getComplex().getShardingColumns().split(",")));
}
for (String shardingColumn : shardingColumns) {
if (shardingColumn != null) {
// 数据库属性是下划线,代码是驼峰命名,所以需要对分表属性进行下划线转驼峰
String camelCase = StrUtil.toCamelCase(shardingColumn);
// 参数添加分表属性
ParameterMapping parameterMapping = new ParameterMapping.Builder(mappedStatement.getConfiguration(), "et." + camelCase, Object.class).build();
boundSql.getParameterMappings().add(parameterMapping);
}
}
}
return invocation.proceed();
}
private MappedStatement getMappedStatement(Invocation invocation) throws NoSuchFieldException, IllegalAccessException {
RoutingStatementHandler routingStatementHandler = (RoutingStatementHandler) invocation.getTarget();
ParameterHandler parameterHandler = routingStatementHandler.getParameterHandler();
// 通过反射获取BoundSql,MappedStatement对象
Field mappedStatementField = parameterHandler.getClass().getDeclaredField("mappedStatement");
mappedStatementField.setAccessible(true);
return (MappedStatement) mappedStatementField.get(parameterHandler);
}
private BoundSql getBoundSql(Invocation invocation) {
RoutingStatementHandler routingStatementHandler = (RoutingStatementHandler) invocation.getTarget();
return routingStatementHandler.getBoundSql();
}
private void getShardingTables() {
if (null != shardTableMap) {
return;
}
synchronized (MybatisPlusInterceptor.class) {
if (null != shardTableMap) {
return;
}
if (null == yamlShardingRuleConfiguration) {
shardTableMap = Collections.emptyMap();
return;
}
shardTableMap = yamlShardingRuleConfiguration.getTables();
}
}
}