我们可以在没有临时文件的情况下在 java 中创建文件/zipfile 作为 InputStream

Question

我需要创建一个测试，我需要在其中传递 InputStream 作为输入，这将是一个 zip 文件。有没有办法我们可以直接创建InputStream，而不是在内存/本地文件副本中创建，以便可以使用流进行测试。

Answer 1

如果我正确理解您的问题，您想从硬编码数据创建一个内存中（即不在磁盘上）的 ZIP 文件。这当然可以使用以下类：

• java.io.ByteArrayInputStream
• java.io.ByteArrayOutputStream
• java.util.zip.ZipOutputStream
• java.util.zip.ZipEntry

这是一个例子：

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

public class Main {

  public static void main(String[] args) throws IOException {
    // Destination of the ZIP file (an in-memory byte array)
    ByteArrayOutputStream boas = new ByteArrayOutputStream();

    /*
     * Write the ZIP file. This creates a single entry named "file.txt"
     * with "Hello, World!" as its contents.
     */
    try (ZipOutputStream zos = new ZipOutputStream(boas)) {
      zos.putNextEntry(new ZipEntry("file.txt"));
      zos.write("Hello, World!".getBytes());
      zos.closeEntry();
    }

    // Create an InputStream to read the raw bytes of the ZIP file
    ByteArrayInputStream bois = new ByteArrayInputStream(boas.toByteArray());

    /*
     * The following writes the ZIP file to disk, specifically to a file named "test.zip"
     * in the working directory. The purpose of this is to allow you to run the code
     * and see a tangible result (i.e. lets you inspect the resulting ZIP file). Obviously
     * you would not do this in your own code since you want to avoid writing the ZIP file
     * to disk.
     * 
     * Note: Will fail if the file already exists.
     */
    Files.copy(bois, Path.of("test.zip"));
  }
}

Answer 2

详细说明上面找到的一些答案，让我描述一下我遇到的情况以及我是如何解决的。我遇到了一种情况，我收到了 PGP 加密的文件（.pgp、.asc 等），所以我必须解密文件（.txt）本身，然后将其传递给流阅读器来解析行条目。而不是在读入之前将解密的文件作为临时文件保存在文件系统中以解析条目。我在内存中做了如下。

我有一个函数将 InputStream 通过管道传输到 ByteArrayOutputStream（而不是 FileOutputStream，检索字节数组并将其初始化为 ByteArrayInputStream 并使用阅读器打印出每个条目

InputStream encFileIn = new FileInputStream("some_encrypted_file.asc");
byte[] decryptFile = util.decryptFile(encFileIn);
BufferedReader reader = new BufferedReader(new InputStreamReader(
                new ByteArrayInputStream(decryptFile), StandardCharsets.UTF_8));
reader.lines().forEach(f -> System.out.println(f));

这里有一些关于在我的解密函数中管道 InputStream 的代码 sn-ps，由于我正在处理 PGP 文件，因此可能不适用于您。所以你可能会尝试直接实例化一个 ByteArrayInputStream

public static void pipeAll(InputStream inStr, OutputStream outStr)
    throws IOException
{
    byte[] bs = new byte[BUFFER_SIZE];
    int numRead;
    while ((numRead = inStr.read(bs, 0, bs.length)) >= 0)
    {
        outStr.write(bs, 0, numRead);
    }
}

public static byte[] readAll(InputStream inStr)
    throws IOException
{
    ByteArrayOutputStream buf = new ByteArrayOutputStream();
    pipeAll(inStr, buf);
    return buf.toByteArray();
}