mirror of https://gitee.com/stylefeng/roses
【8.0】【tenant】【db】更新租户拦截器
parent
1dd597025a
commit
84c5540be2
|
@ -0,0 +1,45 @@
|
|||
/*
|
||||
* Copyright [2020-2030] [https://www.stylefeng.cn]
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*
|
||||
* Guns采用APACHE LICENSE 2.0开源协议,您在使用过程中,需要注意以下几点:
|
||||
*
|
||||
* 1.请不要删除和修改根目录下的LICENSE文件。
|
||||
* 2.请不要删除和修改Guns源码头部的版权声明。
|
||||
* 3.请保留源码和相关描述文件的项目出处,作者声明等。
|
||||
* 4.分发源码时候,请注明软件出处 https://gitee.com/stylefeng/guns
|
||||
* 5.在修改包名,模块名称,项目代码等时,请注明软件出处 https://gitee.com/stylefeng/guns
|
||||
* 6.若您的项目无法满足以上几点,可申请商业授权
|
||||
*/
|
||||
package cn.stylefeng.roses.kernel.db.api.pojo.tenant;
|
||||
|
||||
import lombok.Data;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* 租户相关的表配置
|
||||
*
|
||||
* @author fengshuonan
|
||||
* @since 2023/8/30 10:37
|
||||
*/
|
||||
@Data
|
||||
public class TenantTableProperties {
|
||||
|
||||
/**
|
||||
* 涉及到租户业务的表集合
|
||||
*/
|
||||
private List<String> businessTableList;
|
||||
|
||||
}
|
|
@ -0,0 +1,69 @@
|
|||
package cn.stylefeng.roses.kernel.db.mp.tenant;
|
||||
|
||||
import cn.hutool.core.util.ObjectUtil;
|
||||
import cn.stylefeng.roses.kernel.auth.api.context.LoginContext;
|
||||
import cn.stylefeng.roses.kernel.auth.api.pojo.login.LoginUser;
|
||||
import cn.stylefeng.roses.kernel.db.api.pojo.tenant.TenantTableProperties;
|
||||
import com.baomidou.mybatisplus.extension.plugins.handler.TenantLineHandler;
|
||||
import net.sf.jsqlparser.expression.Expression;
|
||||
import net.sf.jsqlparser.expression.LongValue;
|
||||
import net.sf.jsqlparser.schema.Column;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* 租户隔离插件
|
||||
*
|
||||
* @author fengshuonan
|
||||
* @since 2023/8/30 10:14
|
||||
*/
|
||||
public class ProjectTenantInterceptor implements TenantLineHandler {
|
||||
|
||||
/**
|
||||
* 租户相关的表的配置
|
||||
*/
|
||||
private TenantTableProperties tenantTableProperties;
|
||||
|
||||
public ProjectTenantInterceptor(TenantTableProperties tenantTableProperties) {
|
||||
this.tenantTableProperties = tenantTableProperties;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Expression getTenantId() {
|
||||
LoginUser loginUserNullable = LoginContext.me().getLoginUserNullable();
|
||||
if (loginUserNullable == null) {
|
||||
return null;
|
||||
}
|
||||
if (ObjectUtil.isEmpty(loginUserNullable.getTenantId())) {
|
||||
return null;
|
||||
}
|
||||
return new LongValue(loginUserNullable.getTenantId());
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean ignoreTable(String tableName) {
|
||||
|
||||
if (tenantTableProperties == null) {
|
||||
return true;
|
||||
}
|
||||
|
||||
List<String> businessTableList = tenantTableProperties.getBusinessTableList();
|
||||
if (ObjectUtil.isEmpty(businessTableList)) {
|
||||
return true;
|
||||
}
|
||||
|
||||
for (String tenantTable : businessTableList) {
|
||||
if (tenantTable.equalsIgnoreCase(tableName)) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean ignoreInsert(List<Column> columns, String tenantIdColumn) {
|
||||
return TenantLineHandler.super.ignoreInsert(columns, tenantIdColumn);
|
||||
}
|
||||
|
||||
}
|
|
@ -31,6 +31,12 @@
|
|||
<version>${roses.version}</version>
|
||||
</dependency>
|
||||
|
||||
<!--properties自动提示装载-->
|
||||
<dependency>
|
||||
<groupId>org.springframework.boot</groupId>
|
||||
<artifactId>spring-boot-configuration-processor</artifactId>
|
||||
</dependency>
|
||||
|
||||
</dependencies>
|
||||
|
||||
</project>
|
||||
|
|
|
@ -24,15 +24,20 @@
|
|||
*/
|
||||
package cn.stylefeng.roses.kernel.db.starter;
|
||||
|
||||
import cn.stylefeng.roses.kernel.db.api.pojo.tenant.TenantTableProperties;
|
||||
import cn.stylefeng.roses.kernel.db.mp.dbid.CustomDatabaseIdProvider;
|
||||
import cn.stylefeng.roses.kernel.db.mp.fieldfill.CustomMetaObjectHandler;
|
||||
import cn.stylefeng.roses.kernel.db.mp.injector.CustomInsertBatchSqlInjector;
|
||||
import cn.stylefeng.roses.kernel.db.mp.tenant.ProjectTenantInterceptor;
|
||||
import com.baomidou.mybatisplus.autoconfigure.MybatisPlusAutoConfiguration;
|
||||
import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
|
||||
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
|
||||
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 org.springframework.boot.autoconfigure.AutoConfigureBefore;
|
||||
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
|
||||
import org.springframework.boot.context.properties.ConfigurationProperties;
|
||||
import org.springframework.context.annotation.Bean;
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
|
||||
|
@ -46,6 +51,19 @@ import org.springframework.context.annotation.Configuration;
|
|||
@AutoConfigureBefore(MybatisPlusAutoConfiguration.class)
|
||||
public class ProjectMyBatisPlusAutoConfiguration {
|
||||
|
||||
/**
|
||||
* 设置租户的表的集合,在yml配置中配置 tenant.businessTableList
|
||||
*
|
||||
* @author fengshuonan
|
||||
* @since 2023/8/30 10:39
|
||||
*/
|
||||
@Bean
|
||||
@ConfigurationProperties(prefix = "tenant")
|
||||
@ConditionalOnMissingBean(TenantTableProperties.class)
|
||||
public TenantTableProperties tenantTableProperties() {
|
||||
return new TenantTableProperties();
|
||||
}
|
||||
|
||||
/**
|
||||
* 新的分页插件
|
||||
*
|
||||
|
@ -53,9 +71,12 @@ public class ProjectMyBatisPlusAutoConfiguration {
|
|||
* @since 2020/12/24 13:13
|
||||
*/
|
||||
@Bean
|
||||
public MybatisPlusInterceptor mybatisPlusInterceptor() {
|
||||
public MybatisPlusInterceptor mybatisPlusInterceptor(TenantTableProperties tenantTableProperties) {
|
||||
MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
|
||||
|
||||
// 使用多租户插件
|
||||
interceptor.addInnerInterceptor(tenantLineInnerInterceptor(tenantTableProperties));
|
||||
|
||||
// 使用分页插插件
|
||||
interceptor.addInnerInterceptor(paginationInterceptor());
|
||||
|
||||
|
@ -110,7 +131,7 @@ public class ProjectMyBatisPlusAutoConfiguration {
|
|||
}
|
||||
|
||||
/**
|
||||
* 自定义sqlInjector
|
||||
* 自定义sqlInjector,针对批量插入优化
|
||||
*
|
||||
* @author fengshuonan
|
||||
* @since 2022/9/17 14:28
|
||||
|
@ -120,4 +141,15 @@ public class ProjectMyBatisPlusAutoConfiguration {
|
|||
return new CustomInsertBatchSqlInjector();
|
||||
}
|
||||
|
||||
/**
|
||||
* 多租户插件
|
||||
*
|
||||
* @author fengshuonan
|
||||
* @since 2023/8/30 10:17
|
||||
*/
|
||||
@Bean
|
||||
public TenantLineInnerInterceptor tenantLineInnerInterceptor(TenantTableProperties tenantTableProperties) {
|
||||
return new TenantLineInnerInterceptor(new ProjectTenantInterceptor(tenantTableProperties));
|
||||
}
|
||||
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue