关于多个模块(各自有一套用户体系)使用同一个 Redis 的问题

2024-01-31  本文已影响0人  雁过留声_泪落无痕

背景

多 A B 两个模块(使用同一个域名),都使用了 SaToken 进行鉴权且使用的同一个 redis,并且各自都有自己的用户体系,导致了一个问题:打开两个网页,分别展示 A B 两个模块,把 A 模块网页的链接复制到 B 模块下访问,此时会访问 A 模块的后端逻辑,但是携带的是 B 的 token,然而此 token 在 redis 中又是真实存在的,导致 B 模块的 controller 上的 @SaCheckLogin 注解失效。

解决

大致想到了 2 中解决方案:

  1. 每个模块用不同的 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);
        }
    }

}
  1. 每个模块用不同的 StpUtil.type,即复制 StpUtil 所有代码,比如 StpUtilA 并设置 StpUtilA.type = "login-a" 以及 StpUtilB.type = "login-b"
    这种方式有个弊端,就是需要统一替换所有使用 StpUtil.type 的地方,如 @SaCheckLogin(type = StpUtil.TYPE) ==> @SaCheckLogin(type = StpUtilA.TYPE)
上一篇 下一篇

猜你喜欢

热点阅读