springboot LoginValidateFilter登录过滤器

淩亂°似流年 2022-03-28 10:06 308阅读 0赞

获取header和 request里的token判断，使用jwt的方式进行刷新token。然后返回到header中。 如果没有token或者tokeng过期则返回json 字符串。

import com.kakacl.product_service.utils.ErrorCode;
    import com.kakacl.product_service.utils.JWTUtils;
    import com.kakacl.product_service.utils.Resp;
    import io.jsonwebtoken.Claims;
    import org.apache.commons.lang.StringUtils;
    import org.springframework.beans.factory.annotation.Value;
    import org.springframework.cloud.context.config.annotation.RefreshScope;
    import org.springframework.stereotype.Component;
    
    import javax.servlet.*;
    import javax.servlet.annotation.WebFilter;
    import javax.servlet.http.HttpServletRequest;
    import javax.servlet.http.HttpServletResponse;
    import java.io.IOException;
    import java.io.OutputStream;
    
    /**
     * @author wangwei
     * @version v1.0.0
     * @description 登录验证过滤器Filter
     * @date 2019-01-10
     */
    @Component
    @WebFilter(filterName="LoginValidateFilter", urlPatterns="/*")
    @RefreshScope
    public class LoginValidateFilter implements Filter {
    
        @Value("${version}")
        private String version;
    
        @Override
        public void init(FilterConfig filterConfig) throws ServletException {
    
        }
    
        @Override
        public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {
            String token = ((HttpServletResponse) servletResponse).getHeader("token");
            if(StringUtils.isBlank(token)) {
                token = servletRequest.getParameter("token");
            }
            HttpServletRequest request = (HttpServletRequest) servletRequest;
            HttpServletResponse response = (HttpServletResponse) servletResponse;
            request.setCharacterEncoding("UTF-8");
            response.setCharacterEncoding("UTF-8");
    
            String path = request.getRequestURI();
            if(path.indexOf("/api/"+ version +"/account/sendPhoneCode")> -1 ){
                filterChain.doFilter(servletRequest, servletResponse);
                return;
            }
    
            if(path.indexOf("/api/"+ version +"/account/register")> -1 ){
                filterChain.doFilter(servletRequest, servletResponse);
                return;
            }
    
            if(path.indexOf("/api/"+ version +"/account/login")> -1 ){
                filterChain.doFilter(servletRequest, servletResponse);
                return;
            }
    
            if(StringUtils.isNotBlank(token)) {
                // 判断是否过期 没有过期在header更新token
                try{
                    Claims claims = JWTUtils.parseJWT(token);
                    ((HttpServletResponse) servletResponse).setHeader("token", JWTUtils.createJWT(claims.getId(), claims.getIssuer(), claims.getSubject(), 1000 * 60 * 30));
                    filterChain.doFilter(servletRequest, servletResponse);
                } catch (Exception e) {
                    servletResponse.setContentType("application/json; charset=utf-8");
                    servletResponse.setCharacterEncoding("UTF-8");
                    String userJson = Resp.fail(ErrorCode.UNLOGIN_ERROR).toString();
                    OutputStream out = servletResponse.getOutputStream();
                    out.write(userJson.getBytes("UTF-8"));
                    out.flush();
                }
            } else {
                servletResponse.setContentType("application/json; charset=utf-8");
                servletResponse.setCharacterEncoding("UTF-8");
                String userJson = Resp.fail(ErrorCode.UNLOGIN_ERROR).toString();
                OutputStream out = servletResponse.getOutputStream();
                out.write(userJson.getBytes("UTF-8"));
                out.flush();
            }
        }
    
        @Override
        public void destroy() {
        }
    }