MyBatis-Plus使用動態(tài)表名分表查詢的實現(xiàn)

更新時間：2025年11月28日 10:56:19 作者：彭于晏Yan

本文主要介紹了MyBatis-Plus使用動態(tài)表名分表查詢,主要是動態(tài)修改表名的幾種常見場景,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨著小編來一起學習學習吧

分表策略：根據不同的條件（如年份、月份、區(qū)域等）來選擇查詢不同的表。
多租戶系統(tǒng)：根據租戶的不同，查詢不同的表。
歷史數(shù)據管理：查詢歷史表的數(shù)據時，動態(tài)決定表名（例如每年有一個對應的表，table_2021, table_2022 等）。

1. 引入依賴

 <dependency>
     <groupId>com.baomidou</groupId>
     <artifactId>mybatis-plus-boot-starter</artifactId>
     <version>3.5.3</version>   
 </dependency>

注意：版本須使用3.5.3以上，否則無法使用DynamicTableNameInnerInterceptor類中的setTableNameHandler方法

2. mybatis-plus配置

@Configuration
@Slf4j
public class MybatisPlusConfig {

    @Bean
    public MybatisPlusInterceptor mybatisPlusInterceptor() {
        MybatisPlusInterceptor mybatisPlusInterceptor = new MybatisPlusInterceptor();
        // 數(shù)據權限
        //mybatisPlusInterceptor.addInnerInterceptor(new DataFilterInterceptor());
        // 樂觀鎖
        mybatisPlusInterceptor.addInnerInterceptor(new OptimisticLockerInnerInterceptor());
        // 防止全表更新與刪除
        mybatisPlusInterceptor.addInnerInterceptor(new BlockAttackInnerInterceptor());
        //動態(tài)表名
        DynamicTableNameInnerInterceptor dynamicTableNameInnerInterceptor = new DynamicTableNameInnerInterceptor();
        dynamicTableNameInnerInterceptor.setTableNameHandler((sql, tableName) -> { //匿名內部類
            String tenant = TenantContext.TENANT_DATA.get();
            // 如果不為空，使用動態(tài)表名。
            if (StringUtils.isNotBlank(tenant) && TenantContext.containsTableName(tableName)) {
                log.info("dynamicTableName -- >tableName={}, tenant={}",tableName, tenant);
                return tableName + "_" + tenant;
            }
            return tableName;
        });
        mybatisPlusInterceptor.addInnerInterceptor(dynamicTableNameInnerInterceptor);
        // 分頁插件
        mybatisPlusInterceptor.addInnerInterceptor(new PaginationInnerInterceptor());
        return mybatisPlusInterceptor;
    }
}

3. TenantContext 類：租戶上下文管理器

public class TenantContext {

    // 需要替換的動態(tài)表名
    public static ThreadLocal<String> TENANT_DATA = new ThreadLocal<>();

    //用于記錄哪些表可以使用該動態(tài)表名處理器
    private static List<String> tableNames = Arrays.asList("test_user");

    //設置請求線程的租戶數(shù)據
    public static void setTenantData(String tenant) {
        TENANT_DATA.set(tenant);
    }

    //刪除當前請求線程的租戶數(shù)據
    public static void removeYearData() {
        TENANT_DATA.remove();
    }

    public static Boolean containsTableName(String tableName) {
        if (tableNames.contains(tableName)) {
            return true;
        }
        return false;
    }

}

4. 實體類

@Data
@TableName("test_user")
public class TestUser implements Serializable {
    private String id;
    private String name;
    private String managerId;
    private String salary;
    private String age;
    private String departId;
    private String remark;
    private String province;
}

5. 具體實現(xiàn)

@RestController
@Slf4j
public class TestController {

    @Resource
    private ITestUserService testUserService;

    @GetMapping("/selectList")
    public List<TestUser> selectList(@RequestParam String tenant) {
        return testUserService.selectList(tenant);
    }
}

public interface ITestUserService extends IService<TestUser> {

    List<TestUser> selectList(String tenant);
}

@Service
public class TestUserServiceImpl extends ServiceImpl<TestUserMapper, TestUser> implements ITestUserService {

    @Overridejava
    public List<TestUser> selectList(String tenant) {
        TenantContext.setTenantData(tenant);
        Wrapper<TestUser> queryWrapper = new QueryWrapper<>();
        return baseMapper.selectList(queryWrapper);
    }
}

到此這篇關于MyBatis-Plus使用動態(tài)表名分表查詢的實現(xiàn) 的文章就介紹到這了,更多相關MyBatis-Plus 動態(tài)表查詢內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: