mybatis-plus多租户实现

 获取header中租户id

编写拦截器

@Component
public class HeadersInterceptor implements HandlerInterceptor {

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        String tenantId = request.getHeader("tenant-id");
        //使租户id可全局获取即可,技术不限
        StpUtil.getSession().set("tenantId ", tenantId );
        return true;
    }
}

配置拦截器

@SpringBootConfiguration
@EnableWebMvc
public class InterceptorsConfigure implements WebMvcConfigurer {

    @Autowired
    private HeadersInterceptor headersInterceptor;

    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(headersInterceptor)
                .excludePathPatterns("/tokens","/tokens/**","/users/**")
                .addPathPatterns("/**");
}

mybatis-plus多租户插件

实现租户处理器

@Slf4j
@Component
public class MyTenantLineHandler implements TenantLineHandler {

    /**
     * 租户字段名
     */
    private static final String SYSTEM_TENANT_ID = "tenant_id";

    /**
     * 需要过滤的表
     */
    private static final List<String> IGNORE_TENANT_TABLES = Arrays.asList("s_user");

    /**
     * 获取租户ID值
     */
    @Override
    public Expression getTenantId() {
        return new LongValue(Long.valueOf(StpUtil.getSession().get("tenantId").toString()));
    }

    /**
     * 租户字段名,默认是tenant_id,如果想改成其他字段,在这里返回即可
     */
    @Override
    public String getTenantIdColumn() {
        return SYSTEM_TENANT_ID;
    }

    /**
     * 有不需要进行租户隔离的表,在这里返回true
     */
    @Override
    public boolean ignoreTable(String tableName) {
        return IGNORE_TENANT_TABLES.contains(tableName);
    }
}

在mybatis-plus配置中添加多租户插件

@Configuration
@MapperScan("com.example.tenant.mapper")
public class MybatisPlusConfig {

    @Bean
    public MybatisPlusInterceptor mybatisPlusInterceptor() {
        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
        interceptor.addInnerInterceptor(new TenantLineInnerInterceptor(new MyTenantLineHandler()));
        return interceptor;
    }

}

注:建议数据库表都加上tenant_id字段

  • 4
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值