这篇文章主要介绍“mybatis插件怎么实现自定义改写表名”,在日常操作中,相信很多人在mybatis插件怎么实现自定义改写表名问题上存在疑惑,小编查阅了各式资料,整理出简单好用的操作方法,希望对大家解答”mybatis插件怎么实现自定义改写表名”的疑惑有所帮助!接下来,请跟着小编一起来学习吧!
@Intercepts({@Signature(type = Executor.class, method = "query", args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class}),
@Signature(type = Executor.class, method = "update", args = {MappedStatement.class, Object.class}),})
public class RerouteToTableInterceptor implements Interceptor {
private Map map;
private Set<String> tableSet;
public static boolean openInterceptor = false;
public RerouteToTableInterceptor() {
//标识使用了该插件
setOpenInterceptor(true);
}
@Override
public Object intercept(Invocation invocation) throws Throwable {
Object[] args = invocation.getArgs();
MappedStatement ms = (MappedStatement) args[0];
Object parameterObject = args[1];
BoundSql boundSql = ms.getBoundSql(parameterObject);
String sql = boundSql.getSql();
MySqlStatementParser mySqlStatementParser = new MySqlStatementParser(sql);
SQLStatement statement = mySqlStatementParser.parseStatement();
SQLExprTableSource sqlTableSource = null;
if(statement instanceof SQLSelectStatement){
SQLSelect selectQuery = ((SQLSelectStatement)statement).getSelect();
MySqlSelectQueryBlock sqlSelectQuery = (MySqlSelectQueryBlock)selectQuery.getQuery();
sqlTableSource = (SQLExprTableSource)sqlSelectQuery.getFrom();
}else if(statement instanceof SQLInsertStatement){
SQLInsertStatement sqlInsertStatement = (SQLInsertStatement)statement;
sqlTableSource = sqlInsertStatement.getTableSource();
}else if(statement instanceof SQLUpdateStatement){
SQLUpdateStatement sqlUpdateStatement = (SQLUpdateStatement)statement;
sqlTableSource = (SQLExprTableSource)sqlUpdateStatement.getTableSource();
}else if(statement instanceof SQLDeleteStatement){
SQLDeleteStatement sqlUpdateStatement = (SQLDeleteStatement)statement;
sqlTableSource = (SQLExprTableSource)sqlUpdateStatement.getTableSource();
}
String tableName = sqlTableSource.toString();
if(tableSet.contains(tableName)){
SQLIdentifierExpr sqlExpr = (SQLIdentifierExpr)sqlTableSource.getExpr();
String newTableName = (String)map.get(tableName);
if(!StringUtils.isEmpty(newTableName) && null != newTableName)
sqlExpr.setName(newTableName);
}
BoundSql bs = new BoundSql(ms.getConfiguration(),statement.toString(),boundSql.getParameterMappings(),parameterObject);
MappedStatement newMs = copyFromMappedStatement(ms, new BoundSqlSqlSource(bs));
for (ParameterMapping mapping : boundSql.getParameterMappings()) {
String prop = mapping.getProperty();
if (boundSql.hasAdditionalParameter(prop)) {
bs.setAdditionalParameter(prop, boundSql.getAdditionalParameter(prop));
}
}
args[0] = newMs;
return invocation.proceed();
}
@Override
public Object plugin(Object target) {
return Plugin.wrap(target, this);
}
@Override
public void setProperties(Properties properties) {}
public Map getMap() {
return map;
}
public void setMap(Map map) {
tableSet = map.keySet();
this.map = map;
}
public boolean isOpenInterceptor() {
return openInterceptor;
}
public void setOpenInterceptor(boolean openInterceptor) {
this.openInterceptor = openInterceptor;
}
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();
}
public static class BoundSqlSqlSource implements SqlSource {
private BoundSql boundSql;
public BoundSqlSqlSource(BoundSql boundSql) {
this.boundSql = boundSql;
}
public BoundSql getBoundSql(Object parameterObject) {
return boundSql;
}
}
}
到此,关于“mybatis插件怎么实现自定义改写表名”的学习就结束了,希望能够解决大家的疑惑。理论与实践的搭配能更好的帮助大家学习,快去试试吧!若想继续学习更多相关知识,请继续关注亿速云网站,小编会继续努力为大家带来更多实用的文章!
亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。