1骡和、解決的問題
加了@ResponseBody注解的方法,請求的方式是post的json格式相寇,但如果我們也要通過post的application/x-www-form-urlencoded格式訪問此接口慰于。在不改變此接口的情況下。通過修改參數(shù)解析器HandlerMethodArgumentResovler來兼容兩種請求方法唤衫。
2婆赠、思路
根據(jù)不同的content-type使用不同參數(shù)解析處理器。
Content-Type為application/x-www-form-urlencoded佳励,使用ServletModelAttributeMethodProcessor解析器
Content-Type非application/x-www-form-urlencoded休里,使用RequestResponseBodyMethodProcessor解析器
3、代碼實現(xiàn)
定義自己的參數(shù)解析器赃承,繼承HandlerMethodArgumentResolver
/**
* 自定義參數(shù)解析器
* 功能:實現(xiàn)不同類型參數(shù)的解析
* 使用@RequestBody注解妙黍,Content-Type為application/x-www-form-urlencoded,使用ServletModelAttributeMethodProcessor解析器
* 使用@RequestBody注解瞧剖,Content-Type非application/x-www-form-urlencoded拭嫁,使用RequestResponseBodyMethodProcessor解析器
*
* @author lilei
* @since 2020/05/09
*/
@Log4j
public class CustomMethodArgumentResolver implements HandlerMethodArgumentResolver {
private RequestResponseBodyMethodProcessor requestResponseBodyMethodProcessor;
private ServletModelAttributeMethodProcessor servletModelAttributeMethodProcessor;
private final String APPLICATION_X_WWW_FORM_URLENCODED = "application/x-www-form-urlencoded";
private final String MULTIPART_FORM_DATA = "multipart/form-data";
public CustomMethodArgumentResolver(
RequestResponseBodyMethodProcessor requestResponseBodyMethodProcessor,
ServletModelAttributeMethodProcessor servletModelAttributeMethodProcessor) {
this.requestResponseBodyMethodProcessor = requestResponseBodyMethodProcessor;
this.servletModelAttributeMethodProcessor = servletModelAttributeMethodProcessor;
}
/**
* 參數(shù)解析器適用參數(shù)條件
*
* @param methodParameter
* @return 使用@RequestBody注解 返回true,未使用返回false
* @tips 可以使用自定義注解
*/
@Override
public boolean supportsParameter(MethodParameter methodParameter) {
return methodParameter.hasParameterAnnotation(RequestBody.class);
}
/**
* 參數(shù)解析
* @param methodParameter
* @param modelAndViewContainer
* @param nativeWebRequest
* @param webDataBinderFactory
* @return
* @throws Exception
*/
@Override
public Object resolveArgument(MethodParameter methodParameter, ModelAndViewContainer modelAndViewContainer, NativeWebRequest nativeWebRequest, WebDataBinderFactory webDataBinderFactory) throws Exception {
HttpServletRequest request = nativeWebRequest.getNativeRequest(HttpServletRequest.class);
if (request == null) {
throw new CustomrException(JsonResultCode.CODE_REQUEST_NULL.getName());
}
String contentType = request.getContentType();
//更加請求contentType不同使用不同解析器
log.info("contentType:"+contentType);
if (contentType.contains(APPLICATION_X_WWW_FORM_URLENCODED)||contentType.contains(MULTIPART_FORM_DATA)) {
log.info("Processor:"+servletModelAttributeMethodProcessor);
return servletModelAttributeMethodProcessor.resolveArgument(methodParameter, modelAndViewContainer, nativeWebRequest, webDataBinderFactory);
}
log.info("Processor:"+requestResponseBodyMethodProcessor);
return requestResponseBodyMethodProcessor.resolveArgument(methodParameter, modelAndViewContainer, nativeWebRequest, webDataBinderFactory);
}
}
注冊參數(shù)解析器
/**
* 配置注冊參數(shù)解析器
*/
@Configuration
@Log4j
public class ResolverConfig {
private final RequestMappingHandlerAdapter requestMappingHandlerAdapter;
public ResolverConfig(RequestMappingHandlerAdapter requestMappingHandlerAdapter) {
this.requestMappingHandlerAdapter = requestMappingHandlerAdapter;
}
@PostConstruct
private void addArgumentResolvers() {
log.info("start add argument resolvers...");
// 獲取到的是不可變的集合
List<HandlerMethodArgumentResolver> argumentResolvers = requestMappingHandlerAdapter.getArgumentResolvers();
CustomMethodArgumentResolver customMethodArgumentResolver = getCustomMethodArgumentResolver(argumentResolvers);
//新建參數(shù)解析器列表抓于,然后將新列表添加到requestMappingHandlerAdapter中
List<HandlerMethodArgumentResolver> newArgumentResolverList = new ArrayList<>(argumentResolvers.size() + 1);
// 將自定義方法參數(shù)解析器放置在第一個做粤,并保留原來的解析器
newArgumentResolverList.add(customMethodArgumentResolver );
newArgumentResolverList.addAll(argumentResolvers);
requestMappingHandlerAdapter.setArgumentResolvers(newArgumentResolverList);
log.info("add argument resolvers end!!!");
}
/**
* 獲取自定義方法參數(shù)解析器
* @param argumentResolversList
* @return
*/
private CustomMethodArgumentResolver getCustomMethodArgumentResolver(List<HandlerMethodArgumentResolver> argumentResolversList) {
// 解析Content-Type為application/json的默認解析器
RequestResponseBodyMethodProcessor requestResponseBodyMethodProcessor = null;
// 解析Content-Type為application/x-www-form-urlencoded的默認解析器
ServletModelAttributeMethodProcessor servletModelAttributeMethodProcessor = null;
if (argumentResolversList == null) {
throw new CustomResolverException(JsonResultCode.CODE_RESOLVER_NULL.getName(),JsonResultCode.CODE_RESOLVER_NULL.getCode());
}
for (HandlerMethodArgumentResolver argumentResolver : argumentResolversList) {
if (requestResponseBodyMethodProcessor != null && servletModelAttributeMethodProcessor != null) {
break;
}
if (argumentResolver instanceof RequestResponseBodyMethodProcessor) {
requestResponseBodyMethodProcessor = (RequestResponseBodyMethodProcessor) argumentResolver;
continue;
}
if (argumentResolver instanceof ServletModelAttributeMethodProcessor) {
servletModelAttributeMethodProcessor = (ServletModelAttributeMethodProcessor) argumentResolver;
}
}
if (requestResponseBodyMethodProcessor == null || servletModelAttributeMethodProcessor == null) {
throw new CustomResolverException(JsonResultCode.CODE_RESOLVER_VALUE_NULL.getName(),JsonResultCode.CODE_RESOLVER_VALUE_NULL.getCode());
}
return new CustomMethodArgumentResolver(requestResponseBodyMethodProcessor, servletModelAttributeMethodProcessor);
}
}