spring boot 过滤器

在做后台管理系统的时候,登录是再正常不过了。那么,在登录之后,请求一系列的接口是需要使用token的。那么,问题就出来了,在最开始的登录阶段,登录是不需要token。那我们在登录之后,请求每一个接口我们总不能每一次都在方法中去写获取请求头,然后判断token,这是不合理的。
最有效或者最有用的解决方案就是写一个过滤器,让每一个请求的接口都获取请求头,判断……这一切都在过滤器中完成。然而,在过滤器中,我们还需要踢除登录的接口。

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() {

    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值