Java多文件生成并压缩下载功能(思路详解)

我们来详细的讲解一下“Java多文件生成并压缩下载功能(思路详解)”：

简介

本文讲述的是在Java Web应用中实现多文件生成并压缩下载功能的实现方法，主要的思路是将文件依次读取到内存中，然后利用Java ZipOutputStream类进行压缩，最后将生成的压缩文件发送给客户端。

步骤

第一步：获取文件列表

我们可以通过前端传递一个数组，数组中包含要下载的文件名或者文件路径，我们服务器后台根据传递的参数，转化成文件列表，存放于一个List中。

第二步：将文件依次读取到内存中

我们可以使用Java中的FileInputStream类，为每个文件创建一个输入流，并将它们分别读取到内存中。

List<FileInputStream> fileList = new ArrayList<FileInputStream>();
for (String filePath : filePathList) {
    File file = new File(filePath);
    if(file.exists()) {
        try {
            fileList.add(new FileInputStream(file));
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        }
    }
}

第三步：利用Java ZipOutputStream类进行压缩

我们可以将文件列表的内容利用ZipOutputStream类进行压缩，将所有文件压缩到一个文件夹中，最后压缩成zip格式即可。

ZipOutputStream zipOut = new ZipOutputStream(response.getOutputStream());
for (FileInputStream inputStream : fileList) {
    ZipEntry zipEntry = new ZipEntry(inputStream.getFD().toString());
    zipOut.putNextEntry(zipEntry);
    IOUtils.copy(inputStream, zipOut);
    inputStream.close();
    zipOut.closeEntry();
}

第四步：设置压缩包的文件名并发送给客户端

最后一步，我们需要将生成的zip文件发送给客户端，这里需要设置一下响应头Content-Disposition，这样客户端就可以弹出文件下载框，让用户选择保存文件的路径及文件名。

response.setHeader("Content-Disposition", "attachment;filename=" + zipFileName);
ServletOutputStream outputStream = response.getOutputStream();
zipOut.finish();
IOUtils.closeQuietly(zipOut);
IOUtils.copy(new ByteArrayInputStream(zipFile.toByteArray()), outputStream);
outputStream.flush();

在以上四步实现后，我们就可以愉快地进行多文件生成并压缩下载了！

示例

为了使读者更好地理解上述思路，接下来我们来演示两条简单的示例。

示例一：压缩本地文件并进行下载

@RequestMapping("/download")
public void download(HttpServletRequest request, HttpServletResponse response) throws Exception {
    String filePath1 = "D:\temp\file1.jpg";
    String filePath2 = "D:\temp\file2.txt";
    String[] filePathList = {filePath1, filePath2};
    String zipFileName = "download.zip";
    List<FileInputStream> fileList = new ArrayList<FileInputStream>();
    for (String filePath : filePathList) {
        File file = new File(filePath);
        if(file.exists()) {
            try {
                fileList.add(new FileInputStream(file));
            } catch (FileNotFoundException e) {
                e.printStackTrace();
            }
        }
    }
    ZipOutputStream zipOut = new ZipOutputStream(response.getOutputStream());
    for (FileInputStream inputStream : fileList) {
        ZipEntry zipEntry = new ZipEntry(inputStream.getFD().toString());
        zipOut.putNextEntry(zipEntry);
        IOUtils.copy(inputStream, zipOut);
        inputStream.close();
        zipOut.closeEntry();
    }
    response.setHeader("Content-Disposition", "attachment;filename=" + zipFileName);
    ServletOutputStream outputStream = response.getOutputStream();
    zipOut.finish();
    IOUtils.closeQuietly(zipOut);
    IOUtils.copy(new ByteArrayInputStream(zipFile.toByteArray()), outputStream);
    outputStream.flush();
}

示例二：压缩服务器文件并进行下载

@RequestMapping("/download")
public void download(HttpServletRequest request, HttpServletResponse response) throws Exception {
    String url1 = "http://localhost:8080/files/file1.jpg";
    String url2 = "http://localhost:8080/files/file2.txt";
    String[] urlList = {url1, url2};
    String zipFileName = "download.zip";
    List<FileInputStream> fileList = new ArrayList<FileInputStream>();
    for (String url : urlList) {
        URL fileUrl = new URL(url);
        URLConnection conn = fileUrl.openConnection();
        if(conn.getContentLength() > 0) {
            InputStream inputStream = conn.getInputStream();
            fileList.add(new FileInputStream(IOUtils.toByteArray(inputStream)));
            inputStream.close();
        }
    }
    ZipOutputStream zipOut = new ZipOutputStream(response.getOutputStream());
    for (FileInputStream inputStream : fileList) {
        ZipEntry zipEntry = new ZipEntry(inputStream.getFD().toString());
        zipOut.putNextEntry(zipEntry);
        IOUtils.copy(inputStream, zipOut);
        inputStream.close();
        zipOut.closeEntry();
    }
    response.setHeader("Content-Disposition", "attachment;filename=" + zipFileName);
    ServletOutputStream outputStream = response.getOutputStream();
    zipOut.finish();
    IOUtils.closeQuietly(zipOut);
    IOUtils.copy(new ByteArrayInputStream(zipFile.toByteArray()), outputStream);
    outputStream.flush();
}

至此，我们已经完成了Java多文件生成并压缩下载功能的实现。