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

Compare commits

base: attiya:9b7c24ef7f030c8af330c261edd47a7da6fa80b4
Branches Tags
attiya:main
..
compare: attiya:d1e87ab995e657c19e3421d2e16e8244ad71da93
Branches Tags
attiya:main

No commits in common. "9b7c24ef7f030c8af330c261edd47a7da6fa80b4" and "d1e87ab995e657c19e3421d2e16e8244ad71da93" have entirely different histories.
9b7c24ef7f ... d1e87ab995

2 changed files with 9 additions and 12 deletions
Show Stats Expand all files Collapse all files

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

@ -6,13 +6,10 @@ 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,6 +6,7 @@ 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 {
@ -14,12 +15,11 @@ public class TenantInterceptor implements HandlerInterceptor {
, HttpServletResponse response, Object handler) throws Exception { , HttpServletResponse response, Object handler) throws Exception {
//通过 request 去获取租户 ID //通过 request 去获取租户 ID
if (StpUtil.isLogin()) { Long tenantId = getTenantIdByReuqest(request);
Long tenantId = getTenantIdByReuqest(request);
//设置租户ID到 request 的 attribute //设置租户ID到 request 的 attribute
request.setAttribute("tenantId", tenantId); request.setAttribute("tenantId", tenantId);
return true;
}
return true; return true;
} }