HandlerMethodArgumentResolver 自定义参数解析

概念

• HandlerMethodArgumentResolver 是Spring MVC框架中的一个接口，用于解析Controller方法的参数。它可以将请求的参数转换成方法参数，并将其注入到Controller方法中。
• HandlerMethodArgumentResolver接口有一个方法supportsParameter，用于判断是否支持该参数类型的解析，如果返回true，则会调用resolveArgument方法进行参数解析。
• 通过实现HandlerMethodArgumentResolver接口，我们可以自定义参数解析器，从而实现自定义注解的处理或者自定义参数类型的转换等功能。

public interface HandlerMethodArgumentResolver {
 
    boolean supportsParameter(MethodParameter parameter);
    
    @Nullable
    Object resolveArgument(MethodParameter parameter, @Nullable ModelAndViewContainer mavContainer,
                           NativeWebRequest webRequest, @Nullable WebDataBinderFactory binderFactory) throws Exception;
    
}

应用

自定义注解自动注入当前登录用户

• 自定义注解

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

• 自定义注解参数解析器

@Componetn
public class CurrentUserMethodArgumentResolver implements HandlerMethodArgumentResolver {

    @Resource
    private RedisUtils redisUtils;

    public boolean supportsParameter(MethodParameter parameter) {
        // 当参数中使用了我们的自定义注解的时候
        if (parameter.hasParameterAnnotation(CurrentUser.class)) {
            return true;
        }
        return false;
    }

    public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer, NativeWebRequest webRequest, WebDataBinderFactory binderFactory) throws Exception {
        String token = httpServletRequest.getHeader("token");
        if(token == null){
        	throw new BaseException("token is null");
        }
        Object obj = redisUtils.get(token);
        if(obj == null){
        	throw new BaseException("invalid token");
        }
        return (SysUser)obj;
    }
}

• 配置类

@Configuration
public class ApplicationConfigurer extends WebMvcConfigurerAdapter {
    
    @Resource
    private CurrentUserMethodArgumentResolver currentUserMethodArgumentResolver;
    
    @Override
    public void addArgumentResolvers(List<HandlerMethodArgumentResolver> argumentResolvers) {
        super.addArgumentResolvers(argumentResolvers);
        argumentResolvers.add(currentUserMethodArgumentResolver);
    }
}

• 应用

@RequestMapping("/list")
public ResultInfo<?> list(@CurrentUser SysUser sysUser,String name){
 	// sysUser 就是当前登录用户的信息   
}