使用自定义注解获取信息

有时候会有这样的场景，前端传token，我们可以从token里解析到用户信息，然后其他地方想调用，我们可以存在ThreadLocal里，这样可以在其他地方获取用户信息，这样做太麻烦了，现在有个简单的方法，使用自定义注解，然后写个类实现HandlerMethodArgumentResolver，再配置下就能用了。

@Target(ElementType.PARAMETER)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface User {
}

赋值

/**
 * 给@User注解赋值
 **/
public class UserMethodArgumentResolver implements HandlerMethodArgumentResolver {

    @Override
    public boolean supportsParameter(MethodParameter methodParameter) {
        return methodParameter.hasParameterAnnotation(User.class);
    }

    @Override
    public Object resolveArgument(MethodParameter methodParameter, ModelAndViewContainer modelAndViewContainer, NativeWebRequest webRequest, WebDataBinderFactory webDataBinderFactory) throws Exception {
        // HttpServletRequest servletRequest = webRequest.getNativeRequest(HttpServletRequest.class);
        // servletRequest.getAttribute("userId");
        // 在这里可以根据前端传的用户ID，查库获取用户信息，获取解析token获取用户信息，其他地方使用@User注解获取用户信息
        // String token = webRequest.getParameter("token");
        // String token = webRequest.getHeader("token");
        IOCUser user = new IOCUser();
        user.setId(1);
        user.setName("cq");
        return user;
    }
}

配置

@Configuration
public class MyInterceptorConfig implements WebMvcConfigurer {
    /**
     * 自定义@User注解获取用户信息
     *
     * @param argumentResolvers
     */
    @Override
    public void addArgumentResolvers(List<HandlerMethodArgumentResolver> argumentResolvers) {
        argumentResolvers.add(new UserMethodArgumentResolver());
    }
}

1