mybatis插件怎么实现自定义改写表名

这篇文章主要介绍“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插件怎么实现自定义改写表名”的学习就结束了，希望能够解决大家的疑惑。理论与实践的搭配能更好的帮助大家学习，快去试试吧！若想继续学习更多相关知识，请继续关注亿速云网站，小编会继续努力为大家带来更多实用的文章！