拦截器，获取登录用户信息

改进获取登录用户的方法

1.自定义注解  创建新的@interface

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

/**
 * @author buXianXian
 */
@Retention(RetentionPolicy.RUNTIME)//运行时生效
@Target(ElementType.METHOD)//加载到方法上
public @interface RequiredToken {
}

2.自定义TokenInterceptors类实现HandlerInterceptor中的preHandle---注意@Component注入

/**
 * @author buXianXian
 *
 */
@Component
public class TokenInterceptor implements HandlerInterceptor {
    private static final Logger log = LoggerFactory.getLogger(TokenInterceptor.class);
    private RedisTemplate redisTemplate;

    @Autowired
    public TokenInterceptor(RedisTemplate redisTemplate) {
        this.redisTemplate = redisTemplate;
    }

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        //1.获取当前请求方法上是否存在RequiredToken注解。如果有，说明需要先登录，才能调用该功能
        //需要自己编写注解
        boolean requiredToken = ((HandlerMethod) handler).getMethod().isAnnotationPresent(RequiredToken.class);
        //2.存在该注解
        if (requiredToken) {
            //1.获取token信息
            String token = request.getParameter("token");
            log.info("当前传递的token为: {}", token);
            //2.拼接前缀
            String tokenKey = "session_" + token;
            //3.根据tokenKey获取用户信息
            User o = (User) redisTemplate.opsForValue().get(tokenKey);
            if (o == null) {
                throw new RuntimeException("提示: 令牌无效,无效token!");
            }
            //4.存储到当前请求的上下文中
            request.setAttribute("token", token);
            request.setAttribute("user", o);
        }
        return true;
    }
}

3.配置自定义的拦截器---config

/**
 * @author buXianXian
 * 自定义拦截器
 */
@Configuration
public class MvcConfig implements WebMvcConfigurer {
    //加入自定义的拦截器
    private TokenInterceptor tokenInterceptor;

    @Autowired
    public MvcConfig(TokenInterceptor tokenInterceptor) {
        this.tokenInterceptor = tokenInterceptor;
    }

    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(tokenInterceptor)
                .addPathPatterns("/**");//拦截所有
    }