HandlerMethodArgumentResolver 自定义参数解析

399 阅读1分钟

概念

  • HandlerMethodArgumentResolverSpring 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 就是当前登录用户的信息   
}