关于java:数据分表Mybatis-Plus动态表名最优方案的探索

61次阅读

共计 3256 个字符，预计需要花费 9 分钟才能阅读完成。

大家在应用 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，咱们只须要实现这个接口，并将这个接口利用配置失效，即可实现动静表名。

须要留神的是：

在 mybatis plus 3.4 版本之前，动静表名处理器接口是ITableNameHandler, 须要配合 mybatis plus 分页插件一起应用能力失效。咱们这里只介绍 3.4 版本之后的实现形式。

在 mybatis plus 3.4.3.2 作废该的形式：dynamicTableNameInnerInterceptor.setTableNameHandlerMap(map); 大家如果见到这种形式实现的动静表名，也是过期的实现办法，新版本中该办法曾经删除。

通过我一段时间的实际总结，我的实现类如下（基于 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 版本，30 余篇精品 docker 文章。字母哥博客：zimug.com

正文完