快捷導航

Java實現(xiàn)網(wǎng)絡(luò)文件下載以及下載到指定目錄

更新時間：2024年09月26日 09:01:20 作者：學習游泳的喵

在Spring框架中,StreamUtils和FileCopyUtils兩個工具類提供了方便的文件下載功能,它們都屬于org.springframework.util包,可以通過簡單的方法調(diào)用實現(xiàn)文件流的復(fù)制和下載,這些工具類支持多種參數(shù)傳遞,涵蓋了文件下載的多種場景

前言

最近項目中有使用到pdf文件下載功能，在使用過程中，發(fā)現(xiàn)spring給我們提供了兩個工具類，已經(jīng)實現(xiàn)了文件下載的功能：

StreamUtils和FileCopyUtils，都是org.springframework.util包下面的類，使用非常方便。

相關(guān)代碼

service實現(xiàn)類

	public InputStream getDownloadInputStreamByUrl(String downloadUrl) {
		if (ObjectUtils.isEmpty(downloadUrl)) {
            throw new RuntimeException("無效參數(shù)：fileUrl 不能為空");
        }
		
		InputStream in = null;
		try {
			URL url = new URL(downloadUrl);

			HttpURLConnection conn = (HttpURLConnection) url.openConnection();
			// 設(shè)置超時時間為3秒
			conn.setConnectTimeout(3 * 1000);
			// 防止屏蔽程序抓取而返回403錯誤
			conn.setRequestProperty("User-Agent", "Mozilla/4.0 (compatible; MSIE 5.0; Windows NT; DigExt)");
			// 得到輸入流
			in = conn.getInputStream();
		} catch (Exception e) {
			throw new RuntimeException(e);
		}
		return in;
	}

Controller類

	@GetMapping("/fileDownload/{filename}")
    public void downloadPdfFile(HttpServletResponse response, @PathVariable(name="filename") String filename) {
        try {
            if (StringUtils.isEmpty(filename)) {
            	throw new RuntimeException("文件名不可為空！");
            }
            //通過文件名查找文件對應(yīng)的url
            String downloadUrl = fileDownloadService.getUrlByFiname(filename);
            if (StringUtils.isEmpty(filename)) {
            	throw new RuntimeException("文件對應(yīng)地址不存在");
            }
            // 此處異常方法中已處理
            InputStream in = fileDownloadService.getDownloadInputStreamByUrl(downloadUrl);
   			// 設(shè)置編碼格式
            response.setCharacterEncoding("UTF-8");
            /* 若要下載指定目錄，以下兩行需注釋，否則會下載兩份，一份到指定目錄，一份到瀏覽器下載目錄 */
            // 設(shè)置下載格式
            response.setHeader("Content-Type", "application/pdf");
            // 注釋這行代碼，可實現(xiàn) 預(yù)覽 功能(只限pdf文件)
            response.setHeader("Content-Disposition","attachment;filename=" + URLEncoder.encode(filename, "UTF-8"));
			//下載到指定目錄
			/*
			File file = new File("C:\\tmp");
            if (!file.exists()) {
                //如果文件夾不存在，則創(chuàng)建新的的文件夾
                file.mkdirs();
            }

            File file1 = new File(file.getPath() + File.separator + filename);
           	OutputStream outputStream = new FileOutputStream(file1);
           	FileCopyUtils.copy(in, outputStream);
           	*/

            StreamUtils.copy(in, response.getOutputStream());
        } catch (Exception e) {
        	throw new RuntimeException(e);
        } 
    }

此處通過spring提供的工具類進行文件下載。

工具類中實現(xiàn)方式：

此處也可通過FileCopyUtils.copy(InputStream input, OutputStream output)實現(xiàn)文件下載，其底層也是通過調(diào)用StreamUtils.copy(InputStream in, OutputStream out)方法實現(xiàn)的，不過此方法最后通過finally關(guān)閉了InputStream和OutputStream，邏輯更嚴謹。