当前位置：网站首页>RestTemplate源码debug：可变形参引发的问题

RestTemplate源码debug：可变形参引发的问题

2022-07-29 17:50:00 【ZJH'blog】

今天在做一个跨域请求的时候遇到一个问题，我需要同时从JDK7版本的项目接口中取出设置的请求头和响应的二进制流数据，按着源码分析了半天，把API倒是玩明白了，最后发现是因为没有注意可变形参引发的问题。。因为之前没有用过这个API，所以出这个问题的时候也没往这方面想，纯属是debug排除了所有问题。。。记录一下吧

1.环境模拟

1.1接口提供

提供一个接口，设置一个请求头给response，把图片二进制流写入response

@GetMapping("/pic1")
public void getPic(HttpServletRequest request,HttpServletResponse response) throws IOException {
    response.setHeader("keyy","valuee");
    System.out.println(response.getHeader("keyy"));
    File file = new File("C:\\Users\\zhangjiahao\\Desktop\\变更图片\\班级活动.jpg");
    FileInputStream fis = new FileInputStream(file);
    BufferedInputStream bis = new BufferedInputStream(fis);
    byte[] buf = new byte[2048];
    ServletOutputStream os = response.getOutputStream();
    BufferedOutputStream bos = new BufferedOutputStream(os);
    while(bis.read(buf) != -1){
        bos.write(buf);
    }
    bos.flush();
    bos.close();
    bis.close();
  }

1.2工具类（2个）

前提须知：

getForEntity() 和 exchange()本质上都是调用的excute()
返回值都是ResponseEntity
产生的问题是，这个ResponseEntity最后在getBody()的时候，一个能读取到二进制流，一个读取不到

1.2.1使用getForEntity()

public static String getStreamAndHeader(String url,String headKey,HttpServletResponse response) {
    RestTemplate restTemplate = getSingleRestTemplate();
    ResponseEntity<byte[]> responseEntity = restTemplate.getForEntity(url, null, byte[].class, (Object) null);

    byte[] res = responseEntity.getBody();//获取不到。res=null
    HttpHeaders headers = responseEntity.getHeaders();
    String      value   = headers.getFirst(headKey);
    try {
        ServletOutputStream os = response.getOutputStream();
        os.write(res);
        os.flush();
        os.close();
    }
    catch (IOException e) {
        logger.error("请求接口二进制流写入response时异常{}",e);
    }

    return value;

}

1.2.2使用exchange()

public static void getStreamWithNoParam(String url, HttpMethod method, HttpServletResponse response) {
    RestTemplate restTemplate = getSingleRestTemplate();
    try {
        ResponseEntity<byte[]> exchange = restTemplate.exchange(url, method, null, byte[].class, (Object) null);


        byte[]                 body     = exchange.getBody();//正常获取
        String value = exchange.getHeaders().getFirst("keyy");
        response.setHeader("keyy",value);
        response.getOutputStream().write(body);
    }
    catch (IOException e) {
        logger.error("RestTemplateUtils获取接口二进制流异常");
    }
}