Springboot项目登录校验功能实现

// 设置Cookie示例
@GetMapping("/setCookie")
public String setCookie(HttpServletResponse response) {
    // 创建一个名为"userId"的Cookie，值为"12345"，有效期为1天
    Cookie cookie = new Cookie("userId", "12345");
    cookie.setMaxAge(24 * 60 * 60); // 设置Cookie有效期（秒）
    response.addCookie(cookie); // 将Cookie添加到响应中
    return "Cookie设置成功";
}
// 读取Cookie示例
@GetMapping("/getCookie")
public String getCookie(HttpServletRequest request) {
    Cookie[] cookies = request.getCookies(); // 获取所有Cookie
    if (cookies != null) {
        for (Cookie cookie : cookies) {
            if ("userId".equals(cookie.getName())) { // 查找名为"userId"的Cookie
                return "找到的userId: " + cookie.getValue();
            }
        }
    }
    return "未找到userId Cookie";
}

@RestController
@Slf4j
public class SessionController {
    // 设置Session示例
    @GetMapping("/setSession")
    public String setSession(HttpSession session) {
        session.setAttribute("username", "john_doe"); // 向Session中存储数据
        log.info("Session ID: {}", session.getId()); // 打印Session ID
        return "Session设置成功";
    }
    // 获取Session示例
    @GetMapping("/getSession")
    public String getSession(HttpSession session) {
        String username = (String) session.getAttribute("username"); // 从Session中获取数据
        if (username != null) {
            return "当前登录用户: " + username;
        } else {
            return "Session中未找到用户名";
        }
    }
}

<!--JWT-->
<dependency>
    <groupId>io.jsonwebtoken</groupId>
    <artifactId>jjwt</artifactId>
    <version>0.9.1</version>
</dependency>

// 引入JWT依赖后，使用Jwts工具类生成和校验JWT令牌
public class JwtUtils {
    private static String signKey = "5L2g5aW977yM5pyL5Y+L77yB"; // 签名密钥（自行更换）
    private static Long expire = 43200000L; // 令牌有效期（毫秒）
    // 生成JWT令牌
    public static String generateJwt(Map<String, Object> claims) {
        String jwt = Jwts.builder()
                .addClaims(claims) // 添加自定义信息
                .signWith(SignatureAlgorithm.HS256, signKey) // 使用HS256算法签名
                .setExpiration(new Date(System.currentTimeMillis() + expire)) // 设置令牌有效期
                .compact(); // 生成令牌字符串
        return jwt;
    }
    // 校验JWT令牌
    public static Claims parseJWT(String jwt) {
        Claims claims = Jwts.parser()
                .setSigningKey(signKey) // 设置签名密钥
                .parseClaimsJws(jwt) // 解析令牌
                .getBody(); // 获取令牌中的信息
        return claims;
    }
}

@Slf4j
@WebFilter(urlPatterns = "/*") // 拦截所有请求
public class TokenFilter implements Filter {
    @Override
    public void doFilter(ServletRequest req, ServletResponse resp, FilterChain chain) 
            throws IOException, ServletException {
        HttpServletRequest request = (HttpServletRequest) req;
        HttpServletResponse response = (HttpServletResponse) resp;
        String url = request.getRequestURL().toString();
        if (url.contains("login")) { // 如果是登录请求，直接放行
            chain.doFilter(request, response);
            return;
        }
        String jwt = request.getHeader("token"); // 从请求头中获取JWT令牌
        if (!StringUtils.hasLength(jwt)) { // 令牌为空，返回未授权错误
            response.setStatus(HttpStatus.SC_UNAUTHORIZED);
            return;
        }
        try {
            JwtUtils.parseJWT(jwt); // 校验令牌
        } catch (Exception e) {
            response.setStatus(HttpStatus.SC_UNAUTHORIZED); // 令牌校验失败，返回未授权错误
            return;
        }
        chain.doFilter(request, response); // 令牌校验成功，放行请求
    }
}

@Slf4j
@Component
public class TokenInterceptor implements HandlerInterceptor {
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) 
            throws Exception {
        String url = request.getRequestURL().toString();
        if (url.contains("login")) { // 如果是登录请求，直接放行
            return true;
        }
        String jwt = request.getHeader("token"); // 从请求头中获取JWT令牌
        if (!StringUtils.hasLength(jwt)) { // 令牌为空，返回未授权错误
            response.setStatus(HttpStatus.SC_UNAUTHORIZED);
            return false;
        }
        try {
            JwtUtils.parseJWT(jwt); // 校验令牌
        } catch (Exception e) {
            response.setStatus(HttpStatus.SC_UNAUTHORIZED); // 令牌校验失败，返回未授权错误
            return false;
        }
        return true; // 令牌校验成功，放行请求
    }
}

@Configuration
public class WebConfig implements WebMvcConfigurer {
    @Autowired
    private TokenInterceptor tokenInterceptor;
    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(tokenInterceptor) // 注册拦截器
                .addPathPatterns("/**"); // 拦截所有请求
    }
}