数据分离-未登录状态兼容
This commit is contained in:
parent
5c1bcfec16
commit
924662c500
@ -6,10 +6,13 @@ import org.springframework.web.context.request.RequestContextHolder;
|
||||
|
||||
public class MyTenantFactory implements TenantFactory {
|
||||
|
||||
public Object[] getTenantIds(){
|
||||
public Object[] getTenantIds() {
|
||||
RequestAttributes attributes = RequestContextHolder.getRequestAttributes();
|
||||
Long tenantId = (Long) attributes.getAttribute("tenantId", RequestAttributes.SCOPE_REQUEST);
|
||||
|
||||
return new Object[]{tenantId};
|
||||
if (tenantId != null) {
|
||||
return new Object[]{tenantId};
|
||||
} else {
|
||||
return new Object[]{};
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -6,7 +6,6 @@ import jakarta.servlet.http.HttpServletRequest;
|
||||
import jakarta.servlet.http.HttpServletResponse;
|
||||
import org.springframework.web.servlet.HandlerInterceptor;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
public class TenantInterceptor implements HandlerInterceptor {
|
||||
|
||||
@ -15,11 +14,12 @@ public class TenantInterceptor implements HandlerInterceptor {
|
||||
, HttpServletResponse response, Object handler) throws Exception {
|
||||
|
||||
//通过 request 去获取租户 ID
|
||||
Long tenantId = getTenantIdByReuqest(request);
|
||||
|
||||
//设置租户ID到 request 的 attribute
|
||||
request.setAttribute("tenantId", tenantId);
|
||||
|
||||
if (StpUtil.isLogin()) {
|
||||
Long tenantId = getTenantIdByReuqest(request);
|
||||
//设置租户ID到 request 的 attribute
|
||||
request.setAttribute("tenantId", tenantId);
|
||||
return true;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user