81,094
社区成员
发帖
与我相关
我的任务
分享
@Intercepts(@Signature(type = Executor.class, method = "query", args = { MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class }))
public class QueryConditionHelper implements Interceptor {
@Override
public Object intercept(Invocation invocation) {
Object[] args = invocation.getArgs();
MappedStatement ms = (MappedStatement) args[0];
Object parameter = args[1];
//RowBounds rowBounds = (RowBounds) args[2];
BoundSql boundSql = ms.getBoundSql(parameter);
String sql = boundSql.getSql();
sql+=" AND TABLE.AAA=?";
List<ParameterMapping> parameterMappingList = new ArrayList<>(boundSql.getParameterMappings());
parameterMappingList.add(new ParameterMapping.Builder(ms.getConfiguration(), "AAA", Object.class).build());
BoundSql newBoundSql = new BoundSql(ms.getConfiguration(), sql, parameterMappingList, boundSql.getParameterObject());
// 把新的查询放到statement里
SqlSource sqlSource = new StaticSqlSource(ms.getConfiguration(), sql, parameterMappingList);
MappedStatement newMs = copyFromMappedStatement(ms, sqlSource);
for (ParameterMapping mapping : newBoundSql.getParameterMappings()) {
String prop = mapping.getProperty();
if (boundSql.hasAdditionalParameter(prop)) {
newBoundSql.setAdditionalParameter(prop, boundSql.getAdditionalParameter(prop));
}
}
args[0] = newMs;
return invocation.proceed();
}
private MappedStatement copyFromMappedStatement(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) {
builder.keyProperty(ms.getKeyProperties()[0]);
}
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();
}
}