批量下载

批量下载

         String agent = request.getHeader("USER-AGENT");
        String filenameEncoder = "";
        String zipFileName = eventDate.replace(",", "-");
        //压缩包名称
     
        if (agent.contains("MSIE")) {
            // IE浏览器
            try {
                filenameEncoder = URLEncoder.encode(zipFileName, "utf-8");
            } catch (UnsupportedEncodingException e) {
                e.printStackTrace();
            }
        } else if (agent.contains("Firefox")) {
            // 火狐浏览器
            BASE64Encoder base64Encoder = new BASE64Encoder();
            try {
                filenameEncoder = "=?utf-8?B?" + base64Encoder.encode(zipFileName.getBytes("utf-8")) + "?=";
            } catch (UnsupportedEncodingException e) {
                e.printStackTrace();
            }
        } else {
            // 其它浏览器
            try {
                filenameEncoder = URLEncoder.encode(zipFileName, "utf-8");
            } catch (UnsupportedEncodingException e) {
                log.error("设置文件名编码错误{{}}", e.getMessage());
                e.printStackTrace();
            }
        }

        //2.4设置响应头信息
        //设置文件ContentType类型，这样设置，会自动判断下载文件类型
        response.setContentType("application/octet-stream");
        //设置文件头：最后一个参数是设置下载文件名(假如我们叫a.pdf)
        response.setHeader("Content-Disposition", "attachment;fileName=" + filenameEncoder + ".zip");

        //2.5 写入下载文件
        ZipOutputStream zos = null;
        try {
            zos = new ZipOutputStream(response.getOutputStream());
            for (String fileName : files.keySet()) {
                String fileUrl = files.get(fileName);
                InputStream inputStream = uploadDownloadService.downloadFile(bucketName, fileUrl);
                if (inputStream == null) {
                    continue;
                }
                byte[] buffer = new byte[1024 * 10];
                int data = 0;
                zos.putNextEntry(new ZipEntry(fileName));
                while ((data = inputStream.read(buffer, 0, 1024 * 10)) != -1) {
                    zos.write(buffer, 0, data);
                }
                inputStream.close();
            }
        } catch (IOException e) {
            e.printStackTrace();
            log.error("批量下载文件错误{{}}", e.getMessage());
        }
        try {
            if (zos != null) {
                zos.flush();
                zos.close();
            }
        } catch (IOException e) {
            e.printStackTrace();
            log.error("批量下载文件错误{{}}", e.getMessage());
        }