Filter存用户信息

  @Override
    public void doFilter(ServletRequest request, ServletResponse response,
                         FilterChain chain) throws IOException, ServletException {
        HttpServletRequest httpServletRequest = (HttpServletRequest) request;

        //判断如果是APP请求,则直接通过
        String requestURI = httpServletRequest.getRequestURI();
        if (requestURI.startsWith("/e/welink") ||
                requestURI.startsWith("/e/test/welink") ||
                requestURI.startsWith("/e/approve/waitApprove") ||
                requestURI.startsWith("/e/page/applicationPage") ||
                requestURI.startsWith("/e/ows/user/listRole") ||
                requestURI.endsWith(".ttf") ||
                requestURI.endsWith(".woff") ||
                requestURI.endsWith(".eot") ||
                requestURI.endsWith(".woff2") ||
                requestURI.endsWith(".css") ||
                requestURI.endsWith(".js") ||
                requestURI.endsWith(".png") ||
                requestURI.endsWith(".jpg") ||
                requestURI.endsWith(".wav") ||
                requestURI.endsWith(".xlsx") ||
                requestURI.endsWith(".svg") ||
                requestURI.startsWith("/e/api/csp/") ||
                requestURI.endsWith("/e/api/important_contact/all") ||
                requestURI.contains("getRepOffAndIndustry") ||
                requestURI.contains("getAllQcLine")
                // requestURI.endsWith("/e/promotion/test")
                ) {
            chain.doFilter(request, response);
            return;
        }
        if (requestURI.indexOf("/e/page") != -1) {
            if (isMSBrowser(httpServletRequest)) {
                response.setCharacterEncoding("gbk");
                PrintWriter out = response.getWriter();
                out.println("<div>本系统不支持 IE EDGE 等浏览器,推荐使用 Google Chrome浏览器,谢谢。</div>");
                return;
            }
        }
        HttpSession session = httpServletRequest.getSession();
        UserInfoBean uiBean = (UserInfoBean) session
                .getAttribute(SsoConstants.SESSION_USER_INFO_KEY);
        if (uiBean == null) {   //会话已经超时
            if (CommonUtil.isAjax(httpServletRequest)) {   //ajax请求会话超时,filter不会进入统一异常拦截
                Map<String, String> map = new HashMap<>();
                map.put("code", "401");
                response.setCharacterEncoding("utf-8");
                response.getWriter().print(JSONObject.toJSON(map));
                return;
            }
            try {
                // 从 sso 跳转回来 回到当前访问的 url
                SsoUtil.loginAndRedirect2AppCurrentURL((HttpServletRequest) request,
                        (HttpServletResponse) response);
                return;
            } catch (Exception e) {
                e.printStackTrace();
            }
        } else {
            try {
                User token = null;
                try (Jedis jedis = jedisPool.getResource()) {
                    String redisKey = "userInfo_" + uiBean.getUid();
                    String tokenStr = jedis.get(redisKey);
                    if (StringUtils.isNotBlank(tokenStr)) {
                        token = JSON.parseObject(tokenStr, User.class);
                    }
                    if (token == null) {//|| token.getRootList().size() == 0
                        Result r = userService.login((HttpServletRequest) request,
                                (HttpServletResponse) response);
                        token = (User) r.getData();
                        if ("N".equals(token.getIsVaild())) {
                            if (CommonUtil.isAjax(httpServletRequest)) {   //ajax请求会话超时,filter不会进入统一异常拦截
                                throw new BadRequestException("此账号已被禁用,如需开通,请联系系统管理员 王超伟 w00509928 !");
                            } else {
                                response.setCharacterEncoding("gbk");
                                response.getWriter().println("此账号已被禁用,如需开通,请联系系统管理员 王超伟 w00509928 !");
                                return;
                            }
                        }

                        // "审批"跳转流程
                        if (token.getIsExists()) {
                            // 用户表"存在"用户
                            Boolean redirect2ApplicationPage = approveLoginFilter.redirectWithUserExist(token);
                            if (redirect2ApplicationPage) {
                                ((HttpServletResponse) response).sendRedirect(((HttpServletRequest) request).getContextPath() + "/page/applicationPage?agentId=" + token.getUserAgentId());
                                return;
                            }
                        } else {
                            // 用户表"不存在"用户
                            Map<String, Object> newUserMap = approveLoginFilter.getNewUserByOldAgentId(token); // 查看是否更换账号
                            if (ObjectUtils.isEmpty(newUserMap)) {
                                // 没有更换账号,则表示完全为一个新的账户,按照正常的流程
                                approveLoginFilter.redirectWithUserNotExist(token);
                                ((HttpServletResponse) response).sendRedirect(((HttpServletRequest) request).getContextPath() + "/page/applicationPage?agentId=" + token.getUserAgentId());
                                return;
                            } else {
                                // 更换了账号,则需要重新跳转至 用户表"存在"用户流程
                                User newUser = new User();
                                newUser.setId((String) newUserMap.get("ID"));
                                newUser.setUserAgentId((String) newUserMap.get("USER_AGENTID"));
                                newUser.setUserEmail((String) newUserMap.get("USER_EMAIL"));
                                newUser.setUserPhono((String) newUserMap.get("USER_PHONE"));
                                newUser.setCreateTime((Long) newUserMap.get("CREATE_DATE"));
                                newUser.setRoleId((String) newUserMap.get("ROLE_ID"));
                                newUser.setQc_line((String) newUserMap.get("QC_LINE"));
                                newUser.setRep_off((String) newUserMap.get("REP_OFF"));
                                Boolean redirect2ApplicationPage = approveLoginFilter.redirectWithUserExist(newUser);
                                if (redirect2ApplicationPage) {
                                    ((HttpServletResponse) response).sendRedirect(((HttpServletRequest) request).getContextPath() + "/page/applicationPage?agentId=" + token.getUserAgentId());
                                    return;
                                } else {
                                    // 重新登录
                                    Result loginResult = userService.login((HttpServletRequest) request, (HttpServletResponse) response);
                                    token = (User) loginResult.getData();
                                }
                            }
                        }

                        // 加载整改配置角色
                        userService.setRectRole(token);
                        // 加载菜单、权限等
                        userService.loadUserResource(token);
//                        session.setAttribute("userInfo", token);
                        logUtils.log("登录", null, token.getUserAgentId(), token.getId());
                        dao.commit();
                        jedis.set(redisKey, JSON.toJSONString(token));
                        log.info("从数据库中加载用户信息");
                    }
                    // 设置过期时间 单位(秒)
                    jedis.expire(redisKey, EXP_TIME);
                }
                Method m = AppContext.class.getDeclaredMethod("setCurrentUser", User.class);
                m.setAccessible(true);
                m.invoke(null, token);
                chain.doFilter(request, response);
            } catch (Exception ex) {
                log.error(ex.getMessage(), ex);
            } finally {
                AppContext.clearAll();
                try {
                    dao.close();
                } catch (Exception e) {
                    e.printStackTrace();
                    log.error(e.getMessage(), e);
                }
            }
        }
    }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值