MyBatis-Plus中如何實(shí)現(xiàn)動(dòng)態(tài)表名

更新時(shí)間：2023年07月14日 14:56:05 作者：DAIHAO丶

這篇文章主要介紹了MyBatis-Plus中如何實(shí)現(xiàn)動(dòng)態(tài)表名問(wèn)題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教

MyBatis-Plus實(shí)現(xiàn)動(dòng)態(tài)表名

MyBatis-Plus版本

1、添加MyBatis-Plus依賴

<dependency>
?? ?<groupId>com.baomidou</groupId>
?? ?<artifactId>mybatis-plus-boot-starter</artifactId>
?? ?<version>3.5.1</version>
</dependency>

MyBatis-Plus配置

2、添加MyBatis-Plus配置，利用攔截器獲取到表名給替換

@Configuration
public class MybatisPlusConfig {
? ? static List<String> tableList(){
? ? ? ? List<String> tables = new ArrayList<>();
? ? ? ? //表名
? ? ? ? tables.add("TestUser");
? ? ? ? return tables;
? ? }
?? ?//攔截器，獲取到表名給替換
? ? @Bean
? ? public MybatisPlusInterceptor dynamicTableNameInnerInterceptor() {
? ? ? ? MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
? ? ? ? DynamicTableNameInnerInterceptor dynamicTableNameInnerInterceptor = new DynamicTableNameInnerInterceptor();
? ? ? ? dynamicTableNameInnerInterceptor.setTableNameHandler((sql, tableName) -> {
? ? ? ? ? ? String newTable = null;
? ? ? ? ? ? for (String table : tableList()) {
? ? ? ? ? ? ? ? newTable = RequestDataHelper.getRequestData(table);
? ? ? ? ? ? ? ? if (table.equals(tableName) && newTable!=null){
? ? ? ? ? ? ? ? ? ? tableName = newTable;
? ? ? ? ? ? ? ? ? ? break;
? ? ? ? ? ? ? ? }
? ? ? ? ? ? }
? ? ? ? ? ? return tableName;
? ? ? ? });
? ? ? ? interceptor.addInnerInterceptor(dynamicTableNameInnerInterceptor);
? ? ? ? return interceptor;
? ? }
}

請(qǐng)求參數(shù)傳遞輔助類

3、創(chuàng)建請(qǐng)求參數(shù)傳遞輔助類

public class RequestDataHelper {
? ? /**
? ? ?* 請(qǐng)求參數(shù)存取
? ? ?*/
? ? private static final ThreadLocal<Map<String, Object>> REQUEST_DATA = new ThreadLocal<>();
? ? /**
? ? ?* 設(shè)置請(qǐng)求參數(shù)
? ? ?*
? ? ?* @param requestData 請(qǐng)求參數(shù) MAP 對(duì)象
? ? ?*/
? ? public static void setRequestData(Map<String, Object> requestData) {
? ? ? ? REQUEST_DATA.set(requestData);
? ? }
? ? /**
? ? ?* 獲取請(qǐng)求參數(shù)
? ? ?*
? ? ?* @param param 請(qǐng)求參數(shù)
? ? ?* @return 請(qǐng)求參數(shù) MAP 對(duì)象
? ? ?*/
? ? public static <T> T getRequestData(String param) {
? ? ? ? Map<String, Object> dataMap = getRequestData();
? ? ? ? if (CollectionUtils.isNotEmpty(dataMap)) {
? ? ? ? ? ? return (T) dataMap.get(param);
? ? ? ? }
? ? ? ? return null;
? ? }
? ? /**
? ? ?* 獲取請(qǐng)求參數(shù)
? ? ?*
? ? ?* @return 請(qǐng)求參數(shù) MAP 對(duì)象
? ? ?*/
? ? public static Map<String, Object> getRequestData() {
? ? ? ? return REQUEST_DATA.get();
? ? }
}

使用

4、在程序中使用，注意如果實(shí)際表名與實(shí)體類與不同，可先在實(shí)體類類注明表名@TableName("TestUser")

@Test
public void test1() {
? ? RequestDataHelper.setRequestData(new HashMap<String, Object>() {{
? ? ? ? put("TestUser", "TestUser_" + "2022_05");
? ? }});
?? ?//表名會(huì)被動(dòng)態(tài)替換成"TestUser_2022_05"
?? ?//如果實(shí)際表名與實(shí)體類與不同，可先在實(shí)體類類注明表名@TableName("TestUser")
? ? List<UserEntity> users= userDao.selectList(null);
? ? System.out.println(users);
}

數(shù)據(jù)分表Mybatis Plus動(dòng)態(tài)表名最優(yōu)方案

應(yīng)用場(chǎng)景

大家在使用Mybatis進(jìn)行開發(fā)的時(shí)候，經(jīng)常會(huì)遇到一種情況：按照月份month將數(shù)據(jù)放在不同的表里面，查詢數(shù)據(jù)的時(shí)候需要跟不同的月份month去查詢不同的表。

但是我們都知道，Mybatis是ORM持久層框架，即：實(shí)體關(guān)系映射，實(shí)體Object與數(shù)據(jù)庫(kù)表之間是存在一一對(duì)應(yīng)的映射關(guān)系。比如：

@Data
public class Student {
    private Integer id;
    private String stuName;
    private Integer age;
}

表結(jié)構(gòu)

CREATE TABLE `student` (
    `id` INT(11) NOT NULL AUTO_INCREMENT,
    `stu_name` VARCHAR(64) NOT NULL DEFAULT '0' COMMENT '姓名',
    `age` INT(11) NOT NULL COMMENT '年齡',
    PRIMARY KEY (`id`)
)
COMMENT='學(xué)生表'
COLLATE='utf8_general_ci'
ENGINE=InnoDB
;

Student 實(shí)體類與student表是一一對(duì)應(yīng)的關(guān)系，如果我們希望將學(xué)員表按照月份進(jìn)行分表，比如：student_202206、student_202207、student_202208，即產(chǎn)生了**「一個(gè)實(shí)體類及其Mapper需要操作多個(gè)數(shù)據(jù)庫(kù)分月表，這種情況在Mybatis plus下我們?cè)撊绾尾僮鲾?shù)據(jù)呢？」** 其實(shí)方法有很多，我將我實(shí)踐中總結(jié)出的最優(yōu)方案給大家進(jìn)行說(shuō)明。

動(dòng)態(tài)表名處理器接口實(shí)現(xiàn)

為了處理上述類似的問(wèn)題，mybatis plus提供了動(dòng)態(tài)表名處理器接口TableNameHandler，我們只需要實(shí)現(xiàn)這個(gè)接口，并將這個(gè)接口應(yīng)用配置生效，即可實(shí)現(xiàn)動(dòng)態(tài)表名。

需要注意的是：

在mybatis plus 3.4版本之前，動(dòng)態(tài)表名處理器接口是ITableNameHandler, 需要配合mybatis plus分頁(yè)插件一起使用才能生效。我們這里只介紹3.4版本之后的實(shí)現(xiàn)方式。
在mybatis plus 3.4.3.2 作廢該的方式：dynamicTableNameInnerInterceptor.setTableNameHandlerMap(map); 大家如果見到這種方式實(shí)現(xiàn)的動(dòng)態(tài)表名，也是過(guò)時(shí)的實(shí)現(xiàn)方法，新版本中該方法已經(jīng)刪除。

經(jīng)過(guò)我一段時(shí)間的實(shí)踐總結(jié)，我的實(shí)現(xiàn)類如下（基于mybatis plus 3.4.3.2之后的版本）：

import com.baomidou.mybatisplus.extension.plugins.handler.TableNameHandler;
import java.util.Arrays;
import java.util.List;
/**
 * 按月份參數(shù)，組成動(dòng)態(tài)表名
 */
public class MonthTableNameHandler implements TableNameHandler {
    //用于記錄哪些表可以使用該月份動(dòng)態(tài)表名處理器（即哪些表按月分表）
    private List<String> tableNames;
    //構(gòu)造函數(shù)，構(gòu)造動(dòng)態(tài)表名處理器的時(shí)候，傳遞tableNames參數(shù)
    public MonthTableNameHandler(String ...tableNames) {
        this.tableNames = Arrays.asList(tableNames);
    }
    //每個(gè)請(qǐng)求線程維護(hù)一個(gè)month數(shù)據(jù)，避免多線程數(shù)據(jù)沖突。所以使用ThreadLocal
    private static final ThreadLocal<String> MONTH_DATA = new ThreadLocal<>();
    //設(shè)置請(qǐng)求線程的month數(shù)據(jù)
    public static void setData(String month) {
        MONTH_DATA.set(month);
    }
    //刪除當(dāng)前請(qǐng)求線程的month數(shù)據(jù)
    public static void removeData() {
        MONTH_DATA.remove();
    }
    //動(dòng)態(tài)表名接口實(shí)現(xiàn)方法
    @Override
    public String dynamicTableName(String sql, String tableName) {
        if (this.tableNames.contains(tableName)){
            return tableName + "_" + MONTH_DATA.get();  //表名增加月份后綴
        }else{
            return tableName;   //表名原樣返回
        }
    }
}

大家先對(duì)上面的代碼有一個(gè)基礎(chǔ)了解，看了下面的測(cè)試過(guò)程，再回頭看上面的代碼中的注釋，就比較好理解了。表名處理器寫好了之后，我們要讓它生效，還需要做如下的配置。配置內(nèi)容照葫蘆畫瓢就可以了。需要關(guān)注的部分，我都已經(jīng)給大家添加了注釋。

@Configuration
@MapperScan("com.zimug")
public class MybatisPlusConfig {
    @Bean
    public MybatisPlusInterceptor mybatisPlusInterceptor() {
        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
        DynamicTableNameInnerInterceptor dynamicTableNameInnerInterceptor = new DynamicTableNameInnerInterceptor();
        dynamicTableNameInnerInterceptor.setTableNameHandler(
                //可以傳多個(gè)表名參數(shù)，指定哪些表使用MonthTableNameHandler處理表名稱
                new MonthTableNameHandler("student","teacher") 
        );
        //以攔截器的方式處理表名稱
        interceptor.addInnerInterceptor(dynamicTableNameInnerInterceptor);
        //可以傳遞多個(gè)攔截器，即：可以傳遞多個(gè)表名處理器TableNameHandler
        //interceptor.addInnerInterceptor(dynamicTableNameInnerInterceptor);
        return interceptor;
    }
}

測(cè)試實(shí)現(xiàn)效果

首先創(chuàng)建一個(gè)StudentMapper ，默認(rèn)情況下StudentMapper 只能操作student表，不能操作student_YYYYMM表。

@Mapper
public interface StudentMapper extends BaseMapper<Student> {}

下面我們來(lái)寫一個(gè)單元測(cè)試用例，該測(cè)試用例test函數(shù)模擬一次請(qǐng)求訪問(wèn)的Controller或者service函數(shù)。

@SpringBootTest
class DynamicTableNameTest {
    @Resource
    private StudentMapper studentMapper;
    @Test
    void test() {
        //執(zhí)行數(shù)據(jù)操作之前設(shè)置月份（實(shí)際場(chǎng)景下該參數(shù)從請(qǐng)求參數(shù)中解析）
        MonthTableNameHandler.setData("202208");
        studentMapper.selectById(1); //以id=2查詢student_202208這張表
        //閱后即焚，將ThreadLocal當(dāng)前請(qǐng)求線程的數(shù)據(jù)移除
        MonthTableNameHandler.removeData();
    }
}

當(dāng)我們執(zhí)行這個(gè)單元測(cè)試用例的時(shí)候，我們發(fā)現(xiàn)控制臺(tái)打印出如下信息，注意看SQL的部分，真的是去查詢student_202208這張表了，而不是student表。這說(shuō)明我們的動(dòng)態(tài)表名實(shí)現(xiàn)是成功的。