利用参数解析器HandlerMethodArgumentResolver + AOP，实现注入当前登录用户信息

最近的项目，我们需要频繁的获取APP用户信息，我们可以利用参数解析器HandlerMethodArgumentResolver + AOP，实现注入当前登录用户信息。

一：HandlerMethodArgumentResolver的介绍

public interface HandlerMethodArgumentResolver {

    /**
     * 此解析器是否支持给定的方法参数。
     * 参数：parameter - 要检查的方法参数
     * 返回值： 如果此解析器支持提供的参数，则为true ；否则false
     */
    boolean supportsParameter(MethodParameter parameter);

    /**
    * 将方法参数解析为给定请求的参数值。 ModelAndViewContainer提供对请求模型的访问。 WebDataBinderFactory提供了一种在需要进行数据绑定和类型转换时创建WeDataBinder实例的方法。
    *   参数：
    * parameter -- 要解析的方法参数。此参数必须先前已传递给必须返回true的supportsParameter 。
    * mavContainer – 当前请求的 ModelAndViewContainer
    * webRequest – 当前请求
    * binderFactory – 用于创建WebDataBinder实例的工厂
    * 返回值：
    * 解析的参数值，如果不可解析，则返回null
    * 抛出：
    * Exception ——如果参数值的准备出现错误
    */
    @Nullable
    Object resolveArgument(MethodParameter parameter, @Nullable ModelAndViewContainer mavContainer,
            NativeWebRequest webRequest, @Nullable WebDataBinderFactory binderFactory) throws Exception;

}

简单点说：大家可以看到，这个接口有两个方法，supportsParameter和resolveArgument。方法supportsParameter很好理解，返回值是boolean类型，它的作用是判断Controller层中的参数，是否满足条件，满足条件则执行resolveArgument方法，不满足则跳过。而resolveArgument方法呢，它只有在supportsParameter方法返回true的情况下才会被调用。用于处理一些业务，将返回值赋值给Controller层中的这个参数。因此呢，我们可以将HandlerMethodArgumentResolver理解为是一个参数解析器，我们可以通过写一个类实现HandlerMethodArgumentResolver接口来实现对Controller层中方法参数的修改。

二：真是项目实战

2.1 定义一个用户信息（AOP）

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

2.2 有@LoginUser注解的方法参数，注入当前登录用户

@Component
public class LoginUserHandlerMethodArgumentResolver implements HandlerMethodArgumentResolver {
    @Autowired
    private UserService userService;

    @Override
    public boolean supportsParameter(MethodParameter parameter) {
        return parameter.getParameterType().isAssignableFrom(UserEntity.class) && parameter.hasParameterAnnotation(LoginUser.class);
    }

    @Override
    public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer container,
                                  NativeWebRequest request, WebDataBinderFactory factory) {
        //获取用户ID
        Object object = request.getAttribute(AuthorizationInterceptor.USER_KEY, RequestAttributes.SCOPE_REQUEST);
        if (object == null) {
            return null;
        }
        //获取用户信息,可以考虑一下缓存。
        UserEntity user = userService.getById((Integer) object);
        return user;
    }
}

2.3 WebMvcConfig实现WebMvcConfigurer注册我们添加的拦截器、方法参数解析器等

@Configuration
public class WebMvcConfig implements WebMvcConfigurer {
    @Autowired
    private AuthorizationInterceptor authorizationInterceptor;
    @Autowired
    private LoginUserHandlerMethodArgumentResolver loginUserHandlerMethodArgumentResolver;

    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(authorizationInterceptor).addPathPatterns("/app/**");
    }

    @Override
    public void addArgumentResolvers(List<HandlerMethodArgumentResolver> argumentResolvers) {
        argumentResolvers.add(loginUserHandlerMethodArgumentResolver);
    }
}

2.4 controller 代码直接使用

    @ApiOperation("购物车列表")
    @RequireLogin
    @GetMapping("/list")
    public Result cartList(@LoginUser UserEntity user) {
        CatListVo vo = cartService.cartList(user);
        return Result.ok().data(vo);
    }