資料分表Mybatis Plus動態表名最優方案的探索
- 一、應用場景
- 二、動態表名處理器介面實現
- 三、測試實現效果
一、應用場景
大家在使用Mybatis進行開發的時候,經常會遇到一種情況:按照月份month將資料放在不同的表裡面,查詢資料的時候需要跟不同的月份month去查詢不同的表。
但是我們都知道,Mybatis是ORM持久層框架,即:實體關係對映,實體Object與資料庫表之間是存在一一對應的對映關係。比如:
@Data public class Student { private Integer id; private String stuName; private Integer age; }
表結構
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='學生表' COLLATE='utf8_general_ci' ENGINE=InnoDB ;
Student 實體類與student表是一一對應的關係,如果我們希望將學員表按照月份進行分表,比如:student_202206、student_202207、student_202208,即產生了 「一個實體類及其Mapper需要操作多個數據庫分月表,這種情況在Mybatis plus下我們該如何操作資料呢?」 其實方法有很多,我將我實踐中總結出的最優方案給大家進行說明。
二、動態表名處理器介面實現
為了處理上述類似的問題,mybatis plus提供了動態表名處理器介面 TableNameHandler
,我們只需要實現這個介面,並將這個介面應用配置生效,即可實現動態表名。
需要注意的是:
ITableNameHandler
經過我一段時間的實踐總結,我的實現類如下(基於mybatis plus 3.4.3.2之後的版本):
import com.baomidou.mybatisplus.extension.plugins.handler.TableNameHandler; import java.util.Arrays; import java.util.List; /** * 按月份引數,組成動態表名 */ public class MonthTableNameHandler implements TableNameHandler { //用於記錄哪些表可以使用該月份動態表名處理器(即哪些表按月分表) private List<String> tableNames; //建構函式,構造動態表名處理器的時候,傳遞tableNames引數 public MonthTableNameHandler(String ...tableNames) { this.tableNames = Arrays.asList(tableNames); } //每個請求執行緒維護一個month資料,避免多執行緒資料衝突。所以使用ThreadLocal private static final ThreadLocal<String> MONTH_DATA = new ThreadLocal<>(); //設定請求執行緒的month資料 public static void setData(String month) { MONTH_DATA.set(month); } //刪除當前請求執行緒的month資料 public static void removeData() { MONTH_DATA.remove(); } //動態表名介面實現方法 @Override public String dynamicTableName(String sql, String tableName) { if (this.tableNames.contains(tableName)){ return tableName + "_" + MONTH_DATA.get(); //表名增加月份字尾 }else{ return tableName; //表名原樣返回 } } }
大家先對上面的程式碼有一個基礎瞭解,看了下面的測試過程,再回頭看上面的程式碼中的註釋,就比較好理解了。表名處理器寫好了之後,我們要讓它生效,還需要做如下的配置。配置內容照葫蘆畫瓢就可以了。需要關注的部分,我都已經給大家添加了註釋。
@Configuration @MapperScan("com.zimug") public class MybatisPlusConfig { @Bean public MybatisPlusInterceptor mybatisPlusInterceptor() { MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor(); DynamicTableNameInnerInterceptor dynamicTableNameInnerInterceptor = new DynamicTableNameInnerInterceptor(); dynamicTableNameInnerInterceptor.setTableNameHandler( //可以傳多個表名引數,指定哪些表使用MonthTableNameHandler處理表名稱 new MonthTableNameHandler("student","teacher") ); //以攔截器的方式處理表名稱 interceptor.addInnerInterceptor(dynamicTableNameInnerInterceptor); //可以傳遞多個攔截器,即:可以傳遞多個表名處理器TableNameHandler //interceptor.addInnerInterceptor(dynamicTableNameInnerInterceptor); return interceptor; } }
三、測試實現效果
首先建立一個StudentMapper ,預設情況下StudentMapper 只能操作student表,不能操作student_YYYYMM表。
@Mapper public interface StudentMapper extends BaseMapper<Student> {}
下面我們來寫一個單元測試用例,該測試用例test函式模擬一次請求訪問的Controller或者service函式。
@SpringBootTest class DynamicTableNameTest { @Resource private StudentMapper studentMapper; @Test void test() { //執行資料操作之前設定月份(實際場景下該引數從請求引數中解析) MonthTableNameHandler.setData("202208"); studentMapper.selectById(1); //以id=2查詢student_202208這張表 //閱後即焚,將ThreadLocal當前請求執行緒的資料移除 MonthTableNameHandler.removeData(); } }
當我們執行這個單元測試用例的時候,我們發現控制檯打印出如下資訊,注意看SQL的部分,真的是去查詢student_202208這張表了,而不是student表。這說明我們的動態表名實現是成功的。
「碼文不易,如果您覺得有幫助,請幫忙點選在看或者分享,沒有您的支援我可能無法堅持下去!」歡迎關注公眾號:字母哥雜談,回覆003贈送本文所在專欄《docker修煉之道》的PDF成書版本。
- 吳恩達來信:人工智慧領域的求職小 tips
- EasyCV帶你復現更好更快的自監督演算法-FastConvMAE
- 某車聯網App 通訊協議加密分析(四) Trace Code
- 帶你瞭解CANN的目標檢測與識別一站式方案
- EasyNLP玩轉文字摘要(新聞標題)生成
- PostgreSQL邏輯複製解密
- 基於 CoreDNS 和 K8s 構建雲原生場景下的企業級 DNS
- 迴圈神經網路(RNN)可是在語音識別、自然語言處理等其他領域中引起了變革!
- 技術分享| 分散式系統中服務註冊發現元件的原理及比較
- 利用谷歌地圖採集外貿客戶的電話和手機號碼
- 跟我學Python影象處理丨關於影象金字塔的影象向下取樣和向上取樣
- 帶你掌握如何使用CANN 運算元ST測試工具msopst
- 一招教你如何高效批量匯入與更新資料
- 一步步搞懂MySQL元資料鎖(MDL)
- 你知道如何用 PHP 實現多程序嗎?
- KubeSphere 閘道器的設計與實現(解讀)
- 京東金融客戶端使用者觸達方式的探索與實踐
- 如何使用 Junit Mockito 實踐單元測試
- 演算法基礎(二)| 高精度演算法詳解
- 一文帶你體驗MRS HetuEngine如何實現跨源跨域分析