attiya/ebike-share
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

数据分离-未登录状态兼容

Browse Source
This commit is contained in:
attiya 2025-08-29 10:07:29 +08:00
parent 5c1bcfec16
commit 924662c500
2 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
ebike-user/src/main/java/com/cdzy/user/component/MyTenantFactory.java
View File

@ -6,10 +6,13 @@ import org.springframework.web.context.request.RequestContextHolder;
public class MyTenantFactory implements TenantFactory { public class MyTenantFactory implements TenantFactory {
public Object[] getTenantIds(){ public Object[] getTenantIds() {
RequestAttributes attributes = RequestContextHolder.getRequestAttributes(); RequestAttributes attributes = RequestContextHolder.getRequestAttributes();
Long tenantId = (Long) attributes.getAttribute("tenantId", RequestAttributes.SCOPE_REQUEST); Long tenantId = (Long) attributes.getAttribute("tenantId", RequestAttributes.SCOPE_REQUEST);
if (tenantId != null) {
return new Object[]{tenantId}; return new Object[]{tenantId};
} else {
return new Object[]{};
}
} }
} }

12
ebike-user/src/main/java/com/cdzy/user/component/TenantInterceptor.java
View File

@ -6,7 +6,6 @@ import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse; import jakarta.servlet.http.HttpServletResponse;
import org.springframework.web.servlet.HandlerInterceptor; import org.springframework.web.servlet.HandlerInterceptor;
import java.util.List;
public class TenantInterceptor implements HandlerInterceptor { public class TenantInterceptor implements HandlerInterceptor {
@ -15,11 +14,12 @@ public class TenantInterceptor implements HandlerInterceptor {
, HttpServletResponse response, Object handler) throws Exception { , HttpServletResponse response, Object handler) throws Exception {
//通过 request 去获取租户 ID //通过 request 去获取租户 ID
Long tenantId = getTenantIdByReuqest(request); if (StpUtil.isLogin()) {
Long tenantId = getTenantIdByReuqest(request);
//设置租户ID到 request 的 attribute //设置租户ID到 request 的 attribute
request.setAttribute("tenantId", tenantId); request.setAttribute("tenantId", tenantId);
return true;
}
return true; return true;
} }