下载 zip InputStream Struts2 Java

Question

首先，对不起英语。 我想知道你能不能帮我解决这个问题。 我正在尝试下载我使用 Struts2 从多个字节 [] 创建的 zip 并流式传输结果。

我使用 ZipOutputStream 并设法从中创建一个文件并使用 FileInputStream 读取和下载它，但我的问题是我不想创建一个文件。我只想将 ZipOutputStream 转换为 InputStream（例如转换为 ZipIntputStream）并下载该 ZipInputStream。 为此，我使用以下代码：

public void downloadZip() {
    contentType = "application/octet-stream";
    filename="myZip.zip";
    ByteArrayOutputStream baos = new ByteArrayOutputStream();       
    byte[] bytes;
    try {
        ZipOutputStream zos = new ZipOutputStream(baos);
        ZipEntry ze;
        bytes = otherClass.getBytes("File1");
        ze = new ZipEntry("File1.pdf");
        zos.putNextEntry(ze);
        zos.write(bytes);
        zos.closeEntry();

        bytes = otherClass.getBytes("File2");
        ze = new ZipEntry("File2.pdf");
        zos.putNextEntry(ze);
        zos.write(bytes);
        zos.closeEntry();

        zos.flush();
        inputStream = new ZipInputStream(new ByteArrayInputStream(baos.toByteArray()));
        zos.close();
}
catch(Exception e){...}
}

我的动作 struts.xml

<action...>
    <result name="success" type="stream">
        <param name="contentType">${contentType}</param>
        <param name="inputName">inputStream</param>
        <param name="contentDisposition">attachment;filename="${filename}"</param>
        <param name="bufferSize">1024</param>
    </result>
</action>

问题是浏览器向我显示一条消息，告诉我该文件不是有效的 zip，其大小为 0 字节。

希望我解释清楚，提前非常感谢。

编辑：正如我所评论的，终于我得到了解决方案，它与 leonbloy 的回复非常相似。除了返回 ByteArrayInputStream 我应该在创建 ByteArrayInputStream 之前关闭 ZipOutputStream。这是结果代码，也许对其他人有用：

    ...
    zos.closeEntry();                       
    zos.close();            
    inputStream = new ByteArrayInputStream(baos.toByteArray());
}
catch(Exception e){...}
}

感谢您的帮助。

Answer 1

参数<param name="inputName">inputStream</param> 告诉Struts2 从哪里获取将发送到客户端的原始字节。在您的情况下，您想要发送压缩字节。相反，您正在设置 inputStream=ZipInputStream ，这是一个需要压缩源的流 - 解压缩。你不想这样，你想发送原始压缩字节。

然后替换

inputStream = new ZipInputStream(new ByteArrayInputStream(baos.toByteArray()))

由

inputStream = new ByteArrayInputStream(baos.toByteArray())

它应该可以工作

Answer 2

ZipInputStream 用于解压缩文件并访问未压缩的 zip 内容。您只想将压缩字节发送到浏览器。

创建 zip 文件后，只需使用 response.getOutputStream(baos.toByteArray()) 发送文件的字节即可。

如果您压缩大文件，则在内存中执行所有操作可能会导致内存不足。如果输入文件太大，最好限制输入文件的大小并给用户一个错误。

另外，看起来 inputStream 是一个类变量。对于 servlet，这是一个非常糟糕的主意。基本上你的类不是线程安全的，当两个用户同时访问该页面时，一个用户可能会得到另一个用户的 zip 文件。相反，将响应对象作为参数传递给方法或将 baos 返回给调用者，以便它可以将字节写入正确的响应流。摆脱存储特定于单个用户的信息的类变量，并在调用之间传递它们。

Answer 3

你很幸运，this is exactly what you need。

在这种特定情况下，我通过直接写入响应来绕过框架结果系统。这样，Zip 将立即在客户端系统中创建，并逐步提供，而不是等待精化结束并与 Stream 结果一起输出。

然后没有定义结果：

<action name="createZip" class="foo.bar.CreateZipAction" />

在行动中：

public String execute() {
    try {        
        /* Read the amount of data to be streamed from Database to File System,
           summing the size of all Oracle's BLOB, PostgreSQL's ABYTE etc: 
           SELECT sum(length(my_blob_field)) FROM my_table WHERE my_conditions
        */          
        Long overallSize = getMyService().precalculateZipSize();

        // Tell the browser is a ZIP
        response.setContentType("application/zip"); 
        // Tell the browser the filename, and that it needs to be downloaded instead of opened
        response.addHeader("Content-Disposition", "attachment; filename=\"myArchive.zip\"");        
        // Tell the browser the overall size, so it can show a realistic progressbar
        response.setHeader("Content-Length", String.valueOf(overallSize));      

        ServletOutputStream sos = response.getOutputStream();       
        ZipOutputStream zos = new ZipOutputStream(sos);

        // Set-up a list of filenames to prevent duplicate entries
        HashSet<String> entries = new HashSet<String>();

        /* Read all the ID from the interested records in the database, 
           to query them later for the streams: 
           SELECT my_id FROM my_table WHERE my_conditions */           
        List<Long> allId = getMyService().loadAllId();

        for (Long currentId : allId){
            /* Load the record relative to the current ID:         
               SELECT my_filename, my_blob_field FROM my_table WHERE my_id = :currentId            
               Use resultset.getBinaryStream("my_blob_field") while mapping the BLOB column */
            FileStreamDto fileStream = getMyService().loadFileStream(currentId);

            // Create a zipEntry with a non-duplicate filename, and add it to the ZipOutputStream
            ZipEntry zipEntry = new ZipEntry(getUniqueFileName(entries,fileStream.getFilename()));
            zos.putNextEntry(zipEntry);

            // Use Apache Commons to transfer the InputStream from the DB to the OutputStream
            // on the File System; at this moment, your file is ALREADY being downloaded and growing
            IOUtils.copy(fileStream.getInputStream(), zos);

            zos.flush();
            zos.closeEntry();

            fileStream.getInputStream().close();                    
        }

        zos.close();
        sos.close();    

    } catch (Exception e) {
        logError(e);
    finally {
        IOUtils.closeQuietly(sos);
    }

    return NONE;
}