【8.0】【tenant】【db】更新租户拦截器

kernel-d-db/db-api/src/main/java/cn/stylefeng/roses/kernel/db/api/pojo/tenant/TenantTableProperties.java

@@ -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;
+
+}

kernel-d-db/db-sdk-mp/src/main/java/cn/stylefeng/roses/kernel/db/mp/tenant/ProjectTenantInterceptor.java

@@ -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);
+    }
+
+}

kernel-d-db/db-spring-boot-starter/pom.xml

@@ -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>

kernel-d-db/db-spring-boot-starter/src/main/java/cn/stylefeng/roses/kernel/db/starter/ProjectMyBatisPlusAutoConfiguration.java

@@ -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));
+    }
+
 }