SpringBoot集成JWT

1.导入依赖

		<dependency>
			<groupId>com.auth0</groupId>
			<artifactId>java-jwt</artifactId>
			<version>3.10.3</version>
		</dependency>

2.TokenUtil（token生成）

@Component
public class TokenUtils {

    private static IUserService staticUserService;

    @Resource
    private IUserService userService;

    @PostConstruct
    public void setUserService(){
        staticUserService = userService;
    }
    /**
     * 生成token
     * @param userId
     * @return
     */
    public static String genToken(String userId,String sign){
        return JWT.create().withAudience(userId) // 将 user id 保存到 token 里面
                .withExpiresAt(DateUtil.offsetHour(new Date(),2)) //两小时后token过期
                .sign(Algorithm.HMAC256(sign)); // 以 password 作为 token 的密钥
                //使用提供的userId和sign生成一个token。它使用JWT.create()方法创建一个新的JWT实例，
        // 将userId设置为token的受众（audience），将过期时间设置为当前时间之后两小时，
        // 使用HMAC256算法和提供的sign作为密钥对token进行签名。最后，它返回生成的token。
    }

    /**
     * 获取当前登陆的用户信息
     * @return
     */
    public static User getCurrentUser(){
        /**
         * 获取当前登录用户的信息。它首先使用RequestContextHolder.getRequestAttributes()获取当前的HTTP请求，
         * 然后从请求中获取名为"token"的头部信息。如果token不为空，它解码token以获取受众（audience）中的userId，
         * 然后使用staticUserService根据userId获取用户信息并返回。如果在这个过程中发生任何异常，它将返回null。
         */

            try {
                HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
                String token = request.getHeader(("token"));
                if(StringUtils.isEmpty(token)) {
                    String userId = JWT.decode(token).getAudience().get(0);
                    return staticUserService.getById(Integer.valueOf(userId));
                }
            }catch (Exception e){
                return null;
            }

        return null;
    }
}

3.拦截器验证token

/**
 * 实现了HandlerInterceptor接口，用于拦截请求并进行JWT验证。
 */
public class JwtInterceptor implements HandlerInterceptor {

    @Autowired
    private IUserService userService;

    /**
     * 请求处理之前进行拦截和处理。
     * @param request
     * @param response
     * @param handler
     * @return
     * @throws Exception
     */
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        //请求头中获取名为"token"的值
        String token = request.getHeader("token");
        // 如果不是映射到方法直接通过
        if(!(handler instanceof HandlerMethod)){
            return true;
        }
        if (StringUtils.isEmpty(token)) {
            throw new ServiceException(Constants.CODE_401,"无token，请重新登录");
        }
        //解码token，获取其中的受众（audience）信息，将其赋给userId变量
        String userId;
        try {
            userId = JWT.decode(token).getAudience().get(0);
        } catch (JWTDecodeException j) {
            throw new ServiceException(Constants.CODE_401,"token验证失败，请重新登录");
        }
        //根据token中的userId查询数据库
        User user = userService.getById(userId);
        if (user == null) {
            throw new ServiceException(Constants.CODE_401,"用户不存在，请重新登录");
        }

        // 创建一个JWTVerifier实例，使用用户的密码作为密钥，通过调用verify方法验证token的有效性。
        JWTVerifier jwtVerifier = JWT.require(Algorithm.HMAC256(user.getPassword())).build();
        try {
            jwtVerifier.verify(token);//验证token
        } catch (JWTVerificationException e) {
            throw new ServiceException(Constants.CODE_401,"token验证失败，请重新登录");
        }

        return true;
    }
}

4.注册拦截器

/**
 * 配置拦截器
 */
@Configuration
public class InterceptorConfig implements WebMvcConfigurer {

    /**
     * 这个方法是WebMvcConfigurer接口的addInterceptors方法的实现，用于添加拦截器。
     * 它通过调用registry.addInterceptor(jwtInterceptor())方法将JwtInterceptor拦截器添加到拦截器注册表中。
     * 然后，使用.addPathPatterns("/**")方法指定拦截所有请求。
     * 最后，使用.excludePathPatterns()指定排除的路径，这些路径不会被拦截器处理。这样，通过判断token是否合法，决定是否需要登录。
     *
     */

    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(jwtInterceptor())
                .addPathPatterns("/**")
                .excludePathPatterns("/user/login","/user/register");    // 拦截所有请求，通过判断token是否合法决定需要登录
    }

    /**
     * 使用@Bean注解，将JwtInterceptor实例化为一个Spring bean，
     * 并将其返回。这样其他组件就可以通过依赖注入获取JwtInterceptor的实例。
     * @return
     */
    @Bean
    public JwtInterceptor jwtInterceptor(){
        return new JwtInterceptor();
    }
}