使用自定义注解获取信息

有时候会有这样的场景,前端传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

最后修改:2020 年 09 月 15 日 09 : 20 PM