RestTemplate发送请求下载MP4文件

吗喽.0

已于 2024-08-28 10:30:07 修改

阅读量587

点赞数 9

分类专栏： RestTemplate 文章标签： java spring boot

于 2024-03-12 11:01:06 首次发布

本文链接：https://blog.youkuaiyun.com/m0_49595489/article/details/136645579

版权

RestTemplate 专栏收录该内容

1 篇文章

订阅专栏

本文介绍如何在Spring框架中使用RestTemplate配合HttpMessageConverter处理大文件下载，避免内存溢出问题，通过流式方式将文件保存到本地。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

普通的方式不能下载大文件,会oom,要下载大文件建议使用流式处理下载

import org.springframework.http.converter.HttpMessageConverter;
import org.springframework.http.converter.ByteArrayHttpMessageConverter;
import org.springframework.web.client.RestTemplate;
import org.springframework.http.ResponseEntity;
import org.springframework.http.HttpHeaders;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;

public class FileDownloader {

    public static void downloadFile(String fileUrl, String localFilePath) {
        RestTemplate restTemplate = new RestTemplate();
        
        // 添加一个能处理video/mp4类型响应的HttpMessageConverter
        restTemplate.getMessageConverters().add(new ByteArrayHttpMessageConverter());
        
        ResponseEntity<byte[]> response = restTemplate.getForEntity(fileUrl, byte[].class);
        
        if (response.getStatusCode().is2xxSuccessful()) {
            HttpHeaders headers = response.getHeaders();
            try (InputStream in = new ByteArrayInputStream(response.getBody());
                 FileOutputStream out = new FileOutputStream(localFilePath)) {
                byte[] buffer = new byte[1024];
                int bytesRead;
                while ((bytesRead = in.read(buffer)) != -1) {
                    out.write(buffer, 0, bytesRead);
                }
                System.out.println("File downloaded successfully to: " + localFilePath);
            } catch (IOException e) {
                System.err.println("Error downloading file: " + e.getMessage());
            }
        } else {
            System.err.println("Error downloading file. HTTP status code: " + response.getStatusCodeValue());
        }
    }

    public static void main(String[] args) {
        String fileUrl = "http://example.com/file-to-download.mp4";
        String localFilePath = "/path/to/save/file.mp4";
        downloadFile(fileUrl, localFilePath);
    }
}