mybatis的使用及源码分析(十五) Mybatis-Plus动态表名

在某些情况下，Mybatis-Plus需要使用同一实体对应多个表进行操作，比如按照时间分表的一些操作

Mybatis-Plus使用3.3.1以上的版本

1、定义一个表名处理类，使用ThreadLocal实现

/**
 * 动态表名处理
 * @author zhuquanwen
 * @version 1.0
 * @date 2022/4/11 13:28
 * @since jdk11
 */
public class DynamicTableNameHolder {
    private static final ThreadLocal<String> TABLE_NAME_HOLDER = new ThreadLocal<>();

    public static String get() {
        return TABLE_NAME_HOLDER.get();
    }

    public static void set(String tableName) {
        TABLE_NAME_HOLDER.set(tableName);
    }

    public static void remove() {
        TABLE_NAME_HOLDER.remove();
    }
}

2、注册一个MybatisPlus的动态表名拦截器，并使用上面定义的类获取表名

@Bean
public DynamicTableNameInnerInterceptor getDynamicTableNameInterceptor() {
        DynamicTableNameInnerInterceptor dynamicTableNameInnerInterceptor = new DynamicTableNameInnerInterceptor();
        dynamicTableNameInnerInterceptor.setTableNameHandler((sql, tableName) -> {
            String tableName1 = DynamicTableNameHolder.get();
            return StringUtils.isNotEmpty(tableName1) ? tableName1 : tableName;
        });
        return dynamicTableNameInnerInterceptor;
}

3、测试使用
记得使用后要调用remove方式释放

    /**
     * 测试自定义表名
     * */
    @GetMapping("/dynamic-tablename")
    public ResponseEntity dynamicTableName() {
        ResponseEntity response = getResponse();
        try {
            DynamicTableNameHolder.set("test_copy1");
            List<Test> tests = testMapper.selectList(null);
            response.setValue(tests);
        } finally {
            DynamicTableNameHolder.remove();
        }
        return response;
    }