更新 MybatisConfiguration.java.ftl

SLN/%PUBPRJ%-core/src/main/java/%SYS_PKGPATH%/core/util/config/MybatisConfiguration.java.ftl

+
 <#ibiztemplate>
 TARGET=PSSYSTEM
 </#ibiztemplate>
@@ -16,6 +17,15 @@ import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 import org.mybatis.spring.annotation.MapperScan;
 import java.util.Properties;
+<#if sys.getSaaSMode()==1>
+import cn.ibizlab.resourcescheduling.util.security.AuthenticationUser;
+import com.baomidou.mybatisplus.extension.plugins.tenant.TenantHandler;
+import com.baomidou.mybatisplus.extension.plugins.tenant.TenantSqlParser;
+import net.sf.jsqlparser.expression.Expression;
+import net.sf.jsqlparser.expression.StringValue;
+import java.util.ArrayList;
+import java.util.List;
+</#if>
 
 /**
  * mybatis全局配置类
@@ -52,6 +62,37 @@ public class MybatisConfiguration {
         // paginationInterceptor.setOverflow(false);
         // 设置最大单页限制数量，默认 500 条，-1 不受限制
         paginationInterceptor.setLimit(-1);
+<#if sys.getSaaSMode()==1>
+
+        // 创建SQL解析器集合
+        List<ISqlParser> sqlParserList = new ArrayList<>();
+        // 创建租户SQL解析器
+        TenantSqlParser tenantSqlParser = new TenantSqlParser();
+
+        // 设置租户处理器
+        tenantSqlParser.setTenantHandler(new TenantHandler() {
+
+            @Override
+            public Expression getTenantId(boolean where) {
+                AuthenticationUser user = AuthenticationUser.getAuthenticationUser();
+                return new StringValue(user.getDomain());
+            }
+
+            @Override
+            public String getTenantIdColumn() {
+                return "SRFDCID";
+            }
+
+            @Override
+            public boolean doTableFilter(String tableName) {
+                return false;
+            }
+        });
+
+        sqlParserList.add(tenantSqlParser);
+        paginationInterceptor.setSqlParserList(sqlParserList);
+</#if>        
+
         // 开启 count 的 join 优化,只针对部分 left join
         paginationInterceptor.setCountSqlParser(new JsqlParserCountOptimize(true));
         return paginationInterceptor;