使用HandlerMethodArgumentResolver给Controller注入会员信息

1、在aop中进行校验登录，校验成功后将会员信息保存到ThreadLocal，调用

UserContext.setUser(user)方法;

UserContext 类：

public class UserContext {

    private static ThreadLocal userContext = new ThreadLocal();

   public static void setUser(User user){
       userContext.set(user);
   }

   public static User getUser(){
       return userContext.get();
   }
}

2、写一个类实现HandlerMethodArgumentResolver，有两个实现方法，当supportsParameter返回true时，才会调用resolveArgument方法。

 @Configuration
public class UserArgumentResolver implements HandlerMethodArgumentResolver {
    @Override
    public boolean supportsParameter(MethodParameter methodParameter) {
        Class clazz = methodParameter.getParameterType();
        return clazz == User.class;

    }

    @Override
    public Object resolveArgument(MethodParameter methodParameter, ModelAndViewContainer modelAndViewContainer, NativeWebRequest nativeWebRequest, WebDataBinderFactory webDataBinderFactory) throws Exception {
        return UserContext.getUser();
    }
}

 

3、将第二步的UserArgumentResolver加入到WebMvcConfigurerAdapter中

@Configuration
public class WebConfig extends WebMvcConfigurerAdapter {
    @Autowired
     UsergumentResolver userArgumentResolver;

    @Override
    public void addArgumentResolvers(List argumentResolvers) {
        argumentResolvers.add(userArgumentResolver);
    }
}

4、在Controller的方法参数中直接获得该线程绑定的用户信息 

@RequestMapping("user")
public User getUser(User user){
   
    return user;
}

发现问题？

4、resolveArgument方法中UserContext.getUser();获取不到会员信息为空，导致controller的User为空，经过调试发现

是执行顺序的问题:

HandlerMethodArgumentResolver  ---->aop--------------->controller。HandlerMethodArgumentResolver  快于aop执行

解决问题：

使用拦截器，执行顺序为

拦截器------------->HandlerMethodArgumentResolver -------------controller。

 

注意点：在传统的项目中需要在spring-mvc.xml加入

springboot项目中UserArgumentResolver加个@Service注解

 

WebMvcConfigurerAdapter 中 这个抽象类中的方法可以看一看，addInterceptors可以添加拦截器

@Override
public void addInterceptors(InterceptorRegistry registry) {
}