he2802 преди 4 години
родител
ревизия
8b254e9841
променени са 1 файла, в които са добавени 1 реда и са изтрити 3 реда
  1. 1 3
      zhongzheng-framework/src/main/java/com/zhongzheng/framework/mybatisplus/CustomTenantLineHandler.java

+ 1 - 3
zhongzheng-framework/src/main/java/com/zhongzheng/framework/mybatisplus/CustomTenantLineHandler.java

@@ -70,8 +70,8 @@ public class CustomTenantLineHandler implements TenantLineHandler {
         }*/
         String TenantId = ServletUtils.getRequest().getHeader("TenantId");
         if(!StrUtil.isNotBlank(TenantId)||TenantId==null){
+            TenantId = ServletUtils.getResponse().getHeader("TenantId");
             if(enabledTenant){
-                TenantId = ServletUtils.getResponse().getHeader("TenantId");
                 if(!StrUtil.isNotBlank(TenantId)||TenantId==null){
                     int code = HttpStatus.HTTP_UNAVAILABLE;
                     String msg = "企业非法访问";
@@ -79,8 +79,6 @@ public class CustomTenantLineHandler implements TenantLineHandler {
                     ServletUtils.getResponse().getWriter().flush();
                     ServletUtils.getResponse().getWriter().close();
                 }
-            }else{
-                TenantId = ServletUtils.getResponse().getHeader("TenantId");
             }
         }
         return new LongValue(TenantId);