写在前面 今天我们来聊一聊SpringBoot中的参数解析器,这在某些场景下非常有用。一般来说,在一个Web请求里面参数要么是放在请求地址,要么就是放在请求体里面,极个别的会放在请求头中。
如果请求参数放在请求地址中,那么通常会采用@RequestParam
/@PathVariable
或者如下方式来获取参数:
1 String username = request.getParameter("username");
如果请求参数放在请求体里面,那么通常会采用@RequestBody
或者如下方式来获取参数:
1 String username = request.getParameter("username");
如果请求参数放在请求头里面,那么通常会采用@RequestHeader
或者如下方式来获取参数:
1 String username = request.getHeader("username");
如果参数是JSON形式的,那么会从输入流中获取并解析成JSON字符串,再通过JSON工具转化为POJO对象:
1 2 3 4 BufferedReader reader = new BufferedReader(new InputStreamReader(request.getInputStream())); String json = reader.readLine(); reader.close(); User user = new ObjectMapper().readValue(json, User.class);
无论参数是key-value键值对,还是JSON形式数据,以上几种方式基本上涵盖了日常开发的所有需求。
现在我们以下面的接口为例,来深度分析SpringMVC如何从请求中获取参数:
1 2 3 4 5 6 7 @RestController public class UserController { @GetMapping("/user") public String user(String username){ return "I am" + username; } }
毫无疑问这个username参数肯定是从HttpServletRequest中获取的,那么它是如何获取的呢?
方法参数解析器 HandlerMethodArgumentResolver接口 我们知道在SpringBoot中与Web相关的配置信息都在WebMvcConfigurer
接口中,可以看到该接口中有一个名为addArgumentResolvers
的默认方法,用于添加参数解析器:
1 2 default void addArgumentResolvers(List<HandlerMethodArgumentResolver> resolvers) { }
所以可以看到HandlerMethodArgumentResolver
这个其实就是具体的一些参数解析器,实际上它是一个接口。我们来看HandlerMethodArgumentResolver
这一接口,该接口用于解析方法中的参数:
1 2 3 4 5 6 public interface HandlerMethodArgumentResolver { boolean supportsParameter(MethodParameter parameter); @Nullable Object resolveArgument(MethodParameter parameter, @Nullable ModelAndViewContainer mavContainer, NativeWebRequest webRequest, @Nullable WebDataBinderFactory binderFactory) throws Exception; }
可以看到这个HandlerMethodArgumentResolver
接口中就两个方法,其中supportsParameter()
方法表示是否启用该参数解析器,true表示启用,false表示不启用;resolveArgument()
方法是具体的解析过程,即从HttpServletRequest中取出参数的过程,该方法的返回值就是接口中参数的值。所以如果开发者想自定义参数解析器,只需实现该接口并重写其中的两个方法。
由于SpringBoot采用约定大于配置这一规则,因此建议HandlerMethodArgumentResolver
接口的实现类命名规则为“解析器对应的注解名称”+“MethodArgumentResolver
”。
将下来我们分析几个常用的HandlerMethodArgumentResolver
接口实现类,了解它们对于提升技能有非常大的帮助。
RequestParamMethodArgumentResolver类 通过前面的分析,可以得到这个解析器所对应的注解名称为RequestParam,因此首先我们看一下这个注解的源码:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 @Target({ElementType.PARAMETER}) @Retention(RetentionPolicy.RUNTIME) @Documented public @interface RequestParam { @AliasFor("name") String value() default ""; @AliasFor("value") String name() default ""; boolean required() default true; String defaultValue() default "\n\t\t\n\t\t\n\ue000\ue001\ue002\n\t\t\t\t\n"; }
可以看到这个@RequestParam
注解有四个属性,其中name和value互为各自的别名,required表示参数是否必须传,默认是必须;defaultValue则是参数的默认值。
接着查看一下这个RequestParamMethodArgumentResolver类中的supportsParameter
和resolveName
方法,注意它的resolveArgument
方法名称发生了变化:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 public boolean supportsParameter(MethodParameter parameter) { if (parameter.hasParameterAnnotation(RequestParam.class)) { if (!Map.class.isAssignableFrom(parameter.nestedIfOptional().getNestedParameterType())) { return true; } else { RequestParam requestParam = (RequestParam)parameter.getParameterAnnotation(RequestParam.class); return requestParam != null && StringUtils.hasText(requestParam.name()); } } else if (parameter.hasParameterAnnotation(RequestPart.class)) { return false; } else { parameter = parameter.nestedIfOptional(); if (MultipartResolutionDelegate.isMultipartArgument(parameter)) { return true; } else { return this.useDefaultResolution ? BeanUtils.isSimpleProperty(parameter.getNestedParameterType()) : false; } } } @Nullable protected Object resolveName(String name, MethodParameter parameter, NativeWebRequest request) throws Exception { HttpServletRequest servletRequest = (HttpServletRequest)request.getNativeRequest(HttpServletRequest.class); Object arg; if (servletRequest != null) { arg = MultipartResolutionDelegate.resolveMultipartArgument(name, parameter, servletRequest); if (arg != MultipartResolutionDelegate.UNRESOLVABLE) { return arg; } } arg = null; MultipartRequest multipartRequest = (MultipartRequest)request.getNativeRequest(MultipartRequest.class); if (multipartRequest != null) { List<MultipartFile> files = multipartRequest.getFiles(name); if (!files.isEmpty()) { arg = files.size() == 1 ? files.get(0) : files; } } if (arg == null) { String[] paramValues = request.getParameterValues(name); if (paramValues != null) { arg = paramValues.length == 1 ? paramValues[0] : paramValues; } } return arg; }
简单解释一下上述方法的逻辑: (1)supportsParameter()
方法,如果使用了@RequestParam
注解且不是Map类型,或者是Map类型同时传入了name属性,又或者是没有使用@RequestParam
和@RequestPart
注解且这个参数有多个组成,或者使用默认的解析器且参数的嵌套是简单类型,则使用该参数解析器; (2)resolveName()
方法,首先获取HttpServletRequest
对象,如果该对象存在,则解析请求中的多个参数并返回这些参数的值;如果该对象不存在,但是MultipartRequest
存在,那么从这个MultipartRequest
中通过参数的名称来得到这些参数值,如果参数值存在,那么返回参数值的信息。如果参数值不存在,那么从请求中根据参数名称来得到参数值,如果参数值存在,那么返回参数值的信息。
类似于这种的情况,我们会使用诸如下面的接口:
1 2 3 4 5 6 7 @RestController public class UserController { @GetMapping("/user") public String user(@RequestParam("username") String username){ return "I am " + username; } }
RequestParamMapMethodArgumentResolver类 这个方法参数解析器所对应的注解名称同样也是RequestParam,接着查看一下这个RequestParamMapMethodArgumentResolver类中的supportsParameter
和resolveArgument
方法:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 public boolean supportsParameter(MethodParameter parameter) { RequestParam requestParam = (RequestParam)parameter.getParameterAnnotation(RequestParam.class); return requestParam != null && Map.class.isAssignableFrom(parameter.getParameterType()) && !StringUtils.hasText(requestParam.name()); } public Object resolveArgument(MethodParameter parameter, @Nullable ModelAndViewContainer mavContainer, NativeWebRequest webRequest, @Nullable WebDataBinderFactory binderFactory) throws Exception { ResolvableType resolvableType = ResolvableType.forMethodParameter(parameter); Class valueType; HttpServletRequest servletRequest; Collection parts; Iterator var10; Part part; Map parameterMap; MultipartRequest multipartRequest; if (!MultiValueMap.class.isAssignableFrom(parameter.getParameterType())) { valueType = resolvableType.asMap().getGeneric(new int[]{1}).resolve(); if (valueType == MultipartFile.class) { multipartRequest = MultipartResolutionDelegate.resolveMultipartRequest(webRequest); return multipartRequest != null ? multipartRequest.getFileMap() : new LinkedHashMap(0); } else if (valueType == Part.class) { servletRequest = (HttpServletRequest)webRequest.getNativeRequest(HttpServletRequest.class); if (servletRequest != null && MultipartResolutionDelegate.isMultipartRequest(servletRequest)) { parts = servletRequest.getParts(); LinkedHashMap<String, Part> result = CollectionUtils.newLinkedHashMap(parts.size()); var10 = parts.iterator(); while(var10.hasNext()) { part = (Part)var10.next(); if (!result.containsKey(part.getName())) { result.put(part.getName(), part); } } return result; } else { return new LinkedHashMap(0); } } else { parameterMap = webRequest.getParameterMap(); Map<String, String> result = CollectionUtils.newLinkedHashMap(parameterMap.size()); parameterMap.forEach((key, values) -> { if (values.length > 0) { result.put(key, values[0]); } }); return result; } } else { valueType = resolvableType.as(MultiValueMap.class).getGeneric(new int[]{1}).resolve(); if (valueType == MultipartFile.class) { multipartRequest = MultipartResolutionDelegate.resolveMultipartRequest(webRequest); return multipartRequest != null ? multipartRequest.getMultiFileMap() : new LinkedMultiValueMap(0); } else if (valueType != Part.class) { parameterMap = webRequest.getParameterMap(); MultiValueMap<String, String> result = new LinkedMultiValueMap(parameterMap.size()); parameterMap.forEach((key, values) -> { String[] var3 = values; int var4 = values.length; for(int var5 = 0; var5 < var4; ++var5) { String value = var3[var5]; result.add(key, value); } }); return result; } else { servletRequest = (HttpServletRequest)webRequest.getNativeRequest(HttpServletRequest.class); if (servletRequest != null && MultipartResolutionDelegate.isMultipartRequest(servletRequest)) { parts = servletRequest.getParts(); LinkedMultiValueMap<String, Part> result = new LinkedMultiValueMap(parts.size()); var10 = parts.iterator(); while(var10.hasNext()) { part = (Part)var10.next(); result.add(part.getName(), part); } return result; } else { return new LinkedMultiValueMap(0); } } } }
简单解释一下上述方法的逻辑: (1)supportsParameter()
方法,如果使用了@RequestParam
注解且参数是Map类型,同时@RequestParam
注解中没有设置name属性,那么就可以使用该参数解析器; (2)resolveArgument()
方法,总的来说分为两种情况,一种是MultiValueMap,另一种则是其他的Map。对于MultiValueMap来说,如果它是MultipartFile或者Part类型,那么就可以处理文件上传;如果是其他的则是普通的请求参数。如果是普通的Map,那么就直接从原始请求中获取请求参数,并将这些参数放到一个LinkedMultiValueMap中并返回。
类似于这种的情况,我们会使用诸如下面的接口:
1 2 3 4 5 6 7 @RestController public class UserController { @PostMapping("/user") public void user(@RequestParam MultiValueMap map){ System.out.println(map.get("username")); } }
PrincipalMethodArgumentResolver类 查看一下PrincipalMethodArgumentResolver
类的源码,如下所示:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 public boolean supportsParameter(MethodParameter parameter) { return Principal.class.isAssignableFrom(parameter.getParameterType()); } public Object resolveArgument(MethodParameter parameter, @Nullable ModelAndViewContainer mavContainer, NativeWebRequest webRequest, @Nullable WebDataBinderFactory binderFactory) throws Exception { HttpServletRequest request = (HttpServletRequest)webRequest.getNativeRequest(HttpServletRequest.class); if (request == null) { throw new IllegalStateException("Current request is not of type HttpServletRequest: " + webRequest); } else { Principal principal = request.getUserPrincipal(); if (principal != null && !parameter.getParameterType().isInstance(principal)) { throw new IllegalStateException("Current user principal is not of type [" + parameter.getParameterType().getName() + "]: " + principal); } else { return principal; } } }
简单解释一下上述方法的逻辑: (1)supportsParameter()
方法,用于判断参数类型是否为Principal这一类型,如果是则使用该参数解析器; (2)resolveArgument()
方法,首先从原始请求中获取HttpServletRequest
对象,如果该对象不存在则抛出异常;如果存在则从请求中获取Principal对象并返回。这个Principal对象里面包含登录的用户名称。
类似于这种的情况,我们会使用诸如下面的接口:
1 2 3 4 5 6 7 @RestController public class UserController { @GetMapping("/user") public String user(Principal principal){ return "My name is " + principal.getName(); } }
PathVariableMethodArgumentResolver类 通过前面的分析,可以得到这个解析器所对应的注解名称为PathVariable,因此首先我们看一下这个注解的源码:
1 2 3 4 5 6 7 8 9 10 11 12 @Target({ElementType.PARAMETER}) @Retention(RetentionPolicy.RUNTIME) @Documented public @interface PathVariable { @AliasFor("name") String value() default ""; @AliasFor("value") String name() default ""; boolean required() default true; }
可以看到这个@PathVariable
注解有三个属性,其中name和value互为各自的别名,required表示参数是否必须传,默认是必须。
接着查看一下这个PathVariableMethodArgumentResolver类中的supportsParameter
和resolveName
方法,注意它的resolveArgument方法名称发生了变化:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 public boolean supportsParameter(MethodParameter parameter) { if (!parameter.hasParameterAnnotation(PathVariable.class)) { return false; } else if (!Map.class.isAssignableFrom(parameter.nestedIfOptional().getNestedParameterType())) { return true; } else { PathVariable pathVariable = (PathVariable)parameter.getParameterAnnotation(PathVariable.class); return pathVariable != null && StringUtils.hasText(pathVariable.value()); } } @Nullable protected Object resolveName(String name, MethodParameter parameter, NativeWebRequest request) throws Exception { Map<String, String> uriTemplateVars = (Map)request.getAttribute(HandlerMapping.URI_TEMPLATE_VARIABLES_ATTRIBUTE, 0); return uriTemplateVars != null ? uriTemplateVars.get(name) : null; }
简单解释一下上述方法的逻辑: (1)supportsParameter()
方法,用于判断如果参数上使用了@PathVariable
注解,并且参数的类型不是Map及其子类,则使用该参数解析器。或者是Map类型,则当@PathVariable
注解中的value属性有值时,才使用该参数解析器; (2)resolveName()
方法,用于从请求中获取uriTemplateVars,如果uriTemplateVars不为空,则从uriTemplateVars根据名称来获取值并返回。
类似于这种的情况,我们会使用诸如下面的接口:
1 2 3 4 5 6 7 @RestController public class UserController { @GetMapping("/user/{username}") public String user(@PathVariable("username") String username){ return "My name is " + username; } }
实战 假设如下接口中,我们需要获取用户传入的用户名,此时就可以使用自定义参数解析器这一方式:
1 2 3 4 5 6 7 @RestController public class UserController { @GetMapping("/user") public String user(@CurrentUserName String username){ return "I am" + username; } }
即在方法中通过使用@CurrentUserName
注解从HttpServletRequest中获取当前传入的用户名。要实现这个功能,步骤如下所示:
第一步,创建一个名为method-resolve
的SpringBoot项目,然后添加spring web
和spring test
依赖:
1 2 3 4 5 6 7 8 9 10 11 <dependencies> <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-web</artifactId> </dependency> <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-test</artifactId> <scope>test</scope> </dependency> </dependencies>
第二步,新建一个名为CurrentUserName
的注解:
1 2 3 4 5 @Retention(RetentionPolicy.RUNTIME) @Target({ElementType.PARAMETER}) @Documented public @interface CurrentUserName { }
第三步,新建一个名为CurrentUserNameHandlerMethodArgumentResolver
的类,注意它需要实现HandlerMethodArgumentResolver
接口并重写其中的上述两个方法:
1 2 3 4 5 6 7 8 9 10 11 12 public class CurrentUserNameHandlerMethodArgumentResolver implements HandlerMethodArgumentResolver { @Override public boolean supportsParameter(MethodParameter parameter) { return parameter.getParameterType().isAssignableFrom(String.class) && parameter.hasParameterAnnotation(CurrentUserName.class); } @Override public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer, NativeWebRequest webRequest, WebDataBinderFactory binderFactory) throws Exception { HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest(); return request.getParameter(parameter.getParameterName()); } }
简单解释一下上述方法的逻辑: (1)supportsParameter()
方法,判断当前参数类型是否为String且在参数上使用了@CurrentUserName
注解,这有这样才使用该参数解析器; (2)resolveArgument()
方法,用于返回接口中参数的值,这里直接调用request.getParameter()
方法传入参数名称进而得到参数的值。
第四步,注册自定义参数解析器。定义一个名为WebConfig
的类,注意这个类需要实现WebMvcConfigurer
接口,并重写其中的addArgumentResolvers
默认方法:
1 2 3 4 5 6 7 @Configuration public class WebConfig implements WebMvcConfigurer { @Override public void addArgumentResolvers(List<HandlerMethodArgumentResolver> resolvers) { resolvers.add(new CurrentUserNameHandlerMethodArgumentResolver()); } }
第五步,新建接口类。定义一个名为UserController
的类,里面的代码如下所示:
1 2 3 4 5 6 7 @RestController public class UserController { @GetMapping("/user") public String user(@CurrentUserName String username){ return "I am " + username; } }
第六步,启动项目,访问http://localhost:8080/user?username=melody
链接,可以看到页面显示如下信息:
小结 本文介绍了如何在SpringBoot中通过自定义类实现HandlerMethodArgumentResolver
接口,并重写其中的supportsParameter()
和resolveArgument()
方法来实现自定义参数解析器,同时也剖析了一些常用的参数注解以及背后的原理,最后通过实战学习了如何通过自定义一个注解来实现参数解析器。