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