Spring boot 自定义 Resolver 支持 interface 类型参数

在编写 RestController 层的代码时，由于数据实体类定义了接口及实现类，本着面向接口编程的原则，我使用了接口作为 RestController 方法的入参。

代码大致如下（省略具体业务部分）：

（1）模型接口：

 public interface User {

     long getUserId();

     void setUserId(long userId);

     String getUserName();

     void setUserName(String userName);

     String getCategory();

     void setCategory(String category);
 }

（2）模型实现类

 public class UserImpl implements User{
     private long userId;
     private String userName;
     private String category;

     @Override
     public long getUserId() {
         return userId;
     }

     @Override
     public void setUserId(long userId) {
         this.userId = userId;
     }

     @Override
     public String getUserName() {
         return userName;
     }

     @Override
     public void setUserName(String userName) {
         this.userName = userName;
     }

     @Override
     public String getCategory() {
         return category;
     }

     @Override
     public void setCategory(String category) {
         this.category = category;
     }

 }

（3）RestController POST接口代码

     @PostMapping(value = "/updateUser", consumes = MediaType.APPLICATION_JSON_VALUE)
     public long updateUser(HttpSession session, @RequestBody User user) {
         System.out.println(session.getId());

         System.out.println(user.getUserName());
         System.out.println(user.getUserId());
         return user.getUserId();
     }

（4）前台用的axios发送的请求代码

 const AXIOS = axios.create({
   baseURL: 'http://localhost:9999',
   withCredentials: false,
   headers: {
     Accept: 'application/json',
     'Content-type': 'application/json'
   }
 })

 AXIOS.post('/updateUser', {
   userName: 'testName',
   userId: '123456789',
   category: 'XX'
 })

但在运行测试时发现 Spring boot 本身的默认中并不支持将interface或抽象类作为方法的参数。报了如下错误：

2019-09-08 19:32:22.290 ERROR 12852 --- [nio-9999-exec-9] o.a.c.c.C.[.[.[/].[dispatcherServlet]    
: Servlet.service() for servlet [dispatcherServlet] in context with path [] threw exception 
[Request processing failed; nested exception is org.springframework.http.converter.HttpMessageConversionException: 
Type definition error: [simple type, class com.sample.demo.model.User]; nested exception is com.fasterxml.jackson.databind
.exc.InvalidDefinitionException: Cannot construct instance of `com.sample.demo.model.User` (no Creators, like default 
construct, exist): abstract types either need to be mapped to concrete types, have custom deserializer, 
or contain additional type information at [Source: (PushbackInputStream); line: 1, column: 1]] with root cause
...

大致意思时不存在创建实例的构造函数，抽象类型需要配置映射到具体的实现类。

解决方案一：

于是我上网搜了下解决方法，最终在 StackOverflow 上找到一种解决方案：

@JsonTypeInfo(use = Id.NAME, include = As.PROPERTY, property = "type")
@JsonSubTypes({@JsonSubTypes.Type(value = A.class, name = "A"),
        @JsonSubTypes.Type(value = B.class, name = "B")})
public interface MyInterface {

}

通过添加注解的方式，将接口映射到实现类。

这种方法可以解决方法入参为接口的问题，但同时又会引入一个问题：接口和实现类相互引用，导致循环依赖。而且如果我有很多数据类的接口及实现类的话，每个接口都要写一遍注解。

于是继续探索。。。

解决方案二：

继承 HandlerMethodArgumentResolver  接口实现里面的 supportsParameter  和  resolveArgument 方法。

（1）在supportsParameter 方法中返回支持的类型。其中MODEL_PATH为实体类的包路径，下列代码中默认支持了包内的所有类型。

     @Override
     public boolean supportsParameter(MethodParameter parameter) {
         return parameter.getParameterType().getName().startsWith(MODEL_PATH);
     }

（2）在 resolveArgument 方法中，通过反射生成一个实现类的对象并返回。

  @Override
     public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer modelAndViewContainer,
                                   NativeWebRequest nativeWebRequest, WebDataBinderFactory webDataBinderFactory) throws Exception {
         Class<?> parameterType = parameter.getParameterType();
         String implName = parameterType.getName() + SUFFIX;
         Class<?> implClass = Class.forName(implName);

         if (!parameterType.isAssignableFrom(implClass)) {
             throw new IllegalStateException("type error:" + parameterType.getName());
         }

         Object impl = implClass.newInstance();
         WebDataBinder webDataBinder = webDataBinderFactory.createBinder(nativeWebRequest, impl, parameter.getParameterName());
         ServletRequest servletRequest = nativeWebRequest.getNativeRequest(ServletRequest.class);
         Assert.notNull(servletRequest, "servletRequest is null.");

         ServletRequestDataBinder servletRequestDataBinder = (ServletRequestDataBinder) webDataBinder;
         servletRequestDataBinder.bind(servletRequest);
         return impl;
     }

（3）最后添加到Spring boot 的配置中

     @Bean
     public WebMvcConfigurer webMvcConfigurer() {
         return new WebMvcConfigurerAdapter() {
             @Override
             public void addArgumentResolvers(List<HandlerMethodArgumentResolver> argumentResolvers) {
                 argumentResolvers.add(new MethodInterfaceArgumentResolver());
                 super.addArgumentResolvers(argumentResolvers);
             }
         };
     }

方案二可以解决找不到构造函数的问题，运行不会报错，也不会导致循环依赖，但却没法将前台的数据注入到入参对象中。也就是给方法传入的只是一个刚new出来的UserImpl 对象。

经过测试发现，虽然对post请求无法注入前台数据，但对于get请求，还是可以的：

前台get方法代码：

AXIOS.get('/getUser?userName=Haoye&userId=123456789&category=XX')

后台get方法代码：

     @GetMapping("/getUser")
     public User getUser(User user) {
         System.out.println(user.getUserName());
         return user;
     }

解决方案三：

由于在网上没有找到好的解决方案，我最后通过看Spring boot 源码 + 调试跟踪 + 写demo尝试的方式，终于找到了好的解决方案。

这里先分享下大致的思路：

（1）Spring boot的相关代码应该在 HandlerMethodArgumentResolver 接口对应的包里或者附近。但这样找还是比较慢，因为代码还是很多。

（2）通过打断点，看看哪里调用了 public boolean supportsParameter(MethodParameter parameter) 方法。

于是找到了HandlerMethodArgumentResolverComposite 类调用的地方：

从上图可以看到，当前处理的是第一个参数HttpSession。

（3）先将controller方法的入参先改为UserImpl，也就是实现类，在步骤（2）的截图对应的代码中打断点。

继续调试，找到Spring boot 解析被@RequestBody 注解标注的参数UserImpl user 的时候，用的是什么Resolver。

如下图所示，调用Evaluate窗口获取类型信息，点击 Navigate 跳转到对应的类 RequestResponseBodyMethodProcessor。

（4） RequestResponseBodyMethodProcessor 类中的 resolveArgument 方法源码如下：

     /**
      * Throws MethodArgumentNotValidException if validation fails.
      * @throws HttpMessageNotReadableException if {@link RequestBody#required()}
      * is {@code true} and there is no body content or if there is no suitable
      * converter to read the content with.
      */
     @Override
     public Object resolveArgument(MethodParameter parameter, @Nullable ModelAndViewContainer mavContainer,
             NativeWebRequest webRequest, @Nullable WebDataBinderFactory binderFactory) throws Exception {

         parameter = parameter.nestedIfOptional();
         Object arg = readWithMessageConverters(webRequest, parameter, parameter.getNestedGenericParameterType());
         String name = Conventions.getVariableNameForParameter(parameter);

         if (binderFactory != null) {
             WebDataBinder binder = binderFactory.createBinder(webRequest, arg, name);
             if (arg != null) {
                 validateIfApplicable(binder, parameter);
                 if (binder.getBindingResult().hasErrors() && isBindExceptionRequired(binder, parameter)) {
                     throw new MethodArgumentNotValidException(parameter, binder.getBindingResult());
                 }
             }
             if (mavContainer != null) {
                 mavContainer.addAttribute(BindingResult.MODEL_KEY_PREFIX + name, binder.getBindingResult());
             }
         }

         return adaptArgumentIfNecessary(arg, parameter);
     }

回到最初的问题，导致无法传入interface类型参数的原因是接口无法实例化。那既然如此，我们要修改的地方肯定是Spring boot 尝试实例化接口的地方，也就是实例化失败进而抛出异常的地方。

一路顺腾摸瓜，最终发现 readWithMessageConverters 方法中, 通过给 readWithMessageConverters 方法传入类型信息，最终生成参数实例。

（5） 从（4）中可以看到，相关方法的访问级别为 protected，也就是我们可以通过继承 RequestResponseBodyMethodProcessor 并覆写 readWithMessageConverters 即可。

通过反射，注入 User 接口的实现类型 UserImpl 的class：

 package com.sample.demo.config;

 import org.springframework.core.MethodParameter;
 import org.springframework.http.converter.HttpMessageConverter;
 import org.springframework.http.converter.HttpMessageNotReadableException;
 import org.springframework.web.HttpMediaTypeNotSupportedException;
 import org.springframework.web.context.request.NativeWebRequest;
 import org.springframework.web.servlet.mvc.method.annotation.RequestResponseBodyMethodProcessor;

 import java.io.IOException;
 import java.lang.reflect.Type;
 import java.util.List;

 /**
  * @breaf
  * @author https://cnblogs.com/laishenghao
  * @date 2019/9/7
  * @since 1.0
  **/
 public class ModelRequestBodyMethodArgumentResolver extends RequestResponseBodyMethodProcessor {
     private static final String MODEL_PATH = "com.sample.demo.model";
     private static final String SUFFIX = "Impl";

     public ModelRequestBodyMethodArgumentResolver(List<HttpMessageConverter<?>> converters) {
         super(converters);
     }

     @Override
     public boolean supportsParameter(MethodParameter methodParameter) {
         return super.supportsParameter(methodParameter)
             && methodParameter.getParameterType().getName().startsWith(MODEL_PATH);
     }

     @Override
     protected <T> Object readWithMessageConverters(NativeWebRequest webRequest, MethodParameter parameter, Type paramType)
             throws IOException, HttpMediaTypeNotSupportedException, HttpMessageNotReadableException {
         try {
             Class<?> clazz = Class.forName(paramType.getTypeName() + SUFFIX);
             return super.readWithMessageConverters(webRequest, parameter, clazz);
         } catch (ClassNotFoundException e) {
             return null;
         }
     }

 }

完成上面的代码后，跑了一下，发现并没有什么用，报的错误还是跟最开始的一样。

由此推测，应该是Spring boot 默认配置的 Resolver的优先级比较高，导致我们自定义的并没有生效。

于是继续查找原因，发现自定义的Resolver的优先级几乎垫底了，在远未调用到之前就被它的父类抢了去。

（6）提高自定义 Resolver的优先级。

一个可行的方法是：在Spring boot 框架初始化完成后，获取到所有的Resolver，然后将自定义的加在ArrayList的前面。

 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 import org.springframework.web.method.support.HandlerMethodArgumentResolver;
 import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
 import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;
 import org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerAdapter;

 import javax.annotation.PostConstruct;
 import java.util.ArrayList;
 import java.util.List;

 /**
  * @breaf
  * @blog https://www.cnblogs.com/laishenghao
  * @date 2019/9/7
  * @since 1.0
  **/
 @Configuration
 public class CustomConfigurations {
     @Autowired
     private RequestMappingHandlerAdapter adapter;

     @PostConstruct
     public void prioritizeCustomArgumentMethodHandlers () {
         List<HandlerMethodArgumentResolver> allResolvers = adapter.getArgumentResolvers();
         if (allResolvers == null) {
             allResolvers = new ArrayList<>();
         }
         List<HandlerMethodArgumentResolver> customResolvers = adapter.getCustomArgumentResolvers ();
         if (customResolvers == null) {
             customResolvers = new ArrayList<>();
         }
         ModelRequestBodyMethodArgumentResolver argumentResolver = new ModelRequestBodyMethodArgumentResolver(adapter.getMessageConverters());
         customResolvers.add(0,argumentResolver);

         List<HandlerMethodArgumentResolver> argumentResolvers = new ArrayList<> (allResolvers);
         argumentResolvers.removeAll (customResolvers);
         argumentResolvers.addAll (0, customResolvers);
         adapter.setArgumentResolvers (argumentResolvers);
     }
 }

值得注意的是，getResolvers()方法返回的是不可更改的List，不能直接插入。

至此，自定义参数处理器就可以解析RestController标注的类中的方法的 interface类型参数了。

如果要支持其他类型（比如抽象类、枚举类），或者使用自定义注解标注入参，也可以通过类似的方法来实现。

本文地址：https://www.cnblogs.com/laishenghao/p/11488724.html