武
6 years ago
35 changed files with 409 additions and 144 deletions
@ -1,20 +0,0 @@ |
|||||
//package com.ccsens.ht.bean;
|
|
||||
//
|
|
||||
//import org.springframework.cloud.client.loadbalancer.LoadBalanced;
|
|
||||
//import org.springframework.context.annotation.Bean;
|
|
||||
//import org.springframework.context.annotation.Configuration;
|
|
||||
//import org.springframework.web.client.RestTemplate;
|
|
||||
//
|
|
||||
///**
|
|
||||
// * @description:
|
|
||||
// * @author: wuHuiJuan
|
|
||||
// * @create: 2019/11/26 15:45
|
|
||||
// */
|
|
||||
//@Configuration
|
|
||||
//public class BeanConfig {
|
|
||||
// @Bean
|
|
||||
// @LoadBalanced
|
|
||||
// public RestTemplate restTemplate() {
|
|
||||
// return new RestTemplate();
|
|
||||
// }
|
|
||||
//}
|
|
||||
@ -0,0 +1,22 @@ |
|||||
|
package com.ccsens.ht.config; |
||||
|
|
||||
|
import com.ccsens.ht.intercept.MybatisInterceptor; |
||||
|
import org.springframework.context.annotation.Bean; |
||||
|
import org.springframework.context.annotation.Configuration; |
||||
|
|
||||
|
/** |
||||
|
* @description: |
||||
|
* @author: wuHuiJuan |
||||
|
* @create: 2019/12/11 10:59 |
||||
|
*/ |
||||
|
@Configuration |
||||
|
public class BeanConfig { |
||||
|
/** |
||||
|
* 注册拦截器 |
||||
|
*/ |
||||
|
@Bean |
||||
|
public MybatisInterceptor mybatisInterceptor() { |
||||
|
MybatisInterceptor interceptor = new MybatisInterceptor(); |
||||
|
return interceptor; |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,220 @@ |
|||||
|
package com.ccsens.ht.intercept; |
||||
|
|
||||
|
import cn.hutool.core.collection.CollectionUtil; |
||||
|
import com.ccsens.ht.bean.po.HtDoctorExample; |
||||
|
import com.ccsens.ht.uitl.Constant; |
||||
|
import org.apache.ibatis.binding.MapperMethod; |
||||
|
import org.apache.ibatis.cache.CacheKey; |
||||
|
import org.apache.ibatis.executor.CachingExecutor; |
||||
|
import org.apache.ibatis.executor.Executor; |
||||
|
import org.apache.ibatis.executor.parameter.ParameterHandler; |
||||
|
import org.apache.ibatis.executor.statement.PreparedStatementHandler; |
||||
|
import org.apache.ibatis.executor.statement.StatementHandler; |
||||
|
import org.apache.ibatis.mapping.*; |
||||
|
import org.apache.ibatis.plugin.*; |
||||
|
import org.apache.ibatis.reflection.DefaultReflectorFactory; |
||||
|
import org.apache.ibatis.reflection.MetaObject; |
||||
|
import org.apache.ibatis.reflection.factory.DefaultObjectFactory; |
||||
|
import org.apache.ibatis.reflection.wrapper.DefaultObjectWrapperFactory; |
||||
|
import org.apache.ibatis.session.Configuration; |
||||
|
import org.apache.ibatis.session.ResultHandler; |
||||
|
import org.apache.ibatis.session.RowBounds; |
||||
|
import org.springframework.beans.BeanUtils; |
||||
|
import org.springframework.util.ClassUtils; |
||||
|
|
||||
|
import java.beans.PropertyDescriptor; |
||||
|
import java.lang.reflect.Field; |
||||
|
import java.lang.reflect.InvocationTargetException; |
||||
|
import java.lang.reflect.Method; |
||||
|
import java.sql.PreparedStatement; |
||||
|
import java.util.ArrayList; |
||||
|
import java.util.List; |
||||
|
import java.util.Map; |
||||
|
import java.util.Properties; |
||||
|
|
||||
|
/** |
||||
|
* @description: |
||||
|
* @author: wuHuiJuan |
||||
|
* @create: 2019/12/11 10:58 |
||||
|
*/ |
||||
|
@Intercepts({ |
||||
|
@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} |
||||
|
)*/ |
||||
|
/*@Signature( |
||||
|
type = ParameterHandler.class, |
||||
|
method = "setParameters", |
||||
|
args = {PreparedStatement.class} |
||||
|
)*/ |
||||
|
}) |
||||
|
public class MybatisInterceptor implements Interceptor { |
||||
|
private static final String PARAM_KEY = "is_del"; |
||||
|
@Override |
||||
|
public Object intercept(Invocation invocation) throws Throwable { |
||||
|
|
||||
|
// Object[] args = invocation.getArgs();
|
||||
|
// MappedStatement statement = (MappedStatement) args[0];
|
||||
|
// BoundSql boundSql = statement.getBoundSql(args.length > 1 ? args[1] : null);
|
||||
|
// String sql = boundSql.getSql();
|
||||
|
// String whereDel = "is_del";
|
||||
|
// int whereIndex = sql.indexOf("where") > 0 ? sql.indexOf("where") : sql.indexOf("WHERE");
|
||||
|
// boolean hasParam = true;
|
||||
|
// if (whereIndex < 0) {
|
||||
|
// hasParam = false;
|
||||
|
// sql += " where ";
|
||||
|
// whereIndex = sql.indexOf("where");
|
||||
|
// }
|
||||
|
// //sql中不包含is_del，则添加
|
||||
|
// if (!sql.substring(whereIndex).contains(whereDel)) {
|
||||
|
// StringBuilder sqlBuilder = new StringBuilder();
|
||||
|
// sqlBuilder.append(sql.substring(0, whereIndex + "where".length()))
|
||||
|
// .append(hasParam ? " is_del = 0 and " : " is_del = 0 ")
|
||||
|
// .append(sql.substring(whereIndex + "where".length()));
|
||||
|
// //通过反射修改sql语句
|
||||
|
//// Field field = boundSql.getClass().getDeclaredField("sql");
|
||||
|
//// field.setAccessible(true);
|
||||
|
//// field.set(boundSql, builder.toString());
|
||||
|
//
|
||||
|
// System.out.println("------------------");
|
||||
|
// List<ParameterMapping> parameterMappings = boundSql.getParameterMappings();
|
||||
|
// Configuration configuration = statement.getConfiguration();
|
||||
|
// BoundSql newBoundSql = new BoundSql(configuration, sqlBuilder.toString(), parameterMappings, boundSql.getParameterObject());
|
||||
|
// for (ParameterMapping parameterMapping : parameterMappings) {
|
||||
|
// String prop = parameterMapping.getProperty();
|
||||
|
// if (boundSql.hasAdditionalParameter(prop)) {
|
||||
|
// Object param = boundSql.getAdditionalParameter(prop);
|
||||
|
// newBoundSql.setAdditionalParameter(prop, param);
|
||||
|
// }
|
||||
|
// }
|
||||
|
// BoundSqlSource newSqlSource = new BoundSqlSource(newBoundSql);
|
||||
|
// MappedStatement newMappedStatement = copyFromMappedStatement(
|
||||
|
// statement, newSqlSource);
|
||||
|
// args[MAPPED_STATEMENT_INDEX] = newMappedStatement;
|
||||
|
// }
|
||||
|
|
||||
|
String selectByExample = "selectByExample"; |
||||
|
String selectByPrimaryKey = "selectByPrimaryKey"; |
||||
|
|
||||
|
Object[] args = invocation.getArgs(); |
||||
|
MappedStatement statement = (MappedStatement) args[0]; |
||||
|
if (statement.getId().endsWith(selectByExample)) { |
||||
|
//XXXExample
|
||||
|
Object example = args[1]; |
||||
|
Method method = example.getClass().getMethod("getOredCriteria", null); |
||||
|
//获取到条件数组，第一个是Criteria
|
||||
|
List list = (List)method.invoke(example); |
||||
|
if (CollectionUtil.isEmpty(list)) { |
||||
|
Class clazz = ((ResultMap)statement.getResultMaps().get(0)).getType(); |
||||
|
String exampleName = clazz.getName() + "Example"; |
||||
|
Object paramExample = Class.forName(exampleName).newInstance(); |
||||
|
Method createCriteria = paramExample.getClass().getMethod("createCriteria"); |
||||
|
Object criteria = createCriteria.invoke(paramExample); |
||||
|
Method andIsDelEqualTo = criteria.getClass().getMethod("andIsDelEqualTo", Byte.class); |
||||
|
andIsDelEqualTo.invoke(criteria, Constant.Ht.NUMBER_DEFAULT); |
||||
|
list.add(criteria); |
||||
|
} else { |
||||
|
Object criteria = list.get(0); |
||||
|
Method getCriteria = criteria.getClass().getMethod("getCriteria"); |
||||
|
List params = (List)getCriteria.invoke(criteria); |
||||
|
boolean hasDel = false; |
||||
|
for(Object param: params) { |
||||
|
Method getCondition = param.getClass().getMethod("getCondition"); |
||||
|
Object condition = getCondition.invoke(param); |
||||
|
if ("iis_del =".equals(condition)) { |
||||
|
hasDel = true; |
||||
|
} |
||||
|
} |
||||
|
if (!hasDel) { |
||||
|
Method andIsDelEqualTo = criteria.getClass().getMethod("andIsDelEqualTo", Byte.class); |
||||
|
andIsDelEqualTo.invoke(criteria, Constant.Ht.NUMBER_DEFAULT); |
||||
|
} |
||||
|
|
||||
|
} |
||||
|
|
||||
|
|
||||
|
} else if (statement.getId().endsWith(selectByPrimaryKey)) { |
||||
|
BoundSql boundSql = statement.getBoundSql(args[1]); |
||||
|
String sql = boundSql.getSql() + " and is_del = 0"; |
||||
|
MappedStatement newStatement = newMappedStatement(statement, new BoundSqlSqlSource(boundSql)); |
||||
|
MetaObject msObject = MetaObject.forObject(newStatement, new DefaultObjectFactory(), new DefaultObjectWrapperFactory(),new DefaultReflectorFactory()); |
||||
|
msObject.setValue("sqlSource.boundSql.sql", sql); |
||||
|
args[0] = newStatement; |
||||
|
} |
||||
|
|
||||
|
return invocation.proceed(); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public Object plugin(Object target) { |
||||
|
return Plugin.wrap(target, this); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void setProperties(Properties properties) { |
||||
|
|
||||
|
} |
||||
|
|
||||
|
private MappedStatement newMappedStatement(MappedStatement ms, SqlSource newSqlSource) { |
||||
|
MappedStatement.Builder builder = |
||||
|
new MappedStatement.Builder(ms.getConfiguration(), ms.getId(), newSqlSource, ms.getSqlCommandType()); |
||||
|
builder.resource(ms.getResource()); |
||||
|
builder.fetchSize(ms.getFetchSize()); |
||||
|
builder.statementType(ms.getStatementType()); |
||||
|
builder.keyGenerator(ms.getKeyGenerator()); |
||||
|
if (ms.getKeyProperties() != null && ms.getKeyProperties().length != 0) { |
||||
|
StringBuilder keyProperties = new StringBuilder(); |
||||
|
for (String keyProperty : ms.getKeyProperties()) { |
||||
|
keyProperties.append(keyProperty).append(","); |
||||
|
} |
||||
|
keyProperties.delete(keyProperties.length() - 1, keyProperties.length()); |
||||
|
builder.keyProperty(keyProperties.toString()); |
||||
|
} |
||||
|
builder.timeout(ms.getTimeout()); |
||||
|
builder.parameterMap(ms.getParameterMap()); |
||||
|
builder.resultMaps(ms.getResultMaps()); |
||||
|
builder.resultSetType(ms.getResultSetType()); |
||||
|
builder.cache(ms.getCache()); |
||||
|
builder.flushCacheRequired(ms.isFlushCacheRequired()); |
||||
|
builder.useCache(ms.isUseCache()); |
||||
|
|
||||
|
return builder.build(); |
||||
|
} |
||||
|
|
||||
|
private String getOperateType(Invocation invocation) { |
||||
|
final Object[] args = invocation.getArgs(); |
||||
|
MappedStatement ms = (MappedStatement) args[0]; |
||||
|
SqlCommandType commondType = ms.getSqlCommandType(); |
||||
|
if (commondType.compareTo(SqlCommandType.SELECT) == 0) { |
||||
|
return "select"; |
||||
|
} |
||||
|
if (commondType.compareTo(SqlCommandType.INSERT) == 0) { |
||||
|
return "insert"; |
||||
|
} |
||||
|
if (commondType.compareTo(SqlCommandType.UPDATE) == 0) { |
||||
|
return "update"; |
||||
|
} |
||||
|
if (commondType.compareTo(SqlCommandType.DELETE) == 0) { |
||||
|
return "delete"; |
||||
|
} |
||||
|
return null; |
||||
|
} |
||||
|
// 定义一个内部辅助类，作用是包装sq
|
||||
|
class BoundSqlSqlSource implements SqlSource { |
||||
|
private BoundSql boundSql; |
||||
|
public BoundSqlSqlSource(BoundSql boundSql) { |
||||
|
this.boundSql = boundSql; |
||||
|
} |
||||
|
@Override |
||||
|
public BoundSql getBoundSql(Object parameterObject) { |
||||
|
return boundSql; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
} |
||||
Loading…
Reference in new issue