SpringMVC自定义MethodArgumentResolver学习笔记

  1. 编写MethodArgumentResolver实现类
    public class PropertiesHandlerMethodArgumentResolver implements HandlerMethodArgumentResolver {
          
          
        @Override
        public boolean supportsParameter(MethodParameter parameter) {
          
          
            return Properties.class.isAssignableFrom(parameter.getParameterType());
        }
        @Override
        public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer, NativeWebRequest webRequest, WebDataBinderFactory binderFactory) throws Exception {
          
          
            ServletWebRequest servletWebRequest = (ServletWebRequest)webRequest ;
            HttpServletRequest request = servletWebRequest.getRequest();
            String contentType = request.getHeader("Content-Type") ;
            MediaType mediaType = MediaType.parseMediaType(contentType);
            Charset charset = mediaType.getCharset();
            charset = charset == null ? Charset.forName("UTF-8") : charset ;
            // 请求输入字节流
            ServletInputStream inputStream = request.getInputStream();
            InputStreamReader reader = new InputStreamReader(inputStream,charset);
            Properties properties = new Properties() ;
            // 加载字符流成为Properties对象
            properties.load(reader);
            return properties;
        }
    }
    
  2. 编写配置类将MethodArgumentResolver注册到RequestMappingHandlerAdapter中
    @Configuration
    public class RestWebMvcConfigurer implements WebMvcConfigurer {
          
          
        @Autowired
        private RequestMappingHandlerAdapter requestMappingHandlerAdapter ;
        @PostConstruct
        public void init(){
          
          
            // 获取当前RequestMappingHandlerAdapter中所有的Resolver对象
            //HandlerMethodArgumentResolverComposite.getResolvers()获取集合不可变,所以这里需要新建List
            List<HandlerMethodArgumentResolver> resolvers =
                    requestMappingHandlerAdapter.getArgumentResolvers();
            List<HandlerMethodArgumentResolver> newResolvers = new ArrayList<>(resolvers.size() + 1) ;
            newResolvers.add(new PropertiesHandlerMethodArgumentResolver());
            newResolvers.addAll(resolvers) ;
            // 重置Resolver对象
            requestMappingHandlerAdapter.setArgumentResolvers(newResolvers);
        }
        @Override
        public void addArgumentResolvers(List<HandlerMethodArgumentResolver> resolvers) {
          
          
            // 自定义HandlerMethodArgumentResolver优先级低于内建HandlerMethodArgumentResolver,
            // 详情见:
            // RequestMappingHandlerAdapter.afterPropertiesSet() => {
          
          
            //    List<HandlerMethodArgumentResolver> resolvers = getDefaultArgumentResolvers();
            //    new HandlerMethodArgumentResolverComposite().addResolvers(resolvers)
            // }
            // 所以这里添加了也没用,参数将被MapMethodProcessor处理,因为Properties extends Hashtable
            // 这里调整到@PostConstruct注解的init()中处理
            //resolvers.add(new PropertiesHandlerMethodArgumentResolver());
        }
    }
    
  3. 编写Controller业务方法,注意这里无@RequestBody注解
    @Slf4j
    @RestController
    public class PropertiesRestController {
          
          
        @PostMapping(value = "/add/props",
            consumes = "text/properties;charset=UTF-8")
        public Properties addProp(Properties properties){
          
          
            log.info("====> prop : {}" , properties);
            return properties ;
        }
    }
    
  4. 使用PostMan发送请求,将Header的Content-Type设置为text/properties
    4.1 header内容
        Content-Type设置为text/properties
    4.1 body内容
        username=yicj
        addr=北京
    

猜你喜欢

转载自blog.csdn.net/yichengjie_c/article/details/114293687