背景
多 A B 两个模块(使用同一个域名),都使用了 SaToken 进行鉴权且使用的同一个 redis,并且各自都有自己的用户体系,导致了一个问题:打开两个网页,分别展示 A B 两个模块,把 A 模块网页的链接复制到 B 模块下访问,此时会访问 A 模块的后端逻辑,但是携带的是 B 的 token,然而此 token 在 redis 中又是真实存在的,导致 B 模块的 controller 上的 @SaCheckLogin
注解失效。
解决
大致想到了 2 中解决方案:
- 每个模块用不同的 key 来存 token 到 header 中:如 token-a, token-b。当前端不方便修改的时候,可以后端统一处理:
@WebFilter(urlPatterns = "/*")
@Order(1)
public class ChangeHeaderFilter extends HttpFilter {
@Override
protected void doFilter(HttpServletRequest req, HttpServletResponse res, FilterChain chain) throws IOException, ServletException {
HeaderRequestWrapper httpServletRequestWrapper = new HeaderRequestWrapper(req);
// 这里的 tokenName 是之前和前端约定的 token 名称;
// 由于前端的 token 名称不方便修改,这里通过添加一个新的 header 来传递 token;
// 如将 Authorization:xxx 改为 Authorization-super:xxx
final String oldTokenName = "Authorization";
final String newTokenName = StpUtil.getTokenName();
String authorization = req.getHeader(oldTokenName);
if (authorization != null) {
httpServletRequestWrapper.addHeader(newTokenName, authorization);
}
chain.doFilter(httpServletRequestWrapper, res);
}
private static class HeaderRequestWrapper extends HttpServletRequestWrapper {
public HeaderRequestWrapper(HttpServletRequest request) {
super(request);
}
private final Map<String, String> headerMap = new HashMap<>();
public void addHeader(String name, String value) {
this.headerMap.put(name, value);
}
@Override
public String getHeader(String name) {
String headerValue = super.getHeader(name);
if (this.headerMap.containsKey(name)) {
headerValue = this.headerMap.get(name);
}
return headerValue;
}
@Override
public Enumeration<String> getHeaderNames() {
List<String> names = Collections.list(super.getHeaderNames());
names.addAll(this.headerMap.keySet());
return Collections.enumeration(names);
}
@Override
public Enumeration<String> getHeaders(String name) {
List<String> values = Collections.list(super.getHeaders(name));
if (this.headerMap.containsKey(name)) {
values = Collections.singletonList(this.headerMap.get(name));
}
return Collections.enumeration(values);
}
}
}
- 每个模块用不同的 StpUtil.type,即复制 StpUtil 所有代码,比如 StpUtilA 并设置
StpUtilA.type = "login-a"
以及StpUtilB.type = "login-b"
。
这种方式有个弊端,就是需要统一替换所有使用 StpUtil.type 的地方,如@SaCheckLogin(type = StpUtil.TYPE) ==> @SaCheckLogin(type = StpUtilA.TYPE)
网友评论