New file |
| | |
| | | package com.dy.common.webFilter; |
| | | |
| | | import com.alibaba.fastjson2.JSON; |
| | | import com.dy.common.contant.Constant; |
| | | import com.dy.common.webUtil.BaseResponse; |
| | | import com.dy.common.webUtil.BaseResponseUtils; |
| | | import com.mysql.cj.util.StringUtils ; |
| | | import jakarta.servlet.*; |
| | | 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 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 { |
| | | // 如果不是非过滤项,执行过滤逻辑 |
| | | HttpServletRequest request = (HttpServletRequest) servletRequest; |
| | | HttpServletResponse response = (HttpServletResponse) servletResponse; |
| | | 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, "未得到用户token"); |
| | | //response.sendRedirect("/error"); |
| | | } |
| | | } |
| | | } |
| | | |
| | | @Override |
| | | public void destroy() { |
| | | } |
| | | } |