實現自定義的參數解析器——HandlerMethodArgumentResolver

1.爲何須要本身實現參數解析器

咱們都知道在有註解的接口方法中加上@RequestBody等註解，springMVC會自動的將消息體等地方的裏面參數解析映射到請求的方法參數中。

若是咱們想要的信息不徹底是來自消息體等地方，好比說一部分是消息體，一部分是消息頭，甚至一部分從配置中獲取。這個時候咱們又但願在方法入參進來就將這些信息組裝好。或者說是須要從消息頭裏面去進行token解析認證的時候。

2. HandlerMethodArgumentResolver接口

實現HandlerMethodArgumentResolver來擴展本身的參數解析器。該接口有兩個方法supportsParameter和resolveArgument咱們須要重寫這兩個方法。

 

 

 

supportsParameter是條件，只有這個方法返回true的時候。纔會在調用resolveArgument方法進行參數解析。

Spring源碼中對這兩個方法的調用：

 

 

 

3.實現本身的參數解析器：

 

package com.nijunyang.springboot.argsresolver;

import org.springframework.core.MethodParameter;
import org.springframework.lang.Nullable;
import org.springframework.web.bind.support.WebDataBinderFactory;
import org.springframework.web.context.request.NativeWebRequest;
import org.springframework.web.method.support.HandlerMethodArgumentResolver;
import org.springframework.web.method.support.ModelAndViewContainer;

import javax.servlet.http.HttpServletRequest;

/**
 * @author: create by nijunyang
 * @date:2019/11/30
 */
public class DefArgumentResolver implements HandlerMethodArgumentResolver {
    @Override
    public boolean supportsParameter(MethodParameter methodParameter) {
        //方法參數是User 則使用此解析器
        return User.class.isAssignableFrom(methodParameter.getParameterType());
    }

    @Nullable
    @Override
    public Object resolveArgument(
            MethodParameter methodParameter,
            @Nullable ModelAndViewContainer modelAndViewContainer,
            NativeWebRequest nativeWebRequest,
            @Nullable WebDataBinderFactory webDataBinderFactory) throws Exception {
        HttpServletRequest servletRequest = nativeWebRequest.getNativeRequest(HttpServletRequest.class);
        String info = (String) nativeWebRequest.getAttribute("params", NativeWebRequest.SCOPE_REQUEST);
        //獲取消息頭認證信息，沒有後續操做了，根據業務實際來解析校驗該token
        String token = servletRequest.getHeader("Authorization");
        //本身塞數據進去，也能夠從配置文件獲取數據
        return new User("zhangsan",18);


    }
}

 

4.將本身的解析器加入到容器的解析器集合中

package com.nijunyang.springboot.argsresolver;

import org.springframework.stereotype.Component;
import org.springframework.web.method.support.HandlerMethodArgumentResolver;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;

import java.util.List;

/**
 * @author: create by nijunyang
 * @date:2019/11/30
 */
@Component
//必定要將本身的擴展類加到容器中
public class DefWebMvcConfigurer implements WebMvcConfigurer {

    public void addArgumentResolvers(List<HandlerMethodArgumentResolver> resolvers) {
        resolvers.add(new DefArgumentResolver());
    }
}

固然也能夠用註解的形式注入到DefWebMvcConfigurer 中在添加到集合中，同時還能夠在DefArgumentResolver 中獲取配置的信息。

5.Test

package com.nijunyang.springboot.argsresolver;

import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

/**
 * @author: create by nijunyang
 * @date:2019/11/30
 */
@RestController
public class Controller {

    @GetMapping("/test")
    public String test(User user){
        return user.toString();
    }
}

6.效果：

 

直接能夠獲取到咱們自定返回的對象。