|  |  |  | 
|---|
|  |  |  | import jakarta.servlet.http.HttpServletRequest; | 
|---|
|  |  |  | import jakarta.servlet.http.HttpServletResponse; | 
|---|
|  |  |  | import lombok.extern.slf4j.Slf4j; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | import java.io.IOException; | 
|---|
|  |  |  | import java.io.PrintWriter; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @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); | 
|---|
|  |  |  | HttpServletRequest httpRequest = (HttpServletRequest) servletRequest; | 
|---|
|  |  |  | String requestURI = httpRequest.getRequestURI(); | 
|---|
|  |  |  | // 检查请求是否为非过滤项 | 
|---|
|  |  |  | if (UserTokenNoFilter.urls != null && UserTokenNoFilter.urls.stream().anyMatch(url -> requestURI.startsWith(url))) { | 
|---|
|  |  |  | // 如果是非过滤项,直接调用下一个过滤器或servlet | 
|---|
|  |  |  | 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 { | 
|---|
|  |  |  | // 如果不是非过滤项,执行过滤逻辑 | 
|---|
|  |  |  | HttpServletRequest request = (HttpServletRequest) servletRequest; | 
|---|
|  |  |  | HttpServletResponse response = (HttpServletResponse) servletResponse; | 
|---|
|  |  |  | String token = request.getHeader(Constant.TokenKeyInHeader); | 
|---|
|  |  |  | 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, "未得到用户token"); | 
|---|
|  |  |  | //response.sendRedirect("/error"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | //request.setAttribute(Constant.ErrorKeyInRequest, "未得到用户token"); | 
|---|
|  |  |  | //response.sendRedirect("/error"); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|