java下載多文件夾的文件最終打zip包返回給用戶

業務需求,需要做一個導出用戶單據圖片的功能,需求爲:按單據單號分文件夾分別存儲單據的圖片!最終導出zip包給用戶。通過網上各種找代碼,寫出一個小demo,供各位看官學習!

        
        // 假裝是一個單據集合
        List<String> imgUrls = new ArrayList<>();
        imgUrls.add("http://img1.imgtn.bdimg.com/it/u=2171717409,2602158110&fm=26&gp=0.jpg");
        imgUrls.add("http://images.ofweek.com/Upload/News/2016-5/Finn/53/20160503055505730.jpg");
        imgUrls.add("http://y1.ifengimg.com/a/2016_04/4855d9fcb2ccade.jpg");

        try {
            //zip包的名稱
            String downloadFilename = "圖片.zip";
            downloadFilename = URLEncoder.encode(downloadFilename, "UTF-8");
            // 指明response的返回對象是文件流
            response.setContentType("application/octet-stream");
            // 設置在下載框默認顯示的文件名
            response.setHeader("Content-Disposition", "attachment;filename=" + downloadFilename);
            @Cleanup ZipOutputStream zos = new ZipOutputStream(response.getOutputStream());
            imgUrls.forEach(i -> {
                InputStream fis = null;
                try {
                    // 跨平臺建文件夾和文件名字
                    // File.separator  根據操作系統來判斷用什麼文件分隔符
                    // 這裏只建立了一個文件夾,如果想分類的話,在後面加上想要的文件夾名稱即可
                    ZipEntry zipEntry = new ZipEntry("img_download" + File.separator +imgUrls.indexOf(i)+ ".jpg");
                    // 放到壓縮包裏
                    zos.putNextEntry(zipEntry);
                    // 獲得圖片流
                    fis = getInputStreamByGet(i);
                    if (null != fis) {
                        byte[] buffer = new byte[1024];
                        int r = 0;
                        // 從輸入流讀取一定數量的字節,存到緩存區中
                        while ((r = fis.read(buffer)) != -1) {
                            zos.write(buffer, 0, r);
                        }

                    }
                } catch (Exception e) {
                    e.printStackTrace();
                }
            });
            zos.flush();
        } catch (Exception e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    }
    
    /**
     * 通過get請求得到讀取器響應數據的數據流
     * @param url
     * @return
     */
    private static InputStream getInputStreamByGet(String url) {
        try {
            HttpURLConnection conn = (HttpURLConnection) new URL(url).openConnection();
            conn.setReadTimeout(5000);
            conn.setConnectTimeout(5000);
            conn.setRequestMethod("GET");
            if (conn.getResponseCode() == HttpURLConnection.HTTP_OK) {
                return conn.getInputStream();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
        return null;
    }

寫到這裏就聯想到用這個方法可以從網上爬一些照片到本地!

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章