MybatisPlus拦截器实现数据表分表

最新推荐文章于 2025-03-03 16:08:41 发布

樘棣寂寂

最新推荐文章于 2025-03-03 16:08:41 发布

阅读量1k

点赞数 16

文章标签：数据库

本文链接：https://blog.youkuaiyun.com/m0_66389650/article/details/135673318

版权

文章介绍了如何在项目中通过Mybatisplus的拦截器，利用请求参数和线程变量实现数据表的水平分表，以提高查询效率。通过取模运算确定分表数量，如学生记录表按studentId拆分为16张子表。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

很多项目都会存在一些数据量很大或者数据量增加很快的业务表，由于mysql的数据量达到一定量后会影响我们的查询效率，为了避免该类问题发生，我们需要在项目前期设计的时候针对这两类情况做一个分表的设计。这里的分表指的是水平拆分（只是表名不同，其余字段都一致，主键id不允许重复），对某一个数字取模运算做为拆分后表的后缀名，具体要分多少张表可通过自己实际的项目情况确定。

首先创建请求参数传递的一个辅助类

/**
 * 请求参数传递辅助类
 */
public class RequestDataHelper {
    /**
     * 请求参数存取
     */
    private static final ThreadLocal<Map<String, Object>> REQUEST_DATA = new ThreadLocal<>();

    /**
     * 设置请求参数
     *
     * @param requestData 请求参数 MAP 对象
     */
    public static void setRequestData (Map<String, Object> requestData) {
        REQUEST_DATA.set(requestData);
    }

    /**
     * 获取请求参数
     *
     * @return 请求参数 MAP 对象
     */
    public static Map<String, Object> getRequestData () {
        return REQUEST_DATA.get();
    }
}

该辅助类主要作用有：

        a.设置请求参数
        b.将请求参数存取到线程变量中
        c.获取请求参数

简单的说就是在涉及到需要分表的数据操作时，将请求参数放入线程变量中。然后在拦截器里面获取这个参数，做特定的处理，找到我们具体要操作的那张表。这里的请求参数大家可以把它理解成跟我们分表后的表名产生关联的数据。这里面的线程变量是仅在当前线程下可使用的数据，与其他线程做隔离。在本文中不做详细解释。

在涉及到分表的数据层操作前（Mybatisplus或者Mybatis增删改查数据前），将请求参数放入线程变量

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();

        //动态表插件
        DynamicTableNameInnerInterceptor dynamicTableNameInnerInterceptor = new DynamicTableNameInnerInterceptor();
        dynamicTableNameInnerInterceptor.setTableNameHandler((sql, tableName) -> {
            // 获取参数方法
            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就是调用的时候传入线程变量的请求参数，对它%16就是分了16张表，根据实际业务情况，需要分多少张就把%后面的数字改为多少。