mybatis plus 实现自动给每个sql条件增加租户Id字段的原理及使用
首先,用到了ThreadaLocal,一个TenantContextUtil类有多个ThreadLocal的属性,包含租户id和是否忽略租户等等字段。
在租户拦截器中实现对TenantContextUtil属性值的填充清除操作。每次请求的RequestHeader中有租户id,从请求头拿取填充的。
===================================================================
在配置类中注入如下bean,至于为什么注入这个bean就可以实现,其他一般都是重写什么方法之类的,这里我猜测是根据MybatisPlusInterceptor实例的bean类型来看看有没有这个拦截器的、
@Bean
public MybatisPlusInterceptor mybatisPlusInterceptor() {
MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
interceptor.addInnerInterceptor(new TenantLineInnerInterceptor(new TenantLineHandler() {
//这个方法是条件拼接租户字段的值是多少,我们是从前面ThreadLocal中获取返回
//StringValue是Expression 的子类
@Override
public Expression getTenantId() {
return new StringValue(TenantContext.getTenantId());
}
//这个方法是条件拼接租户字段的字段名称:
//这个字段名称一般是放在一个类中定义好的
@Override
public String getTenantIdColumn() {
return "tenant_id";
}
//这个类是获取是否忽略拼接租户这个字段条件
//也是从TenantContext.getIgnore()获取。
@Override
public boolean ignoreTable(String tableName) {
return TenantContext.getIgnore()
}
}));
return interceptor;
}
这三个方法注意ignoreTable()这个方法,拦截器是每次执行sql就会自动拼接,因此,我们业务中构造wrapper时,就可以调用TenantContext.setIgnore(true)更改threadLocal的这个值,间接达到,在拦截器这里TenantContext.getIgnore()获取返回的就是我们业务代码更改的true,就会忽略条件添加租户字段