自定义HttpMessageConverter实现RestTemplate的exchange方法返回自定义格式数据

一 概述

 实现如下效果代码,且可正常获取到返回数据:

ResponseEntity resEntity = restTemplate
                                    .exchange(url, HttpMethod.GET, requestEntity, JsonObject.class,uriVariables);

 restTemplate的get和POST请求都有类型参数设置,所以可以直接返回转换成自定义的格式.比如:

restTemplate.getForObject("http://www.baiud.com",JsonObject.class);

但是exchange是无法指定返回类型的,只能使用spring默认提供的几种转换器转换的格式,定义其他格式会报错或者返回实体body为空

restTemplate的post方法支持设置header,但是get方法不支持,假如现在有个需求,使用get方法需要设置header权限,且返回我想要的格式,那么自定义HttpMessageConverter就派上用场了

也提现了spring非常强大的可扩展性

二 步骤

新建自定义转换类:

/**
 * 接口请求返回数据格式转换处理类,只处理String->JsonObject
 */
public class StringToJsonObjectMessageConverter extends AbstractHttpMessageConverter {

    public StringToJsonObjectMessageConverter() {
        super(StandardCharsets.UTF_8,MediaType.APPLICATION_JSON,MediaType.APPLICATION_XML); //设置支付类型
    }
    /**
     * HttpMessageConvert是否处理接受的这个类
     */
    @Override
    protected boolean supports(Class aClass) {
        //isAssignableFrom: 判定此Class对象所表示的类或接口与指定的 Class 参数所表示的类或接口是否相同，或是否是其超类或超接口
        return JsonObject.class==aClass;
    }

    //对数据进行处理并返回
    @Override
    protected JsonObject readInternal(Classextends JsonObject> aClass, HttpInputMessage httpInputMessage) throws IOException, HttpMessageNotReadableException {
        //数据转换
        String temp = StreamUtils.copyToString(httpInputMessage.getBody(),Charset.forName("UTF-8"));
        return Tools.getGsonInstance().fromJson(temp, JsonObject.class);
    }

    //处理如何输出数据到response
    @Override
    protected void writeInternal(JsonObject jsonObject, HttpOutputMessage httpOutputMessage) throws IOException, HttpMessageNotWritableException {
        //可继续处理转换后的格式数据
        httpOutputMessage.getBody().write(jsonObject.getAsByte());
    }
}

配置修改自定义类顺序:

    @Bean
    public RestTemplate restTemplate() {
        HttpComponentsClientHttpRequestFactory httpRequestFactory = new HttpComponentsClientHttpRequestFactory();
        httpRequestFactory.setConnectionRequestTimeout(30 * 1000);
        httpRequestFactory.setConnectTimeout(30 * 3000);
        httpRequestFactory.setReadTimeout(30 * 3000);
        RestTemplate restTemplate = new RestTemplate(httpRequestFactory);
        // 支持返回JsonObject类型数据,需要在String之前,因为String支付所有的类型会直接解析
        restTemplate.getMessageConverters().set(1,new StringToJsonObjectMessageConverter());
        //spring转换默认编码是ISO_8859_1,如果乱码可修改支持中文编码
        restTemplate.getMessageConverters().add(new StringHttpMessageConverter(StandardCharsets.UTF_8));
        return restTemplate;
    }

断点查看:

三 原理

exchage方法源码点进去,看一下就知道了

        //获取HttpMessageConverter集合
        Iterator var8 = RestTemplate.this.getMessageConverters().iterator();
        //按照索引顺序进行迭代循环解析
        while(var8.hasNext()) {
            HttpMessageConverter messageConverter = (HttpMessageConverter)var8.next();
            if (messageConverter instanceof GenericHttpMessageConverter) {
                GenericHttpMessageConverter