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