一.Configuration
@Configuration public class MybatisPlusConfiguration { public static ThreadLocal<String> inputTableName = new ThreadLocal<>(); @Bean public PaginationInterceptor paginationInterceptor(){ PaginationInterceptor paginationInterceptor = new PaginationInterceptor(); List<ISqlParser> sqlParserList = new ArrayList<>(); DynamicTableNameParser dynamicTableNameParser = new DynamicTableNameParser(); HashMap<String, ITableNameHandler> map = new HashMap<>(); map.put("user", (metaObject, sql, tableName) -> inputTableName.get()); dynamicTableNameParser.setTableNameHandlerMap(map); sqlParserList.add(dynamicTableNameParser); paginationInterceptor.setSqlParserList(sqlParserList); paginationInterceptor.setSqlParserFilter(metaObject -> { MappedStatement statement = SqlParserHelper.getMappedStatement(metaObject);
//如果是selectById方法,則表名不會被替換,其他的會被動態替換表名 return "com.mp.dao.UserMapper.selectById".equals(statement.getId()); }); return paginationInterceptor; } }
二.測試
傳入參數
@Test public void testTableNAme(){ MybatisPlusConfiguration.inputTableName.set("user_2020"); userMapper.selectList(null); }
三.運行結果
可以觀察到查詢的表名已經被替換