MybatisPlus攔截器如何實現(xiàn)數(shù)據(jù)表分表

更新時間：2024年11月06日 09:29:12 作者：樘棣寂寂

為了解決MySQL中大數(shù)據(jù)量的查詢效率問題,采用水平拆分策略,通過取模運算確定表后綴,實現(xiàn)數(shù)據(jù)的有效管理,設計分表時,需利用線程變量存取請求參數(shù),并通過攔截器確定操作的具體表名,從而優(yōu)化數(shù)據(jù)處理性能,此方法適用于業(yè)務表數(shù)據(jù)量大或快速增長的場景

MybatisPlus攔截器實現(xiàn)數(shù)據(jù)表分表

很多項目都會存在一些數(shù)據(jù)量很大或者數(shù)據(jù)量增加很快的業(yè)務表，由于mysql的數(shù)據(jù)量達到一定量后會影響我們的查詢效率，為了避免該類問題發(fā)生，我們需要在項目前期設計的時候針對這兩類情況做一個分表的設計。

這里的分表指的是水平拆分（只是表名不同，其余字段都一致，主鍵id不允許重復），對某一個數(shù)字取模運算做為拆分后表的后綴名，具體要分多少張表可通過自己實際的項目情況確定。

首先創(chuàng)建請求參數(shù)傳遞的一個輔助類

/**
 * 請求參數(shù)傳遞輔助類
 */
public class RequestDataHelper {
    /**
     * 請求參數(shù)存取
     */
    private static final ThreadLocal<Map<String, Object>> REQUEST_DATA = new ThreadLocal<>();

    /**
     * 設置請求參數(shù)
     *
     * @param requestData 請求參數(shù) MAP 對象
     */
    public static void setRequestData (Map<String, Object> requestData) {
        REQUEST_DATA.set(requestData);
    }

    /**
     * 獲取請求參數(shù)
     *
     * @return 請求參數(shù) MAP 對象
     */
    public static Map<String, Object> getRequestData () {
        return REQUEST_DATA.get();
    }
}

該輔助類主要作用有：

a.設置請求參數(shù)
b.將請求參數(shù)存取到線程變量中
c.獲取請求參數(shù)

簡單的說就是在涉及到需要分表的數(shù)據(jù)操作時，將請求參數(shù)放入線程變量中。

然后在攔截器里面獲取這個參數(shù)，做特定的處理，找到我們具體要操作的那張表。

這里的請求參數(shù)大家可以把它理解成跟我們分表后的表名產生關聯(lián)的數(shù)據(jù)。

這里面的線程變量是僅在當前線程下可使用的數(shù)據(jù)，與其他線程做隔離。

在本文中不做詳細解釋。

在涉及到分表的數(shù)據(jù)層操作前（Mybatisplus或者Mybatis增刪改查數(shù)據(jù)前）

將請求參數(shù)放入線程變量

RequestDataHelper.setRequestData(Collections.singletonMap("studentId", param.getStudentId()));
        LambdaQueryWrapper<StudentRecordEntity> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(StudentRecordEntity::getStudentId, param.getStudentId());
        List<StudentRecordEntity> studentRecordEntityList = super.list(queryWrapper);

Mybatisplus 攔截器代碼

/**
     * Mybatis plus 攔截器
     */
    @Bean
    public MybatisPlusInterceptor mybatisPlusInterceptor() {
        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();

        //動態(tài)表插件
        DynamicTableNameInnerInterceptor dynamicTableNameInnerInterceptor = new DynamicTableNameInnerInterceptor();
        dynamicTableNameInnerInterceptor.setTableNameHandler((sql, tableName) -> {
            // 獲取參數(shù)方法
            if (tableName.equalsIgnoreCase("t_student_record")) {
                Map<String, Object> paramMap = RequestDataHelper.getRequestData();
                long studentId= Long.parseLong(String.valueOf(paramMap.get("studentId")));
                int mod = (int) (studentId% 16);
                return tableName + "_" + mod;

            } else {
                return tableName;
            }
        });
        interceptor.addInnerInterceptor(dynamicTableNameInnerInterceptor);
        return interceptor;
    }

這里面t_student_record表就是要拆分的表，如果有多個，在if里面寫多個。

studentId就是調用的時候傳入線程變量的請求參數(shù)，對它%16就是分了16張表，根據(jù)實際業(yè)務情況，需要分多少張就把%后面的數(shù)字改為多少。