spring boot单元测试之RestTemplate(二)

上篇博客中，简单介绍了RestTemplate，只是用到了单元测试环节，如果在正式开发环境使用RestTemplate调用远程接口，还有一些配置要做。

一、配置类

由于Spring boot没有对RestTemplate做自动配置，所以我们需要写一个配置类引入RestTemplate。

@Configuration

public class RestTemplateConfig {

    @Bean

    public RestTemplate restTemplate(ClientHttpRequestFactory factory) {

        return new RestTemplate(factory);

    }

    @Bean

    public ClientHttpRequestFactory simpleClientHttpRequestFactory() {

        SimpleClientHttpRequestFactory factory = new SimpleClientHttpRequestFactory();

        factory.setReadTimeout(5000);//单位为ms

        factory.setConnectTimeout(5000);//单位为ms

        return factory;

    }

}

二、工具类

在实际开发过程中，可能对于调用远程接口有多种参数情况，因此写了一个工具类用来支持。

RestTemplate功能非常强大，Post请求的时候特别要注意，body参数必须为MultiValueMap类型。

@Service

public class RestTemplateService {

    @Autowired

    private RestTemplate restTemplate;

    /**

     * 实测可用

     */

    public JSONObject doGet(String url) {

        JSONObject json = restTemplate.getForEntity(url, JSONObject.class).getBody();

        return json;

    }

    /**

     * 实测可用

     */

    public JSONObject doGet(String url, HttpHeaders headers) {

        HttpEntity<String> entity = new HttpEntity<>(null, headers);

        ResponseEntity<JSONObject> exchange = restTemplate.exchange(url, HttpMethod.GET, entity, JSONObject.class);

        return exchange.getBody();

    }

    /**

     * 实测可用，body参数必须为MultiValueMap类型

     */

    public JSONObject doPost(String url, MultiValueMap<String, Object> param) {

        JSONObject json = restTemplate.postForEntity(url, param, JSONObject.class).getBody();

        return json;

    }

    /**

     * 实测可用，body参数必须为MultiValueMap类型

     */

    public JSONObject doPost(String url, MultiValueMap<String, Object> params, HttpHeaders headers) {

        JSONObject json = restTemplate.postForObject(url, new HttpEntity<>(params, headers), JSONObject.class);

        return json;

    }

}

巴特西

spring boot单元测试之RestTemplate(二)

一、配置类

二、工具类

最新文章

热门文章