use of org.apache.ibatis.executor.statement.BaseStatementHandler in project mybatis.flying by limeng32.
the class AutoMapperInterceptor method intercept.
@Override
public Object intercept(Invocation invocation) throws Throwable {
StatementHandler statementHandler = (StatementHandler) invocation.getTarget();
MetaObject metaStatementHandler = getRealObj(statementHandler);
String originalSql = (String) metaStatementHandler.getValue(DELEGATE_BOUNDSQL_SQL);
Configuration configuration = (Configuration) metaStatementHandler.getValue(DELEGATE_CONFIGURATION);
Object parameterObject = metaStatementHandler.getValue(DELEGATE_BOUNDSQL_PARAMETEROBJECT);
FlyingModel flyingModel = CookOriginalSql.fetchFlyingFeature(originalSql);
MappedStatement mappedStatement = (MappedStatement) metaStatementHandler.getValue(DELEGATE_MAPPEDSTATEMENT);
if (flyingModel.isHasFlyingFeature()) {
if ((flyingModel.getDataSourceId() != null) && !((Connection) invocation.getArgs()[0]).getCatalog().equalsIgnoreCase(flyingModel.getConnectionCatalog())) {
ApplicationContext applicationContext = ApplicationContextProvider.getApplicationContext();
if (applicationContext != null) {
DataSource dataSource = (DataSource) applicationContext.getBean(flyingModel.getDataSourceId());
if (dataSource == null) {
throw new AutoMapperException(AutoMapperExceptionEnum.cannotFindAssignedDataSourceInContext.description());
}
Connection connection = ((SmartDataSource) (applicationContext.getBean(flyingModel.getDataSourceId()))).getConnection();
invocation.getArgs()[0] = connection;
} else {
throw new AutoMapperException(AutoMapperExceptionEnum.cannotFindApplicationContextProvider.description());
}
}
String newSql = "";
switch(flyingModel.getActionType()) {
case count:
newSql = SqlBuilder.buildCountSql(parameterObject);
break;
case delete:
newSql = SqlBuilder.buildDeleteSql(parameterObject);
break;
case insert:
newSql = SqlBuilder.buildInsertSql(parameterObject, flyingModel);
break;
case select:
newSql = SqlBuilder.buildSelectSql(mappedStatement.getResultMaps().get(0).getType(), flyingModel);
break;
case selectAll:
newSql = SqlBuilder.buildSelectAllSql(parameterObject, flyingModel);
break;
case selectOne:
newSql = SqlBuilder.buildSelectOneSql(parameterObject, flyingModel);
break;
case update:
newSql = SqlBuilder.buildUpdateSql(parameterObject, flyingModel);
break;
case updatePersistent:
newSql = SqlBuilder.buildUpdatePersistentSql(parameterObject, flyingModel);
break;
default:
break;
}
logger.warn(new StringBuffer("Auto generated sql:").append(newSql).toString());
SqlSource sqlSource = buildSqlSource(configuration, newSql, parameterObject.getClass());
List<ParameterMapping> parameterMappings = sqlSource.getBoundSql(parameterObject).getParameterMappings();
metaStatementHandler.setValue(DELEGATE_BOUNDSQL_SQL, sqlSource.getBoundSql(parameterObject).getSql());
metaStatementHandler.setValue(DELEGATE_BOUNDSQL_PARAMETERMAPPINGS, parameterMappings);
}
/* 开始处理分页问题 */
if (invocation.getTarget() instanceof RoutingStatementHandler) {
BaseStatementHandler delegate = (BaseStatementHandler) ReflectHelper.getValueByFieldName(statementHandler, DELEGATE);
mappedStatement = (MappedStatement) ReflectHelper.getValueByFieldName(delegate, MAPPEDSTATEMENT);
BoundSql boundSql = delegate.getBoundSql();
if (parameterObject == null) {
throw new AutoMapperException(AutoMapperExceptionEnum.parameterObjectIsNull);
} else if (parameterObject instanceof Conditionable) {
Conditionable condition = (Conditionable) parameterObject;
String sql = boundSql.getSql();
if (condition.getLimiter() != null) {
Connection connection = (Connection) invocation.getArgs()[0];
String countSql = new StringBuffer("select count(0) from (").append(sql).append(") myCount").toString();
PreparedStatement countStmt = connection.prepareStatement(countSql);
BoundSql countBS = new BoundSql(mappedStatement.getConfiguration(), countSql, boundSql.getParameterMappings(), parameterObject);
setParameters(countStmt, mappedStatement, countBS, parameterObject);
ResultSet rs = countStmt.executeQuery();
int count = 0;
if (rs.next()) {
count = rs.getInt(1);
}
rs.close();
countStmt.close();
condition.getLimiter().setTotalCount(count);
}
String pageSql = generatePageSql(sql, condition);
ReflectHelper.setValueByFieldName(boundSql, SQL, pageSql);
} else {
}
}
/* 调用原始statementHandler的prepare方法完成原本的逻辑 */
statementHandler = (StatementHandler) metaStatementHandler.getOriginalObject();
statementHandler.prepare((Connection) invocation.getArgs()[0], mappedStatement.getTimeout());
/* 传递给下一个拦截器处理 */
return invocation.proceed();
}
Aggregations