he2802 4 лет назад
Родитель
Сommit
0999e6aef7

+ 5 - 1
zhongzheng-admin-saas/src/main/java/com/zhongzheng/core/config/SwaggerConfig.java

@@ -127,10 +127,14 @@ public class SwaggerConfig
 
     private List<Parameter> getGlobalOperationParameters() {
         List<Parameter> pars = new ArrayList<>();
-        ParameterBuilder parameterBuilder = new ParameterBuilder();
         // header query cookie
+        // header query cookie
+        ParameterBuilder parameterBuilder = new ParameterBuilder();
         parameterBuilder.name("X-Auth-Token").description("token").modelRef(new ModelRef("string")).parameterType("header").defaultValue("test").required(false);
         pars.add(parameterBuilder.build());
+        ParameterBuilder parameterBuilder1 = new ParameterBuilder();
+        parameterBuilder1.name("AuthorizationToken").description("AuthorizationToken").modelRef(new ModelRef("string")).parameterType("header").defaultValue("BearerSuper 123").required(false);
+        pars.add(parameterBuilder1.build());
         return pars;
     }
 }

+ 3 - 1
zhongzheng-admin-saas/src/main/resources/application.yml

@@ -201,7 +201,9 @@ mybatis-plus:
       updateStrategy: NOT_EMPTY
       # 字段验证策略之 select,在 select 的时候的字段验证策略既 wrapper 根据内部 entity 生成的 where 条件
       selectStrategy: NOT_EMPTY
-
+  # 开启多租户
+  tenant:
+    enabled-tenant: false
 # PageHelper分页插件
 pagehelper:
   helperDialect: mysql

+ 3 - 1
zhongzheng-admin/src/main/resources/application.yml

@@ -201,7 +201,9 @@ mybatis-plus:
       updateStrategy: NOT_EMPTY
       # 字段验证策略之 select,在 select 的时候的字段验证策略既 wrapper 根据内部 entity 生成的 where 条件
       selectStrategy: NOT_EMPTY
-
+  # 开启多租户
+  tenant:
+    enabled-tenant: true
 # PageHelper分页插件
 pagehelper:
   helperDialect: mysql

+ 3 - 1
zhongzheng-api/src/main/resources/application.yml

@@ -201,7 +201,9 @@ mybatis-plus:
       updateStrategy: NOT_EMPTY
       # 字段验证策略之 select,在 select 的时候的字段验证策略既 wrapper 根据内部 entity 生成的 where 条件
       selectStrategy: NOT_EMPTY
-
+  # 开启多租户
+  tenant:
+    enabled-tenant: true
 # PageHelper分页插件
 pagehelper:
   helperDialect: mysql

+ 11 - 5
zhongzheng-framework/src/main/java/com/zhongzheng/framework/mybatisplus/CustomTenantLineHandler.java

@@ -11,6 +11,7 @@ import lombok.SneakyThrows;
 import net.sf.jsqlparser.expression.Expression;
 import net.sf.jsqlparser.expression.LongValue;
 import org.apache.commons.compress.utils.Lists;
+import org.springframework.beans.factory.annotation.Value;
 
 import java.util.ArrayList;
 import java.util.List;
@@ -28,6 +29,9 @@ public class CustomTenantLineHandler implements TenantLineHandler {
      */
     public static final String TENANT_ID_COLUMN = "tenant_id";
 
+    @Value("${mybatis-plus.tenant.enabled-tenant:true}")
+    private boolean enabledTenant;
+
     /**
      * 忽略添加租户ID的表
      */
@@ -52,11 +56,13 @@ public class CustomTenantLineHandler implements TenantLineHandler {
     public Expression getTenantId() {
         String TenantId = ServletUtils.getRequest().getHeader("TenantId");
         if(!StrUtil.isNotBlank(TenantId)){
-            int code = HttpStatus.HTTP_UNAVAILABLE;
-            String msg = "企业非法访问";
-            ServletUtils.renderString( ServletUtils.getResponse(), JSON.toJSONString(AjaxResult.error(code, msg)));
-            ServletUtils.getResponse().getWriter().flush();
-            ServletUtils.getResponse().getWriter().close();
+            if(enabledTenant){
+                int code = HttpStatus.HTTP_UNAVAILABLE;
+                String msg = "企业非法访问";
+                ServletUtils.renderString( ServletUtils.getResponse(), JSON.toJSONString(AjaxResult.error(code, msg)));
+                ServletUtils.getResponse().getWriter().flush();
+                ServletUtils.getResponse().getWriter().close();
+            }
         }
         return new LongValue(TenantId);
     }