85 lines
3.2 KiB
Java
85 lines
3.2 KiB
Java
package com.rj.config;
|
||
|
||
import com.baomidou.mybatisplus.annotation.DbType;
|
||
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
|
||
import com.baomidou.mybatisplus.extension.plugins.handler.TenantLineHandler;
|
||
import com.baomidou.mybatisplus.extension.plugins.inner.OptimisticLockerInnerInterceptor;
|
||
import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
|
||
import com.baomidou.mybatisplus.extension.plugins.inner.TenantLineInnerInterceptor;
|
||
import com.rj.tenant.TenantContextHolder;
|
||
import net.sf.jsqlparser.expression.Expression;
|
||
import net.sf.jsqlparser.expression.StringValue;
|
||
import org.springframework.context.annotation.Bean;
|
||
import org.springframework.context.annotation.Configuration;
|
||
import org.springframework.transaction.annotation.EnableTransactionManagement;
|
||
|
||
import java.util.HashSet;
|
||
import java.util.Set;
|
||
|
||
/**
|
||
* MyBatis Plus 配置类
|
||
*
|
||
* <p>包含：</p>
|
||
* <ul>
|
||
* <li>多租户拦截器：根据当前线程中的租户ID自动为SQL拼接 tenant_id 条件；</li>
|
||
* <li>分页插件；</li>
|
||
* <li>乐观锁插件。</li>
|
||
* </ul>
|
||
*/
|
||
@Configuration
|
||
@EnableTransactionManagement
|
||
public class MybatisPlusConfig {
|
||
|
||
/**
|
||
* MyBatis-Plus 主拦截器配置
|
||
*/
|
||
@Bean
|
||
public MybatisPlusInterceptor mybatisPlusInterceptor(TenantLineHandler tenantLineHandler) {
|
||
MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
|
||
|
||
// 1. 多租户插件（需优先添加）
|
||
interceptor.addInnerInterceptor(new TenantLineInnerInterceptor(tenantLineHandler));
|
||
|
||
// 2. 分页插件
|
||
interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.MYSQL));
|
||
|
||
// 3. 乐观锁插件
|
||
interceptor.addInnerInterceptor(new OptimisticLockerInnerInterceptor());
|
||
|
||
return interceptor;
|
||
}
|
||
|
||
/**
|
||
* 多租户处理器：定义如何获取租户ID、租户字段名以及忽略的表
|
||
*/
|
||
@Bean
|
||
public TenantLineHandler tenantLineHandler() {
|
||
// 需要忽略多租户的表（不自动拼接 tenant_id）
|
||
Set<String> ignoreTables = new HashSet<>();
|
||
ignoreTables.add("tenant"); // 租户表本身
|
||
ignoreTables.add("industry_tags"); // 行业标签（示例：如认为是公共字典）
|
||
ignoreTables.add("menu"); // 菜单表（不需要租户隔离）
|
||
ignoreTables.add("ai_prompts"); // AI 提示词全局配置，表无 tenant_id 字段
|
||
|
||
return new TenantLineHandler() {
|
||
|
||
@Override
|
||
public Expression getTenantId() {
|
||
String tenantId = TenantContextHolder.getTenantId();
|
||
// 没有租户ID时，返回 null 由 MyBatis-Plus 决定处理策略，通常为不过滤或抛错（视版本而定）
|
||
return tenantId == null ? null : new StringValue(tenantId);
|
||
}
|
||
|
||
@Override
|
||
public String getTenantIdColumn() {
|
||
return "tenant_id";
|
||
}
|
||
|
||
@Override
|
||
public boolean ignoreTable(String tableName) {
|
||
// 忽略名单内的表不做租户隔离
|
||
return ignoreTables.contains(tableName);
|
||
}
|
||
};
|
||
}
|
||
} |