spring boot 过滤器

在做后台管理系统的时候，登录是再正常不过了。那么，在登录之后，请求一系列的接口是需要使用token的。那么，问题就出来了，在最开始的登录阶段，登录是不需要token。那我们在登录之后，请求每一个接口我们总不能每一次都在方法中去写获取请求头，然后判断token，这是不合理的。

最有效或者最有用的解决方案就是写一个过滤器，让每一个请求的接口都获取请求头，判断......这一切都在过滤器中完成。然而，在过滤器中，我们还需要踢除登录的接口。

java 复制代码

package com.bnc.s99.filter;

import com.bnc.s99.common.RedisClass;
import com.bnc.s99.utils.JwtUtil;
import jakarta.servlet.*;
import jakarta.servlet.annotation.WebFilter;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.io.IOException;
import java.io.PrintWriter;
import java.util.List;
import java.util.Map;
import java.util.Objects;

import static org.hibernate.internal.util.collections.ArrayHelper.indexOf;


@Component
public class MyFilter implements Filter {
    @Autowired
    private RedisClass redisClass;

    @Override
    public void init(FilterConfig filterConfig) {

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



        String uri = req.getRequestURI();
        //在uriArr 的数组中存放的就是不需要经过 过滤器的接口
        //也就是说，当请求这数组中的接口时就会自动跳过，不需要经过这过滤器
        String[] uriArr = new String[]{
                "/admin/login",
                "/admin/loginOut",
                "/admin/getCode",
                "/admin/test"
        };
        int index = indexOf(uriArr , uri);
        if (index != -1) {
            chain.doFilter(request, response);
            return;
        }

        String headerToken = req.getHeader("token");
        if(headerToken == null) {
            resp.setStatus(403); // 403 Forbidden
            resp.setContentType("application/json");
            resp.setCharacterEncoding("UTF-8");
            try (PrintWriter out = resp.getWriter()) {
                // 创建一个简单的JSON响应体
                String jsonResponse = "{\"error\": \"没有token\" , \"code\": \"403\" , \"data\": null}";
                out.write(jsonResponse);
            }
            return;
        }

        Boolean bol = JwtUtil.isTokenExpired(headerToken);
        if(bol){
            resp.setStatus(403); // 403 Forbidden
            resp.setContentType("application/json");
            resp.setCharacterEncoding("UTF-8");
            try (PrintWriter out = resp.getWriter()) {
                // 创建一个简单的JSON响应体
                String jsonResponse = "{\"error\": \"token已过期\" , \"code\": \"403\" , \"data\": null}";
                out.write(jsonResponse);
            }
            return;
        }

        Map userinfo = JwtUtil.parseToken(headerToken);
        String username = userinfo.get("username").toString();
        String redisToken = redisClass.getString(username);
        if(Objects.equals(redisToken, headerToken)){
            resp.setStatus(403); // 403 Forbidden
            resp.setContentType("application/json");
            resp.setCharacterEncoding("UTF-8");
            try (PrintWriter out = resp.getWriter()) {
                // 创建一个简单的JSON响应体
                String jsonResponse = "{\"error\": \"token已注销\" , \"code\": \"403\" , \"data\": null}";
                out.write(jsonResponse);
            }
            return;
        }
        chain.doFilter(request, response);
    }
    @Override
    public void destroy() {

    }
}