From beb074bf390cdd09b1c8089e395db5f421efc34c Mon Sep 17 00:00:00 2001 From: Fancy <Fancy.fx@outlook.com> Date: 星期三, 23 十月 2024 10:58:53 +0800 Subject: [PATCH] 优化代码 --- pms-parent/pms-common/src/main/java/com/dy/common/webFilter/UserTokenFilter.java | 60 ++++++++++++++++++++++++++++++------------------------------ 1 files changed, 30 insertions(+), 30 deletions(-) diff --git a/pms-parent/pms-common/src/main/java/com/dy/common/webFilter/UserTokenFilter.java b/pms-parent/pms-common/src/main/java/com/dy/common/webFilter/UserTokenFilter.java index 15c7f37..97fc80a 100644 --- a/pms-parent/pms-common/src/main/java/com/dy/common/webFilter/UserTokenFilter.java +++ b/pms-parent/pms-common/src/main/java/com/dy/common/webFilter/UserTokenFilter.java @@ -14,46 +14,46 @@ import java.io.PrintWriter; /** - * 鐧诲綍鐢ㄦ埛token杩囨睙鍣� - * 姝ょ被娌℃湁娉ㄨВ @Component 鐨勪互鍦ㄩ泦鎴愬叾鐨勫瓙妯″潡涓姞鍏ユ杩囨睙鍣� + * 鐧诲綍鐢ㄦ埛token杩囨护鍣� + * 姝ょ被娌℃湁娉ㄨВ @Component 锛� 浠ュ湪闆嗘垚鍏剁殑瀛愭ā鍧椾腑鍔犲叆姝よ繃婊ゅ櫒 */ @Slf4j public class UserTokenFilter implements Filter { @Override - public void init(FilterConfig filterConfig) throws ServletException { - } - - @Override public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException { HttpServletRequest request = (HttpServletRequest) servletRequest; HttpServletResponse response = (HttpServletResponse) servletResponse; - String token = request.getHeader(Constant.UserTokenKeyInHeader); - if(!StringUtils.isNullOrEmpty(token)){ - UserTokenContext.set(token); + String requestURI = request.getRequestURI(); + // 妫�鏌ヨ姹傛槸鍚︿负闈炶繃婊ら」 + if (UserTokenNoFilter.urls != null && UserTokenNoFilter.urls.stream().anyMatch(url -> requestURI.startsWith(url))) { + // 濡傛灉鏄潪杩囨护椤癸紝鐩存帴璋冪敤涓嬩竴涓繃婊ゅ櫒鎴杝ervlet filterChain.doFilter(servletRequest, servletResponse); - }else{ - PrintWriter pw = null ; - try { - BaseResponse<?> res = BaseResponseUtils.buildToLogin() ; - String jsonString = JSON.toJSONString(res); - response.setCharacterEncoding("UTF-8"); - response.setContentType("application/json; charset=utf-8"); - pw = response.getWriter() ; - pw.write(jsonString); - pw.flush(); - } catch (IOException e) { - e.printStackTrace(); - }finally { - if(pw != null){ - pw.close(); + } else { + // 濡傛灉涓嶆槸闈炶繃婊ら」锛屾墽琛岃繃婊ら�昏緫 + String token = request.getHeader(Constant.UserTokenKeyInHeader); + if(!StringUtils.isNullOrEmpty(token)){ + UserTokenContext.set(token); + filterChain.doFilter(servletRequest, servletResponse); + }else{ + PrintWriter pw = null ; + try { + BaseResponse<?> res = BaseResponseUtils.buildToLogin(); + String jsonString = JSON.toJSONString(res); + response.setCharacterEncoding("UTF-8"); + response.setContentType("application/json; charset=utf-8"); + pw = response.getWriter() ; + pw.write(jsonString); + pw.flush(); + } catch (IOException e) { + e.printStackTrace(); + }finally { + if(pw != null){ + pw.close(); + } } + //request.setAttribute(Constant.ErrorKeyInRequest, "鏈緱鍒扮敤鎴穞oken"); + //response.sendRedirect("/error"); } - //request.setAttribute(Constant.ErrorKeyInRequest, "鏈緱鍒扮敤鎴穞oken"); - //response.sendRedirect("/error"); } - } - - @Override - public void destroy() { } } \ No newline at end of file -- Gitblit v1.8.0