前言

前幾天筆者在寫Rest接口的時候，看到了一種傳值方式是以前沒有寫過的，就萌生了一探究竟的想法。在此之前，有篇文章曾涉及到這個話題，但那篇文章著重于處理流程的分析，并未深入。

本文重點來看幾種傳參方式，看看它們都是如何被解析并應(yīng)用到方法參數(shù)上的。

一、HTTP請求處理流程

不論在SpringBoot還是SpringMVC中，一個HTTP請求會被DispatcherServlet類接收，它本質(zhì)是一個Servlet，因為它繼承自HttpServlet。在這里，Spring負(fù)責(zé)解析請求，匹配到Controller類上的方法，解析參數(shù)并執(zhí)行方法，最后處理返回值并渲染視圖。

image

我們今天的重點在于解析參數(shù)，對應(yīng)到上圖的目標(biāo)方法調(diào)用這一步驟。既然說到參數(shù)解析，那么針對不同類型的參數(shù)，肯定有不同的解析器。Spring已經(jīng)幫我們注冊了一堆這東西。

它們有一個共同的接口HandlerMethodArgumentResolver。supportsParameter用來判斷方法參數(shù)是否可以被當(dāng)前解析器解析，如果可以就調(diào)用resolveArgument去解析。

public interface HandlerMethodArgumentResolver {
    //判斷方法參數(shù)是否可以被當(dāng)前解析器解析
    boolean supportsParameter(MethodParameter var1);
    //解析參數(shù)
    @Nullable
    Object resolveArgument(MethodParameter var1, 
            @Nullable ModelAndViewContainer var2, 
            NativeWebRequest var3, 
            @Nullable WebDataBinderFactory var4)throws Exception;
}

二、RequestParam

在Controller方法中，如果你的參數(shù)標(biāo)注了RequestParam注解，或者是一個簡單數(shù)據(jù)類型。

@RequestMapping("/test1")
@ResponseBody
public String test1(String t1, @RequestParam(name = "t2",required = false) String t2,HttpServletRequest request){
    logger.info("參數(shù):{},{}",t1,t2);
    return "Java";
}

我們的請求路徑是這樣的：http://localhost:8080/test1?t1=Jack&t2=Java

如果按照以前的寫法，我們直接根據(jù)參數(shù)名稱或者RequestParam注解的名稱從Request對象中獲取值就行。比如像這樣：

String parameter = request.getParameter("t1");

在Spring中，這里對應(yīng)的參數(shù)解析器是RequestParamMethodArgumentResolver。與我們的想法差不多，就是拿到參數(shù)名稱后，直接從Request中獲取值。

protected Object resolveName(String name, MethodParameter parameter, 
        NativeWebRequest request) throws Exception {
        
    HttpServletRequest servletRequest = request.getNativeRequest(HttpServletRequest.class);
    //...省略部分代碼...
    if (arg == null) {
        String[] paramValues = request.getParameterValues(name);
        if (paramValues != null) {
            arg = paramValues.length == 1 ? paramValues[0] : paramValues;
        }
    }
    return arg;
}

三、RequestBody

如果我們需要前端傳輸更多的參數(shù)內(nèi)容，那么通過一個POST請求，將參數(shù)放在Body中傳輸是更好的方式。當(dāng)然，比較友好的數(shù)據(jù)格式當(dāng)屬JSON。

image

面對這樣一個請求，我們在Controller方法中可以通過RequestBody注解來接收它，并自動轉(zhuǎn)換為合適的Java Bean對象。

@ResponseBody
@RequestMapping("/test2")
public String test2(@RequestBody SysUser user){
    logger.info("參數(shù)信息:{}",JSONObject.toJSONString(user));
    return "Hello";
}

在沒有Spring的情況下，我們考慮一下如何解決這一問題呢？

首先呢，還是要依靠Request對象。對于Body中的數(shù)據(jù)，我們可以通過request.getReader()方法來獲取，然后讀取字符串，最后通過JSON工具類再轉(zhuǎn)換為合適的Java對象。

比如像下面這樣：

@RequestMapping("/test2")
@ResponseBody
public String test2(HttpServletRequest request) throws IOException {
    BufferedReader reader = request.getReader();
    StringBuilder builder = new StringBuilder();
    String line;
    while ((line = reader.readLine()) != null){
        builder.append(line);
    }
    logger.info("Body數(shù)據(jù)：{}",builder.toString());
    SysUser sysUser = JSONObject.parseObject(builder.toString(), SysUser.class);
    logger.info("轉(zhuǎn)換后的Bean：{}",JSONObject.toJSONString(sysUser));
    return "Java";
}

當(dāng)然，在實際場景中，上面的SysUser.class需要動態(tài)獲取參數(shù)類型。

在Spring中，RequestBody注解的參數(shù)會由RequestResponseBodyMethodProcessor類來負(fù)責(zé)解析。

它的解析由父類AbstractMessageConverterMethodArgumentResolver負(fù)責(zé)。整個過程我們分為三個步驟來看。

1、獲取請求輔助信息

在開始之前需要先獲取請求的一些輔助信息，比如HTTP請求的數(shù)據(jù)格式，上下文Class信息、參數(shù)類型Class、HTTP請求方法類型等。

protected <T> Object readWithMessageConverters(){
                   
    boolean noContentType = false;
    MediaType contentType;
    try {
        contentType = inputMessage.getHeaders().getContentType();
    } catch (InvalidMediaTypeException var16) {
        throw new HttpMediaTypeNotSupportedException(var16.getMessage());
    }
    if (contentType == null) {
        noContentType = true;
        contentType = MediaType.APPLICATION_OCTET_STREAM;
    }
    Class<?> contextClass = parameter.getContainingClass();
    Class<T> targetClass = targetType instanceof Class ? (Class)targetType : null;
    if (targetClass == null) {
        ResolvableType resolvableType = ResolvableType.forMethodParameter(parameter);
        targetClass = resolvableType.resolve();
    }
    HttpMethod httpMethod = inputMessage instanceof HttpRequest ?
            ((HttpRequest)inputMessage).getMethod() : null; 

    //.......
}

2、確定消息轉(zhuǎn)換器

上面獲取到的輔助信息是有作用的，就是要確定一個消息轉(zhuǎn)換器。消息轉(zhuǎn)換器有很多，它們的共同接口是HttpMessageConverter。在這里，Spring幫我們注冊了很多轉(zhuǎn)換器，所以需要循環(huán)它們，來確定使用哪一個來做消息轉(zhuǎn)換。

如果是JSON數(shù)據(jù)格式的，會選擇MappingJackson2HttpMessageConverter來處理。它的構(gòu)造函數(shù)正是指明了這一點。

public MappingJackson2HttpMessageConverter(ObjectMapper objectMapper) {
    super(objectMapper, new MediaType[]{
        MediaType.APPLICATION_JSON, 
        new MediaType("application", "*+json")});
}

3、解析

既然確定了消息轉(zhuǎn)換器，那么剩下的事就很簡單了。通過Request獲取Body，然后調(diào)用轉(zhuǎn)換器解析就好了。

protected <T> Object readWithMessageConverters(){
    if (message.hasBody()) {
      HttpInputMessage msgToUse = this.getAdvice().beforeBodyRead(message, parameter, targetType, converterType);
      body = genericConverter.read(targetType, contextClass, msgToUse);
      body = this.getAdvice().afterBodyRead(body, msgToUse, parameter, targetType, converterType);
    }
}

再往下就是Jackson包的內(nèi)容了，不再深究。雖然寫出來的過程比較啰嗦，但實際上主要就是為了尋找兩個東西：

方法解析器RequestResponseBodyMethodProcessor

消息轉(zhuǎn)換器MappingJackson2HttpMessageConverter

都找到之后調(diào)用方法解析即可。

四、GET請求參數(shù)轉(zhuǎn)換Bean

還有一種寫法是這樣的，在Controller方法上用Java Bean接收。

@RequestMapping("/test3")
@ResponseBody
public String test3(SysUser user){
    logger.info("參數(shù)：{}",JSONObject.toJSONString(user));
    return "Java";
}

然后用GET方法請求：

http://localhost:8080/test3?id=1001&name=Jack&password=1234&address=北京市海淀區(qū)

URL后面的參數(shù)名稱對應(yīng)Bean對象里面的屬性名稱，也可以自動轉(zhuǎn)換。那么，這里它又是怎么做的呢 ？

筆者首先想到的就是Java的反射機制。從Request對象中獲取參數(shù)名稱，然后和目標(biāo)類上的方法一一對應(yīng)設(shè)置值進(jìn)去。

比如像下面這樣：

public String test3(SysUser user,HttpServletRequest request)throws Exception {
    //從Request中獲取所有的參數(shù)key 和 value
    Map<String, String[]> parameterMap = request.getParameterMap();
    Iterator<Map.Entry<String, String[]>> iterator = parameterMap.entrySet().iterator();
    //獲取目標(biāo)類的對象
    Object target = user.getClass().newInstance();
    Field[] fields = target.getClass().getDeclaredFields();
    while (iterator.hasNext()){
        Map.Entry<String, String[]> next = iterator.next();
        String key = next.getKey();
        String value = next.getValue()[0];
        for (Field field:fields){
            String name = field.getName();
            if (key.equals(name)){
                field.setAccessible(true);
                field.set(target,value);
                break;
            }
        }
    }
    logger.info("userInfo:{}",JSONObject.toJSONString(target));
    return "Python";
}

除了反射，Java還有一種內(nèi)省機制可以完成這件事。我們可以獲取目標(biāo)類的屬性描述符對象，然后拿到它的Method對象， 通過invoke來設(shè)置。

private void setProperty(Object target,String key,String value) {
    try {
        PropertyDescriptor propDesc = new PropertyDescriptor(key, target.getClass());
        Method method = propDesc.getWriteMethod();
        method.invoke(target, value);
    } catch (Exception e) {
        e.printStackTrace();
    }
}

然后在上面的循環(huán)中，我們就可以調(diào)用這個方法來實現(xiàn)。

while (iterator.hasNext()){
    Map.Entry<String, String[]> next = iterator.next();
    String key = next.getKey();
    String value = next.getValue()[0];
    setProperty(userInfo,key,value);
}

為什么要說到內(nèi)省機制呢？因為Spring在處理這件事的時候，最終也是靠它處理的。

簡單來說，它是通過BeanWrapperImpl來處理的。關(guān)于BeanWrapperImpl有個很簡單的使用方法：

SysUser user = new SysUser();
BeanWrapper wrapper = new BeanWrapperImpl(user.getClass());

wrapper.setPropertyValue("id","20001");
wrapper.setPropertyValue("name","Jack");

Object instance = wrapper.getWrappedInstance();
System.out.println(instance);

wrapper.setPropertyValue最后就會調(diào)用到BeanWrapperImpl#BeanPropertyHandler.setValue()方法。

它的setValue方法和我們上面的setProperty方法大致相同。

private class BeanPropertyHandler extends PropertyHandler {
    //屬性描述符
    private final PropertyDescriptor pd;
    public void setValue(@Nullable Object value) throws Exception {
        //獲取set方法
        Method writeMethod = this.pd.getWriteMethod();
        ReflectionUtils.makeAccessible(writeMethod);
        //設(shè)置
        writeMethod.invoke(BeanWrapperImpl.this.getWrappedInstance(), value);
    }
}

通過上面的方式，就完成了GET請求參數(shù)到Java Bean對象的自動轉(zhuǎn)換。

回過頭來，我們再看Spring。雖然我們上面寫的很簡單，但真正用起來還需要考慮的很多很多。Spring中處理這種參數(shù)的解析器是ServletModelAttributeMethodProcessor。

它的解析過程在其父類ModelAttributeMethodProcessor.resolveArgument()方法。整個過程，我們也可以分為三個步驟來看。

1、獲取目標(biāo)類的構(gòu)造函數(shù)

根據(jù)參數(shù)類型，先生成一個目標(biāo)類的構(gòu)造函數(shù)，以供后面綁定數(shù)據(jù)的時候使用。

2、創(chuàng)建數(shù)據(jù)綁定器WebDataBinder

WebDataBinder繼承自DataBinder。而DataBinder主要的作用，簡言之就是利用BeanWrapper給對象的屬性設(shè)值。

3、綁定數(shù)據(jù)到目標(biāo)類，并返回

在這里，又把WebDataBinder轉(zhuǎn)換成ServletRequestDataBinder對象，然后調(diào)用它的bind方法。

接下來有個很重要的步驟是，將request中的參數(shù)轉(zhuǎn)換為MutablePropertyValues pvs對象。

然后接下來就是循環(huán)pvs，調(diào)用setPropertyValue設(shè)置屬性。當(dāng)然了，最后調(diào)用的其實就是BeanWrapperImpl#BeanPropertyHandler.setValue()。

下面有段代碼可以更好的理解這一過程，效果是一樣的：

//模擬Request參數(shù)
Map<String,Object> map = new HashMap();
map.put("id","1001");
map.put("name","Jack");
map.put("password","123456");
map.put("address","北京市海淀區(qū)");

//將request對象轉(zhuǎn)換為MutablePropertyValues對象
MutablePropertyValues propertyValues = new MutablePropertyValues(map);
SysUser sysUser = new SysUser();
//創(chuàng)建數(shù)據(jù)綁定器
ServletRequestDataBinder binder = new ServletRequestDataBinder(sysUser);
//bind數(shù)據(jù)
binder.bind(propertyValues);
System.out.println(JSONObject.toJSONString(sysUser));

五、自定義參數(shù)解析器

我們說所有的消息解析器都實現(xiàn)了HandlerMethodArgumentResolver接口。我們也可以定義一個參數(shù)解析器，讓它實現(xiàn)這個接口就好了。

首先，我們可以定義一個RequestXuner注解。

@Target({ElementType.PARAMETER})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface RequestXuner {
    String name() default "";
    boolean required() default false;
    String defaultValue() default "default";
}

然后是實現(xiàn)了HandlerMethodArgumentResolver接口的解析器類。

public class XunerArgumentResolver implements HandlerMethodArgumentResolver {
    @Override
    public boolean supportsParameter(MethodParameter parameter) {
        return parameter.hasParameterAnnotation(RequestXuner.class);
    }

    @Override
    public Object resolveArgument(MethodParameter methodParameter,
                                  ModelAndViewContainer modelAndViewContainer,
                                  NativeWebRequest nativeWebRequest,
                                  WebDataBinderFactory webDataBinderFactory){
    
        //獲取參數(shù)上的注解
        RequestXuner annotation = methodParameter.getParameterAnnotation(RequestXuner.class);
        String name = annotation.name();
        //從Request中獲取參數(shù)值
        String parameter = nativeWebRequest.getParameter(name);
        return "HaHa，"+parameter;
    }
}

不要忘記需要配置一下。

@Configuration
public class WebMvcConfiguration extends WebMvcConfigurationSupport {
    @Override
    protected void addArgumentResolvers(List<HandlerMethodArgumentResolver> resolvers) {
        resolvers.add(new XunerArgumentResolver());
    }
}

一頓操作后，在Controller中我們可以這樣使用它：

@ResponseBody
@RequestMapping("/test4")
public String test4(@RequestXuner(name="xuner") String xuner){
    logger.info("參數(shù)：{}",xuner);
    return "Test4";
}

六、總結(jié)

本文內(nèi)容通過相關(guān)示例代碼展示了Spring中部分解析器解析參數(shù)的過程。說到底，無論參數(shù)如何變化，參數(shù)類型再怎么復(fù)雜。

它們都是通過HTTP請求發(fā)送過來的，那么就可以通過HttpServletRequest來獲取到一切。Spring做的就是通過注解，盡量適配大部分應(yīng)用場景。