mybatis-plus動態表名實現方法

2022-02-10 13:01:08

1.使用場景

一個mybatis entity 對應多張表（表明不同的表–> 多張表結構一致只有表名稱不同），在使用時，可以動態對映表名稱。
比如：按照時間分表，某些業務冷熱資料分離後資料存在不同的表中等

2.一定要時常注意工具的更新，不吃虧

因為工作需要，最近需要實現一個類似如下業務等功能互動：

簡單的描述就是：有一部分的表資料，採用了外部同步的模式，同步到了doris中，在業務中採用雙資料來源的方式，資料寫在mysql，讀走doris （此處不考慮資料同步的延遲問題 ，這是個大問題，但我們這篇文章先不考慮這個問題）。

要實現上面的方案，需要解決以下幾個問題：

雙資料來源的問題doris/mysql中庫和表的名稱不一致（例如：上圖綠色打底的表，在mysql和doris中表的結構一致，但在mysql中庫.表名為： databaseA.tableA ，在doris中則為 databaseB.ods_tableB_s）

在之前專案研究中，因為有自己去自定義dao層mapper的方法，mybatis表中的方法對應的sql在專案啟動時就已經生成好了對應的sql模板，在程式執行中，只是用引數代替即可。

例如上圖，因為我之前用的mybatis-plus 的版本比較低，在生成模板時實際上就已經確定了表的名稱。這就導致了無法在使用的時候重新替換。為了實現這個功能，自己實現的時候也是一直往我如何在生成模板語句的時候就直接把我想要的表名替換替換進去。在實現的差不多的時候，同事來了一句，mybatis-plus不是支援的嗎？ hhhh，所以說，還是要去關注一下你在專案中所用到的jar包的迭代更新哦，會發現很多，然後開始偷懶。

3.對mybatis-plus進行升級

我把版本從：3.3.0 升級到了 3.4.3.2
啟動專案的時候發現服務啟動失敗了，著急嘿～

然後在解決問題的過程中發現時因為jar包衝突了。
然後發現了idea中一個很好使用的工具【在pom.xml檔案中要，右擊滑鼠進入到Maven -> show dependence 】可以很直觀的看見衝突法jar包，然後去解決對應的衝突就可以了（這裡不多說了，給大家一個學習連結傳送門：https://blog.csdn.net/daerzei/article/details/82344569）

4.將原來的mybatis的外掛設定調整成新版本的，並新增上動態表名的模組

4.1 調整mybatis plus 設定

    @Bean
    public MybatisPlusInterceptor mybatisPlusInterceptor() {
        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
        interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.MYSQL));

        DynamicTableNameInnerInterceptor dynamicTableNameInnerInterceptor = new DynamicTableNameInnerInterceptor();
        //新增上自己實現的表名處理器DayRecordTableNameHandler
        dynamicTableNameInnerInterceptor.setTableNameHandler(new DayRecordTableNameHandler());
        interceptor.addInnerInterceptor(dynamicTableNameInnerInterceptor);

        return interceptor;
    }

4.2 自定義表名稱處理器

結合threadlocal 來確定對對應的業務邏輯使用什麼表名稱（我這裡是利用將資料來源資訊放入到threadlocal中，然後來對映不同的表名稱），下面的程式碼範例是一個簡化版：

public class DayRecordTableNameHandler implements TableNameHandler {
    private static final Logger log = LoggerFactory.getLogger(DayRecordTableNameHandler.class);

    public DayRecordTableNameHandler(){
    }

    @Override
    public String dynamicTableName(String sql, String tableName) {
        Object dataSource = HnThreadLocal.get(SystemConstants.DATASOURCE);
        if(Objects.isNull(dataSource)){
            return tableName;
        }
        if(Objects.equals(tableName,"zsh_day_record")){
            int dataSourceIntValue = (int)dataSource;
            if(Objects.equals(dataSourceIntValue,1)){
                log.info("表名替換:"+tableName+"--->sql:ods_zsh_day_record_s");
                return "ods_zsh_day_record_s";
            }
        }
        return tableName;
    }

}

4.3 利用AOP切換資料來源

這裡想省略了，多資料來源用的是dynamic-datasource-spring-boot-starter
AOP監聽多是 @DS ，依據註解value來判斷資料來源是不是走doris