基于RestTemplate扩展,支持Get请求传递javaBean参数

784 阅读2分钟

一、背景说明

项目中使用了RestTemplate来对第三方接口进行请求。使用get方式请求查询接口时,需要将参数拼接到url中,不支持使用JavaBean传参,编写代码不友好。

二、开发常见写法

如果是是Get请求,通常需要这么写:

RestTemplate一般写法.jpeg

如果不使用占位符的写法,服务端将收不到参数。

不正确写法.jpeg

服务端接口定义如下,请求结果是参数解析失败,返回400错误码

服务端无法接收参数导致400.jpeg

三、扩展实践

这个时候就想,能不能直接传map,或者一个JavaBean来实现传参数呢? 于是查看源码,发现RestTemplate有扩展点可以进行扩展,下面就开始扩展之路了。 RestTemplate有一个接口,专门处理请求url的,

public interface UriTemplateHandler {

//这个方法就是我们传入的参数map

URI expand(String uriTemplate, Map<String, ?> uriVariables);

URI expand(String uriTemplate, Object... uriVariables);

}

于是自定义一个UriTemplateHandler,来对请求url进行处理。

代码如下:

public class GetUriTemplateHandler implements UriTemplateHandler {

private UriTemplateHandler uriTemplateHandler = new DefaultUriBuilderFactory();

@Override

public URI expand(String uriTemplate, Map<String, ?> uriVariables) {

//利用UriComponentsBuilder将请求参数追加到url,GetParamMultiValueMap是自定义的一个map

uriTemplate = UriComponentsBuilder

.fromHttpUrl(uriTemplate)

.queryParams(new GetParamMultiValueMap(uriVariables)).toUriString();

log.info("处理之后的url ==> {}", uriTemplate);

return uriTemplateHandler.expand(uriTemplate, uriVariables);

}

@Override

public URI expand(String uriTemplate, Object... uriVariables) {

return uriTemplateHandler.expand(uriTemplate, uriVariables);

}

}

GetParamMultiValueMap实现如下:

public class GetParamMultiValueMap<K,V> implements MultiValueMap<K, V> {

private final Map<K, List<V>> map;

public GetParamMultiValueMap(Map<K, V> paramMap) {

Assert.notNull(paramMap, "'map' must not be null");

//将请求参数转换成MultiValueMap

this.map = new HashMap<>();

paramMap.forEach((k, v) -> {

if (v instanceof List) {

map.put(k, (List<V>) v);

} else {

map.put(k, Lists.newArrayList(v));

}

});

}

}

GetUriTemplateHandler实现好之后,只需要在RestTemplate实例化之后将默认的UriTemplateHandler替换就可以使用自己实现的功能了

@Configuration

public class RestTemplateConfig {

@ConditionalOnClass(value = RestTemplate.class)

@Bean

public RestTemplate restTemplate() {

RestTemplate restTemplate = new RestTemplate();

//默认使用的jdk的http请求组件 替换成okhttp3

restTemplate.setRequestFactory(new OkHttp3ClientHttpRequestFactory());

//使用自定义的uri处理器,实现参数追加到url的功能

restTemplate.setUriTemplateHandler(new GetUriTemplateHandler());

return restTemplate;

}

}

下面再看测试结果:

验证处理参数.jpeg

收到了服务端的正确返回。

服务端正常返回.png

四、再次扩展

现在还是传递的map,能不能直接传一个JavaBean呢?

答案是可以的,通过对RestTemplate进行包装,

借助apache-commont包将JavaBean转换成map

将参数Bean转换成map.jpeg

注入到Spring容器

@Bean

public RestTemplateWrapper restTemplateWrapper(RestTemplate restTemplate) {

return new RestTemplateWrapper(restTemplate);

}

测试使用bean可以使用可以成功传递参数。

测试传递bean.jpeg

成功接收服务的响应

ok receive message ===> test

上面内容就是基于RestTemplate对Get请求参数扩展的处理了,如果有更多的实践方案欢迎一起探讨。