使用RestTemplate访问https请求并获取返回的输入流

最新推荐文章于 2024-11-09 17:34:14 发布

原创最新推荐文章于 2024-11-09 17:34:14 发布 · 4.4k 阅读

7 ·

CC 4.0 BY-SA版权

文章标签：

#https #java

本文介绍了如何在Java中利用RestTemplate库执行HTTPS请求，并详细讲解了如何正确处理返回的输入流，以便于进行数据解析和处理。

代码

package com.hyzs.szcg.doc.utils;

import com.hyzs.intelligentgateway.core.service.IntelligentGatewayService;
import lombok.extern.slf4j.Slf4j;
import org.apache.http.util.EntityUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.core.io.Resource;
import org.springframework.http.*;
import org.springframework.http.converter.HttpMessageConverter;
import org.springframework.stereotype.Component;
import org.springframework.web.client.RestTemplate;

import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;

@Slf4j
@Component
public class TencentApiUtil {
    @Value("${apiUrl}")
    private String apiUrl;
    @Autowired
    IntelligentGatewayService intelligentGatewayService;
    /**
     * GET
     *
     * @date 2019年11月28日
     */
    public  byte[] doGetHttps(String params) {
        // HttpHeaders  自己另外创建
        HttpHeaders httpHeaders = this.intelligentGatewayService.buildHttpHeaders();
        HttpEntity<String> requestEntity = new  HttpEntity<>(httpHeaders);
        RestTemplate restTemplate=new RestTemplate();
//如果返回的是String 则改用 String 若为文件流 则改用 Resource
        ResponseEntity<Resource> responseEntity = restTemplate.exchange(apiUrl+params, HttpMethod.GET,requestEntity, Resource.class);
        if (responseEntity.getStatusCodeValue()!= HttpStatus.OK.value()){
            return null;
        }
        if (responseEntity != null) {
            String contentType=responseEntity.getHeaders().getContentType().toString();
            if("image/png".equals(contentType)){
                InputStream responseInputStream=null;
                try {
                    responseInputStream = responseEntity.getBody().getInputStream();
                } catch (IOException e) {
                    e.printStackTrace();
                }
                if (responseInputStream == null) {
                    return null;
                } else {
                    try {
                        return toByteArray(responseInputStream);
                    } catch (IOException e) {
                        log.error("图片数据转化为byte[]失败");
                        return null;
                    }
                }
            }else
            {
                log.error("调用腾讯api失败="+ responseEntity.getBody());
            }
        }
        return null;
    }
    public static byte[] toByteArray(InputStream input) throws IOException {
        ByteArrayOutputStream output = new ByteArrayOutputStream();
        byte[] buffer = new byte[1024*4];
        int n = 0;
        while (-1 != (n = input.read(buffer))) {
            output.write(buffer, 0, n);
        }
        return output.toByteArray();
    }
}