mybatis-plus 动态表名简易使用

news/2024/10/21 9:59:23/

场景：由于有些表是分表的，需要给表名添加后缀才能正确地访问表，如sys_user_2024_01

代码

依赖版本

        <dependency><groupId>com.baomidou</groupId><artifactId>mybatis-plus-boot-starter</artifactId><version>3.4.3.4</version><scope>compile</scope></dependency><dependency><groupId>com.baomidou</groupId><artifactId>mybatis-plus-extension</artifactId><version>3.4.3.4</version><scope>compile</scope></dependency>

mybatisplus_21">mybatisplus配置类

java">/***  MybatisPlus 配置文件*/
@Configuration
@MapperScan({"com.xxx.dataana.mapper"})
public class MybatisPlusConfig {@Beanpublic MybatisPlusInterceptor mybatisPlusInterceptor() {MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();//动态表名实现DynamicTableNameInnerInterceptor dynamicTableNameInnerInterceptor = new DynamicTableNameInnerInterceptor();//可以传多个表名参数，指定哪些表使用TableNameSuffixHandler处理表名称dynamicTableNameInnerInterceptor.setTableNameHandler(new TableNameSuffixHandler("sys_user"));//以拦截器的方式处理表名称interceptor.addInnerInterceptor(dynamicTableNameInnerInterceptor);return interceptor;}

TableNameSuffixHandler

java">/*** mybatis-plus 成动态表名控制器*/
public class TableNameSuffixHandler implements TableNameHandler {/*** 用于记录哪些表可以使用该动态表名处理器（即哪些表需要分表）*/private final List<String> tableNames;/*** 构造函数，构造动态表名处理器的时候，传递tableNames参数* @param tableNames*/public TableNameSuffixHandler(String... tableNames) {this.tableNames = Arrays.asList(tableNames);}/*** 每个请求线程维护一个suffix数据，避免多线程数据冲突。所以使用ThreadLocal*/private static final ThreadLocal<String> TABLE_SUFFIX = new ThreadLocal<>();/*** 设置请求线程的表名后缀数据* @param tableNameSuffix*/public static void setSuffix(String tableNameSuffix) {TABLE_SUFFIX.set(tableNameSuffix);}/*** 删除请求线程的表名后缀数据* 注： 使用完必须释放*/public static void removeSuffix() {TABLE_SUFFIX.remove();}/*** 动态表名接口实现方法* @param sql* @param tableName* @return*/@Overridepublic String dynamicTableName(String sql, String tableName) {if (this.tableNames.contains(tableName)) {String suffix = TABLE_SUFFIX.get();if(suffix != null){//表名增加后缀return tableName + suffix;}}//表名原样返回return tableName;}
}

使用

java">    public ServiceResponse deleteRelaEventByRelaId(String anaVerId, String relaId) {LambdaUpdateWrapper<SysUser> luw = Wrappers.lambdaUpdate(SysUser.class).set(SysUser::getStatus, "0").set(SysUser::getUpdateTime, new Date()).eq(SysUser::getId, relaId);TableNameSuffixHandler.setSuffix("_2024_01");this.update(luw);TableNameSuffixHandler.removeSuffix();return ServiceResponse.createSuccess();}