更新 SaaSTenantHandler.java.ftl

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

-A
\ No newline at end of file
+<#ibiztemplate>
+TARGET=PSSYSTEM
+</#ibiztemplate>
+<#if sys.getSaaSMode()==1>
+package ${pub.getPKGCodeName()}.core.util.config;
+
+import ${pub.getPKGCodeName()}.util.security.AuthenticationUser;
+import com.baomidou.mybatisplus.extension.plugins.tenant.TenantHandler;
+import net.sf.jsqlparser.expression.Expression;
+import net.sf.jsqlparser.expression.StringValue;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.context.properties.EnableConfigurationProperties;
+import org.springframework.stereotype.Component;
+
+
+@Component
+@EnableConfigurationProperties(SaaSTenantProperties.class)
+public class SaaSTenantHandler implements TenantHandler {
+
+    @Autowired
+    private SaaSTenantProperties saaSTenantProperties;
+
+    @Override
+    public Expression getTenantId(boolean where) {
+        AuthenticationUser user = AuthenticationUser.getAuthenticationUser();
+        return new StringValue(user.getDomain());
+    }
+
+    @Override
+    public String getTenantIdColumn() {
+        return saaSTenantProperties.getColumn();
+    }
+
+    @Override
+    public boolean doTableFilter(String tableName) {
+        tableName = tableName.replace("`","") ;
+        if (saaSTenantProperties.getSysTables().contains(tableName))
+            return true;
+        return false;
+    }
+
+}
+</#if>
\ No newline at end of file