1. 程式人生 > 實用技巧 >mybatis plus 動態表名 SQL 解析【我】

mybatis plus 動態表名 SQL 解析【我】

轉:

mybatis plus 動態表名 SQL 解析器

說明:

其中 put 後面的 “User”就是bean的@TableName("User")註解中配的表名,這個表不一定在資料庫存在,實際執行sql時是在dynamicTableName方法中返回的表名。另外下面這段程式碼可以配多行,代表對不同的表名進行過濾。

        tableNameHandlerMap.put("User", new ITableNameHandler() {
            @Override
            public String dynamicTableName(MetaObject metaObject, String sql, String tableName) {
                
return myTableName.get();//返回null不會替換 注意 多租戶過濾會將它一塊過濾不會替換@SqlParser(filter=true) 可不會替換 } });

比如再加一段,表示對sys_user2過濾:

        tableNameHandlerMap.put("sys_user2",new ITableNameHandler(){
            @Override
            public String dynamicTableName(MetaObject metaObject, String sql, String tableName) {
                
// 自定義表名規則,或者從配置檔案、request上下文中讀取 // 假設這裡的使用者表根據年份來進行分表操作 Date date = new Date(); String year = String.format("%tY", date); // 返回最後需要操作的表名,sys_user_2019 return "sys_user_" + year; } });

eg:

import com.baomidou.mybatisplus.core.parser.ISqlParser;
import com.baomidou.mybatisplus.extension.parsers.DynamicTableNameParser; import com.baomidou.mybatisplus.extension.parsers.ITableNameHandler; import com.baomidou.mybatisplus.extension.plugins.PaginationInterceptor; import org.apache.ibatis.reflection.MetaObject; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; import java.util.*; /** * @Description myBatisPlus的配置類 * @createTime 2020年06月28日 15:39:00 */ @Configuration public class MyBatisPlusConfig { public static ThreadLocal<String> myTableName = new ThreadLocal<>(); @Bean public PaginationInterceptor paginationInterceptor() { PaginationInterceptor paginationInterceptor = new PaginationInterceptor(); //分頁查詢上限重新賦值 paginationInterceptor.setLimit(100000); // 建立SQL解析器集合 List<ISqlParser> sqlParserList = new ArrayList<>(); // 動態表名SQL解析器 DynamicTableNameParser dynamicTableNameParser = new DynamicTableNameParser(); Map<String, ITableNameHandler> tableNameHandlerMap = new HashMap<>(); // Map的key就是需要替換的原始表名 tableNameHandlerMap.put("t_a_dc_check_xxx1", new ITableNameHandler() { @Override public String dynamicTableName(MetaObject metaObject, String sql, String tableName) { // 自定義表名規則,或者從配置檔案、request上下文中讀取 // 假設這裡的使用者表根據年份來進行分表操作 Date date = new Date(); //DateTimeFormatter dateTimeFormatter = new DateTimeFormatter(LocalDateTime.now()); String year = String.format("_%tY%tm", date,date); // 返回最後需要操作的表名,table_name_YYYYMM月表 return "t_a_dc_check_xxx1" + year; } }); tableNameHandlerMap.put("t_a_dc_check_xxx2", new ITableNameHandler() { @Override public String dynamicTableName(MetaObject metaObject, String sql, String tableName) { // 自定義表名規則,或者從配置檔案、request上下文中讀取 // 假設這裡的使用者表根據年份來進行分表操作 Date date = new Date(); //DateTimeFormatter dateTimeFormatter = new DateTimeFormatter(LocalDateTime.now()); String year = String.format("_%tY%tm", date,date); // 返回最後需要操作的表名,sys_user_2019 return "t_a_dc_check_xxx2" + year; } }); //增加一個銷賬月表表名過濾器 tableNameHandlerMap.put("dt_xxx_item_month", new ITableNameHandler() { @Override public String dynamicTableName(MetaObject metaObject, String sql, String tableName) { return myTableName.get();//返回null不會替換 注意 多租戶過濾會將它一塊過濾不會替換@SqlParser(filter=true) 可不會替換 } }); dynamicTableNameParser.setTableNameHandlerMap(tableNameHandlerMap); sqlParserList.add(dynamicTableNameParser); paginationInterceptor.setSqlParserList(sqlParserList); return paginationInterceptor; } }

使用時直接在查詢前加上設定表名即可:

   QueryWrapper<ItemMonth> itemMonthQueryWrapper = new QueryWrapper<>();
            itemMonthQueryWrapper.eq("BILL_ID",billId);
            MyBatisPlusConfig.myTableName.set("user_1234");
            List<AcctItemMonth> list = itemMonthService.list(itemMonthQueryWrapper);