下载多个文件 Java Spring

Question

我正在尝试在我的Spring-MVC应用程序中使用一个HTTP get请求来下载多个文件。

我已经查看了其他帖子，说你可以压缩文件并发送这个文件，但在我的情况下不理想，因为文件无法直接从应用程序访问。要获取文件，我必须查询REST接口，该接口从hbase或hadoop流式传输文件。

我可能有超过1 GB的文件，因此将文件下载到存储库中，将其压缩并将其发送到客户端需要太长时间。（考虑到大文件已经被压缩，再次压缩不会使它们更小）。

我看到这里和那里说可以使用multipart-response一次下载多个文件，但我无法得到任何结果。这是我的代码：

String boundaryTxt = "--AMZ90RFX875LKMFasdf09DDFF3";
response.setContentType("multipart/x-mixed-replace;boundary=" + boundaryTxt.substring(2));
ServletOutputStream out = response.getOutputStream();
        
// write the first boundary
out.write(("\r\n"+boundaryTxt+"\r\n").getBytes());

String contentType = "Content-type: application/octet-stream\n";
        
for (String s:files){
    System.out.println(s);
    String[] split = s.trim().split("/");
    db = split[1];
    key = split[2]+"/"+split[3]+"/"+split[4];
    filename = split[4];
            
    out.write((contentType + "\r\n").getBytes());
    out.write(("\r\nContent-Disposition: attachment; filename=" +filename+"\r\n").getBytes());
    
    InputStream is = null;
    if (db.equals("hadoop")){
        is = HadoopUtils.get(key);
    }
    else if (db.equals("hbase")){
        is = HbaseUtils.get(key);
    }
    else{
        System.out.println("Wrong db with name: " + db);
    }
    byte[] buffer = new byte[9000]; // max 8kB for http get
    int data;
    while((data = is.read(buffer)) != -1) { 
        out.write(buffer, 0, data);
    } 
    is.close(); 
       
    // write bndry after data
    out.write(("\r\n"+boundaryTxt+"\r\n").getBytes());
    response.flushBuffer();
    }
// write the ending boundary
out.write((boundaryTxt + "--\r\n").getBytes());
response.flushBuffer();
out.close();
}   

奇怪的是，我的浏览器会得到不同的结果。在Chrome中什么都没有发生（查看了控制台），而在Firefox中，我得到了一个提示，要求为每个文件下载，但它既没有正确的类型也没有正确的名称（控制台中也没有任何内容）。

我的代码有什么错误吗？如果没有，有没有其他替代方法？

编辑

我还看到了这篇文章：无法将multipart/mixed请求发送到基于Spring MVC的REST服务 编辑2 这个文件的内容就是我想要的，但为什么我不能得到正确的文件名，为什么Chrome不能下载任何东西？

- Whitefret

---

你可以在JS中循环下载请求，每个请求完成后会触发一个新的下载。我认为这比绕过多个下载触发器更容易。 - We are Borg

@WeareBorg，我只有基本的JS知识，你能给我一些资源吗？ - Whitefret

很遗憾，我是一名后端开发人员，甚至不会JS。如果您愿意，我可以给您提供Java的ZIP下载代码。 - We are Borg

@WeareBorg 可能有一种方法可以使用HDFS制作zip文件，问题是我还有Hbase中的文件。 - Whitefret

@WeareBorg 目前我正在进行的工作是纯流式传输，因此我不会在服务器上保留数据（除了缓存）。但我无法使多部分正常工作。现在问题已经解决。 - Whitefret

Answer 1

这是通过zip进行下载的方法：

try {
      List<GroupAttachments> groupAttachmentsList = attachIdList.stream().map(this::getAttachmentObjectOnlyById).collect(Collectors.toList()); // Get list of Attachment objects
            Person person = this.personService.getCurrentlyAuthenticatedUser();
            String zipSavedAt = zipLocation + String.valueOf(new BigInteger(130, random).toString(32)); // File saved location
            byte[] buffer = new byte[1024];
            FileOutputStream fos = new FileOutputStream(zipSavedAt);
            ZipOutputStream zos = new ZipOutputStream(fos);

                GroupAttachments attachments = getAttachmentObjectOnlyById(attachIdList.get(0));

                    for (GroupAttachments groupAttachments : groupAttachmentsList) {
                            Path path = Paths.get(msg + groupAttachments.getGroupId() + "/" +
                                    groupAttachments.getFileIdentifier());   // Get the file from server from given path
                            File file = path.toFile();
                            FileInputStream fis = new FileInputStream(file);
                            zos.putNextEntry(new ZipEntry(groupAttachments.getFileName()));
                            int length;

                            while ((length = fis.read(buffer)) > 0) {
                                zos.write(buffer, 0, length);
                            }
                            zos.closeEntry();
                            fis.close();

                    zos.close();
                    return zipSavedAt;
            }
        } catch (Exception ignored) {
        }
        return null;
    }

用于下载zip的控制器方法:

 @RequestMapping(value = "/URL/{mode}/{token}")
    public void downloadZip(HttpServletResponse response, @PathVariable("token") String token,
                            @PathVariable("mode") boolean mode) {
        response.setContentType("application/octet-stream");
        try {
            Person person = this.personService.getCurrentlyAuthenticatedUser();
            List<Integer> integerList = new ArrayList<>();
            String[] integerArray = token.split(":");
            for (String value : integerArray) {
                integerList.add(Integer.valueOf(value));
            }
            if (!mode) {
                String zipPath = this.groupAttachmentsService.downloadAttachmentsAsZip(integerList);
                File file = new File(zipPath);
                response.setHeader("Content-Length", String.valueOf(file.length()));
                response.setHeader("Content-Disposition", "attachment; filename=\"" + person.getFirstName() + ".zip" + "\"");
                InputStream is = new FileInputStream(file);
                FileCopyUtils.copy(IOUtils.toByteArray(is), response.getOutputStream());
                response.flushBuffer();
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

如果有疑问，请告知，祝您愉快。

更新

将字节数组压缩成ZIP文件。您可以像我给出的第一种方法一样在循环中使用此代码：

public static byte[] zipBytes(String filename, byte[] input) throws IOException {
    ByteArrayOutputStream baos = new ByteArrayOutputStream();
    ZipOutputStream zos = new ZipOutputStream(baos);
    ZipEntry entry = new ZipEntry(filename);
    entry.setSize(input.length);
    zos.putNextEntry(entry);
    zos.write(input);
    zos.closeEntry();
    zos.close();
    return baos.toByteArray();
}

- We are Borg

Answer 2

您可以使用multipart/x-mixed-replace内容类型来实现。您可以像这样添加：response.setContentType("multipart/x-mixed-replace;boundary=END"); 并通过循环遍历文件并将每个文件写入响应输出流中。

您可以查看这个示例作为参考。

另一种方法是创建一个REST端点，允许您下载单个文件，然后针对每个文件单独调用此端点。

- Kuncheria