快捷導航

SpringBoot使用RestTemplate如何通過http請求將文件下載到本地

更新時間：2025年02月10日 08:49:12 作者：lijiabin417

文章介紹了如何通過編寫代碼批量下載文件,解決了沒有提供批量下載接口的問題,首先篩選出需要下載的文件ID,然后通過后端代碼發(fā)起HTTP請求,將下載的資源寫入本地文件中,總結(jié)了實現(xiàn)方式和注意事項,希望能為類似任務(wù)提供參考

背景

最近被安排了一個活，純體力的重復性工作，將開發(fā)一個項目的指定資源通過現(xiàn)有的下載接口下載下來。

思路

因為沒有提供批量下載接口，同時下載的資源需要自己篩選，想著這樣人工處理特別麻煩，個人也沒有什么進步，就想著寫一段代碼處理，這樣處理起來準確，效率高，后續(xù)有類似的任務(wù)還可以將這段代碼改改后繼續(xù)使用。

1.篩選

篩選出需要的下載文件的id，這個可以根據(jù)業(yè)務(wù)調(diào)整，不一定是id，具體的實現(xiàn)根據(jù)業(yè)務(wù)邏輯實現(xiàn)；

2.下載

下載方式有兩種，一種是通過postman或者apifox這種工具，通過提前構(gòu)建參數(shù)和提供寫有參數(shù)的文件，批量出發(fā)請求，這種只是大致想到的可以不靠編碼實現(xiàn)的方式；另外一種是借助后端代碼發(fā)起http請求，將下載的資源寫入本地文件中。下面簡單介紹第二種，需要時可以批量下載文件。

實現(xiàn)方式

資源下載方法

public static void downloadFile(String url, HttpHeaders headers, RequestParams requestParams, String outputPath) throws IOException {


        // 創(chuàng)建請求實體
        HttpEntity<RequestParams> entity = new HttpEntity<>(requestParams, headers);
        // 初始化RestTemplate，也可以自己構(gòu)建配置類。
        final RestTemplate restTemplate = new RestTemplate();
        // 請求配置
        ResponseEntity<Resource> response = restTemplate.exchange(
                URI.create(url),
                HttpMethod.POST,
                entity,
                Resource.class
        );

        // 請求成功校驗
        if (response.getStatusCode().is2xxSuccessful() && response.getBody() != null) {
            // 獲取資源
            Resource resource = response.getBody();

            // 確保目錄存在
            File outputFile = new File(outputPath);
            File parentDir = outputFile.getParentFile();
            // 文件存在校驗
            if (parentDir != null && !parentDir.exists()) {
                if (!parentDir.mkdirs()) {
                    throw new IOException("Failed to create directory: " + parentDir);
                }
            }
            //獲取流
            try (InputStream inputStream = resource.getInputStream();
                 OutputStream outputStream = Files.newOutputStream(outputFile.toPath())) {
                StreamUtils.copy(inputStream, outputStream);
            }
        } else {
            throw new IOException("Failed to download file: " + response.getStatusCode());
        }
    }

實體類（可以自己定義）

@Data
public class RequestParams implements Serializable {
    private Long id;
}

函數(shù)調(diào)用

    public static void downloadPerPaper(PaperInfo paperInfos) throws IOException {
		//url
        String url="https://xxxx.com";
        //請求頭
        HttpHeaders headers = new HttpHeaders();
        headers.set("xxx","xxxxx");
        //請求參數(shù)
        final RequestParams requestParams = new RequestParams();
        requestParams.setId(1L);
        //存在存放地址
        String path="xx/xx/xx.doc";
        //調(diào)用下載方法
        downloadFile(url,headers,requestParams,path);
    }