Spring中使用LocalDateTime、LocalDate等引數作為入參
0x0 背景
專案中使用LocalDateTime系列作為dto中時間的型別,但是spring收到引數後總報錯,為了全域性配置時間型別轉換,嘗試瞭如下3中方法。
注:本文基於Springboot2.0測試,如果無法生效可能是spring版本較低導致的。PS:如果你的Controller中的LocalDate型別的引數啥註解(RequestParam、PathVariable等)都沒加,也是會出錯的,因為預設情況下,解析這種引數使用ModelAttributeMethodProcessor進行處理,而這個處理器要通過反射例項化一個物件出來,然後再對物件中的各個引數進行convert,但是LocalDate類沒有建構函式,無法反射例項化因此會報錯!!!
0x1 當LocalDateTime作為RequestParam或者PathVariable時
這種情況要和時間作為Json字串時區別對待,因為前端json轉後端pojo底層使用的是Json序列化Jackson工具(HttpMessgeConverter);而時間字串作為普通請求引數傳入時,轉換用的是Converter,兩者有區別哦。
在這種情況下,有如下幾種方案:
1. 使用Converter
import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; import org.springframework.core.convert.converter.Converter; import org.springframework.http.converter.HttpMessageConverter; import java.time.LocalDate; import java.time.LocalDateTime; import java.time.format.DateTimeFormatter; @Configuration public class DateConfig { @Bean public Converter<String,LocalDate> localDateConverter() { return new Converter<>() { @Override public LocalDate convert(String source) { return LocalDate.parse(source,DateTimeFormatter.ofPattern("yyyy-MM-dd")); } }; } @Bean public Converter<String,LocalDateTime> localDateTimeConverter() { return new Converter<>() { @Override public LocalDateTime convert(String source) { return LocalDateTime.parse(source,DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss")); } }; } }
以上兩個bean會注入到spring mvc的引數解析器(好像叫做ParameterConversionService),當傳入的字串要轉為LocalDateTime類時,spring會呼叫該Converter對這個入參進行轉換。
2. 使用ControllerAdvice配合initBinder
@ControllerAdvice public class GlobalExceptionHandler { @InitBinder protected void initBinder(WebDataBinder binder) { binder.registerCustomEditor(LocalDate.class,new PropertyEditorSupport() { @Override public void setAsText(String text) throws IllegalArgumentException { setValue(LocalDate.parse(text,DateTimeFormatter.ofPattern("yyyy-MM-dd"))); } }); binder.registerCustomEditor(LocalDateTime.class,new PropertyEditorSupport() { @Override public void setAsText(String text) throws IllegalArgumentException { setValue(LocalDateTime.parse(text,DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss"))); } }); binder.registerCustomEditor(LocalTime.class,new PropertyEditorSupport() { @Override public void setAsText(String text) throws IllegalArgumentException { setValue(LocalTime.parse(text,DateTimeFormatter.ofPattern("HH:mm:ss"))); } }); } }
從名字就可以看出來,這是在controller做環切(這裡面還可以全域性異常捕獲),在引數進入handler之前進行轉換;轉換為我們相應的物件。
0x2 當LocalDateTime作為Json形式傳入
這種情況下,如同上文描述,要利用Jackson的json序列化和反序列化來做:
@Configuration public class JacksonConfig { /** 預設日期時間格式 */ public static final String DEFAULT_DATE_TIME_FORMAT = "yyyy-MM-dd HH:mm:ss"; /** 預設日期格式 */ public static final String DEFAULT_DATE_FORMAT = "yyyy-MM-dd"; /** 預設時間格式 */ public static final String DEFAULT_TIME_FORMAT = "HH:mm:ss"; @Bean public ObjectMapper objectMapper(){ ObjectMapper objectMapper = new ObjectMapper(); // objectMapper.disable(SerializationFeature.WRITE_DATES_AS_TIMESTAMPS); // objectMapper.disable(DeserializationFeature.ADJUST_DATES_TO_CONTEXT_TIME_ZONE); JavaTimeModule javaTimeModule = new JavaTimeModule(); javaTimeModule.addSerializer(LocalDateTime.class,new LocalDateTimeSerializer(DateTimeFormatter.ofPattern(DEFAULT_DATE_TIME_FORMAT))); javaTimeModule.addSerializer(LocalDate.class,new LocalDateSerializer(DateTimeFormatter.ofPattern(DEFAULT_DATE_FORMAT))); javaTimeModule.addSerializer(LocalTime.class,new LocalTimeSerializer(DateTimeFormatter.ofPattern(DEFAULT_TIME_FORMAT))); javaTimeModule.addDeserializer(LocalDateTime.class,new LocalDateTimeDeserializer(DateTimeFormatter.ofPattern(DEFAULT_DATE_TIME_FORMAT))); javaTimeModule.addDeserializer(LocalDate.class,new LocalDateDeserializer(DateTimeFormatter.ofPattern(DEFAULT_DATE_FORMAT))); javaTimeModule.addDeserializer(LocalTime.class,new LocalTimeDeserializer(DateTimeFormatter.ofPattern(DEFAULT_TIME_FORMAT))); objectMapper.registerModule(javaTimeModule).registerModule(new ParameterNamesModule()); return objectMapper; } }
0x3 來個完整的配置吧
package com.fly.hi.common.config; import com.fasterxml.jackson.core.JsonGenerator; import com.fasterxml.jackson.core.JsonParser; import com.fasterxml.jackson.core.JsonProcessingException; import com.fasterxml.jackson.databind.*; import com.fasterxml.jackson.datatype.jsr310.JavaTimeModule; import com.fasterxml.jackson.datatype.jsr310.deser.LocalDateDeserializer; import com.fasterxml.jackson.datatype.jsr310.deser.LocalDateTimeDeserializer; import com.fasterxml.jackson.datatype.jsr310.deser.LocalTimeDeserializer; import com.fasterxml.jackson.datatype.jsr310.ser.LocalDateSerializer; import com.fasterxml.jackson.datatype.jsr310.ser.LocalDateTimeSerializer; import com.fasterxml.jackson.datatype.jsr310.ser.LocalTimeSerializer; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; import org.springframework.core.convert.converter.Converter; import java.io.IOException; import java.text.ParseException; import java.text.SimpleDateFormat; import java.time.LocalDate; import java.time.LocalDateTime; import java.time.LocalTime; import java.time.format.DateTimeFormatter; import java.util.Date; @Configuration public class DateConfig { /** 預設日期時間格式 */ public static final String DEFAULT_DATE_TIME_FORMAT = "yyyy-MM-dd HH:mm:ss"; /** 預設日期格式 */ public static final String DEFAULT_DATE_FORMAT = "yyyy-MM-dd"; /** 預設時間格式 */ public static final String DEFAULT_TIME_FORMAT = "HH:mm:ss"; /** * LocalDate轉換器,用於轉換RequestParam和PathVariable引數 */ @Bean public Converter<String,DateTimeFormatter.ofPattern(DEFAULT_DATE_FORMAT)); } }; } /** * LocalDateTime轉換器,用於轉換RequestParam和PathVariable引數 */ @Bean public Converter<String,DateTimeFormatter.ofPattern(DEFAULT_DATE_TIME_FORMAT)); } }; } /** * LocalTime轉換器,用於轉換RequestParam和PathVariable引數 */ @Bean public Converter<String,LocalTime> localTimeConverter() { return new Converter<>() { @Override public LocalTime convert(String source) { return LocalTime.parse(source,DateTimeFormatter.ofPattern(DEFAULT_TIME_FORMAT)); } }; } /** * Date轉換器,用於轉換RequestParam和PathVariable引數 */ @Bean public Converter<String,Date> dateConverter() { return new Converter<>() { @Override public Date convert(String source) { SimpleDateFormat format = new SimpleDateFormat(DEFAULT_DATE_TIME_FORMAT); try { return format.parse(source); } catch (ParseException e) { throw new RuntimeException(e); } } }; } /** * Json序列化和反序列化轉換器,用於轉換Post請求體中的json以及將我們的物件序列化為返回響應的json */ @Bean public ObjectMapper objectMapper(){ ObjectMapper objectMapper = new ObjectMapper(); objectMapper.disable(SerializationFeature.WRITE_DATES_AS_TIMESTAMPS); objectMapper.disable(DeserializationFeature.ADJUST_DATES_TO_CONTEXT_TIME_ZONE); //LocalDateTime系列序列化和反序列化模組,繼承自jsr310,我們在這裡修改了日期格式 JavaTimeModule javaTimeModule = new JavaTimeModule(); javaTimeModule.addSerializer(LocalDateTime.class,new LocalTimeDeserializer(DateTimeFormatter.ofPattern(DEFAULT_TIME_FORMAT))); //Date序列化和反序列化 javaTimeModule.addSerializer(Date.class,new JsonSerializer<>() { @Override public void serialize(Date date,JsonGenerator jsonGenerator,SerializerProvider serializerProvider) throws IOException { SimpleDateFormat formatter = new SimpleDateFormat(DEFAULT_DATE_TIME_FORMAT); String formattedDate = formatter.format(date); jsonGenerator.writeString(formattedDate); } }); javaTimeModule.addDeserializer(Date.class,new JsonDeserializer<>() { @Override public Date deserialize(JsonParser jsonParser,DeserializationContext deserializationContext) throws IOException,JsonProcessingException { SimpleDateFormat format = new SimpleDateFormat(DEFAULT_DATE_TIME_FORMAT); String date = jsonParser.getText(); try { return format.parse(date); } catch (ParseException e) { throw new RuntimeException(e); } } }); objectMapper.registerModule(javaTimeModule); return objectMapper; } }
0x4 深入研究SpringMVC資料繫結過程
接下來進入debug模式,看看mvc是如何將我們request中的引數繫結到我們controller層方法入參的:
寫一個簡單controller,下個斷點看看方法呼叫棧:
@GetMapping("/getDate") public LocalDateTime getDate(@RequestParam LocalDate date,@RequestParam LocalDateTime dateTime,@RequestParam Date originalDate) { System.out.println(date); System.out.println(dateTime); System.out.println(originalDate); return LocalDateTime.now(); }
斷住以後,我們看下方法呼叫棧中一些關鍵方法:
//進入DispatcherServlet doService:942,DispatcherServlet //處理請求 doDispatch:1038,DispatcherServlet //生成呼叫鏈(前處理、實際呼叫方法、後處理) handle:87,AbstractHandlerMethodAdapter //反射獲取到實際呼叫方法,準備開始呼叫 invokeHandlerMethod:895,RequestMappingHandlerAdapter invokeAndHandle:102,ServletInvocableHandlerMethod //這裡是關鍵,引數從這裡開始獲取到 invokeForRequest:142,InvocableHandlerMethod doInvoke:215,InvocableHandlerMethod //這個是Java reflect呼叫,因此一定是在這之前獲取到的引數 invoke:566,Method
根據上述分析,發現invokeForRequest:142,InvocableHandlerMethod這裡的程式碼是用來拿到實際引數的:
@Nullable public Object invokeForRequest(NativeWebRequest request,@Nullable ModelAndViewContainer mavContainer,Object... providedArgs) throws Exception { //這個方法是獲取引數的,在這裡下個斷 Object[] args = getMethodArgumentValues(request,mavContainer,providedArgs); if (logger.isTraceEnabled()) { logger.trace("Arguments: " + Arrays.toString(args)); } //這裡開始呼叫方法 return doInvoke(args); }
進入這個方法看看是什麼操作:
protected Object[] getMethodArgumentValues(NativeWebRequest request,Object... providedArgs) throws Exception { //獲取方法引數陣列,包含了入參資訊,比如型別、泛型等等 MethodParameter[] parameters = getMethodParameters(); //這個用來存放一會從request parameter轉換的引數 Object[] args = new Object[parameters.length]; for (int i = 0; i < parameters.length; i++) { MethodParameter parameter = parameters[i]; parameter.initParameterNameDiscovery(this.parameterNameDiscoverer); //這裡看起來沒啥卵用(providedArgs為空) args[i] = resolveProvidedArgument(parameter,providedArgs); //這裡開始獲取到方法實際呼叫的引數,步進 if (this.argumentResolvers.supportsParameter(parameter)) { //從名字就看出來:引數解析器解析引數 args[i] = this.argumentResolvers.resolveArgument(parameter,request,this.dataBinderFactory); continue; } } return args; }
進入resolveArgument看看:
public Object resolveArgument(MethodParameter parameter,NativeWebRequest webRequest,@Nullable WebDataBinderFactory binderFactory) throws Exception { //根據方法入參,獲取對應的解析器 HandlerMethodArgumentResolver resolver = getArgumentResolver(parameter); //開始解析引數(把請求中的parameter轉為方法的入參) return resolver.resolveArgument(parameter,webRequest,binderFactory); }
這裡根據引數獲取相應的引數解析器,看看內部如何獲取的:
//遍歷,呼叫supportParameter方法,跟進看看 for (HandlerMethodArgumentResolver methodArgumentResolver : this.argumentResolvers) { if (methodArgumentResolver.supportsParameter(parameter)) { result = methodArgumentResolver; this.argumentResolverCache.put(parameter,result); break; } }
這裡,遍歷引數解析器,查詢有沒有適合的解析器!那麼,有哪些引數解析器呢(我測試的時候有26個)???我列出幾個重要的看看,是不是很眼熟!!!
{RequestParamMethodArgumentResolver@7686} {PathVariableMethodArgumentResolver@8359} {RequestResponseBodyMethodProcessor@8366} {RequestPartMethodArgumentResolver@8367}
我們進入最常用的一個解析器看看他的supportsParameter方法,發現就是通過引數註解來獲取相應的解析器的。
public boolean supportsParameter(MethodParameter parameter) { //如果引數擁有註解@RequestParam,則走這個分支(知道為什麼上文要對RequestParam和Json兩種資料區別對待了把) if (parameter.hasParameterAnnotation(RequestParam.class)) { //這個似乎是對Optional型別的引數進行處理的 if (Map.class.isAssignableFrom(parameter.nestedIfOptional().getNestedParameterType())) { RequestParam requestParam = parameter.getParameterAnnotation(RequestParam.class); return (requestParam != null && StringUtils.hasText(requestParam.name())); } else { return true; } } //...... }
也就是說,對於@RequestParam和@RequestBody以及@PathVariable註解的引數,SpringMVC會使用不通的引數解析器進行資料繫結!
那麼,這三種解析器分別使用什麼Converter解析引數呢?我們分別進入三種解析器看一看:
首先看下RequestParamMethodArgumentResolver發現內部使用WebDataBinder進行資料繫結,底層使用的是ConversionService (也就是我們的Converter注入的地方)
WebDataBinder binder = binderFactory.createBinder(webRequest,null,namedValueInfo.name); //通過DataBinder進行資料繫結的 arg = binder.convertIfNecessary(arg,parameter.getParameterType(),parameter); //跟進convertIfNecessary() public <T> T convertIfNecessary(@Nullable Object value,@Nullable Class<T> requiredType,@Nullable MethodParameter methodParam) throws TypeMismatchException { return getTypeConverter().convertIfNecessary(value,requiredType,methodParam); } //繼續跟進,看到了把 ConversionService conversionService = this.propertyEditorRegistry.getConversionService(); if (editor == null && conversionService != null && newValue != null && typeDescriptor != null) { TypeDescriptor sourceTypeDesc = TypeDescriptor.forObject(newValue); if (conversionService.canConvert(sourceTypeDesc,typeDescriptor)) { try { return (T) conversionService.convert(newValue,sourceTypeDesc,typeDescriptor); } catch (ConversionFailedException ex) { // fallback to default conversion logic below conversionAttemptEx = ex; } } }
然後看下RequestResponseBodyMethodProcessor發現使用的轉換器是HttpMessageConverter型別的:
//resolveArgument方法內部呼叫下面進行引數解析 Object arg = readWithMessageConverters(webRequest,parameter,parameter.getNestedGenericParameterType()); //step into readWithMessageConverters(),我們看到這裡的Converter是HttpMessageConverter for (HttpMessageConverter<?> converter : this.messageConverters) { Class<HttpMessageConverter<?>> converterType = (Class<HttpMessageConverter<?>>) converter.getClass(); GenericHttpMessageConverter<?> genericConverter = (converter instanceof GenericHttpMessageConverter ? (GenericHttpMessageConverter<?>) converter : null); if (genericConverter != null ? genericConverter.canRead(targetType,contextClass,contentType) : (targetClass != null && converter.canRead(targetClass,contentType))) { if (message.hasBody()) { HttpInputMessage msgToUse = getAdvice().beforeBodyRead(message,targetType,converterType); body = (genericConverter != null ? genericConverter.read(targetType,msgToUse) : ((HttpMessageConverter<T>) converter).read(targetClass,msgToUse)); body = getAdvice().afterBodyRead(body,msgToUse,converterType); } else { body = getAdvice().handleEmptyBody(null,message,converterType); } break; } }
最後看下PathVariableMethodArgumentResolver發現 和RequestParam走的執行路徑一致(二者都是繼承自AbstractNamedValueMethodArgumentResolver解析器),因此程式碼就不貼了。
0xFF總結
如果要轉換request傳來的引數到我們指定的型別,根據入參註解要進行區分:
如果是RequestBody,那麼通過配置ObjectMapper(這個玩意兒會注入到Jackson的HttpMessagConverter裡面,即MappingJackson2HttpMessageConverter中)來實現Json格式資料的序列化和反序列化;
如果是RequestParam或者PathVariable型別的引數,通過配置Converter實現引數轉換(這些Converter會注入到ConversionService中)。
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。