Question

我想编写一个方法，从单个InputStream中读取ZIP中的多个XML文件。

该方法将打开一个ZipInputStream，并在每个xml文件上获取相应的InputStream，并将其提供给我的XML解析器。以下是该方法的框架：

private void readZip(InputStream is) throws IOException {

    ZipInputStream zis = new ZipInputStream(is);
    ZipEntry entry = zis.getNextEntry();

    while (entry != null) {

        if (entry.getName().endsWith(".xml")) {

            // READ THE STREAM
        }
        entry = zis.getNextEntry();
    }
}

有问题的部分是“// READ THE STREAM”。我有一个工作解决方案，它包括创建一个ByteArrayInputStream，并用它提供我的解析器。但它使用缓冲区，对于大文件，我得到一个OutOfMemoryError。这是代码，如果有人仍然感兴趣：

int count;
byte buffer[] = new byte[2048];
ByteArrayOutputStream out = new ByteArrayOutputStream();
while ((count = zis.read(buffer)) != -1) { out.write(buffer, 0, count); }       
InputStream is = new ByteArrayInputStream(out.toByteArray());

理想的解决方案是使用原始ZipInputStream提供解析器。它应该有效，因为如果我只用扫描仪打印条目内容就可以了：

Scanner sc = new Scanner(zis);
while (sc.hasNextLine())
{
    System.out.println(sc.nextLine());
}

但是...我正在使用的解析器（jdom2，但我也尝试过javax.xml.parsers.DocumentBuilderFactory）在解析数据后关闭流：/。所以我无法获得下一个条目并继续。

所以最后问题是：

有人知道没有关闭其流的DOM解析器吗？
还有另一种方法可以从ZipEntry获得InputStream吗？

感谢。

Answer 1

您可以包装ZipInputStream并拦截对close()的调用。

Answer 2

感谢halfbit，我最终得到了自己的ZipInputStream类，它覆盖了close方法：

import java.io.IOException;
import java.io.InputStream;
import java.util.zip.ZipInputStream;

public class CustomZipInputStream extends ZipInputStream {

    private boolean _canBeClosed = false;

    public CustomZipInputStream(InputStream is) {
        super(is);
    }

    @Override
    public void close() throws IOException {

        if(_canBeClosed) super.close();
    }

    public void allowToBeClosed() { _canBeClosed = true; }
}

Answer 3

Tim解决方案的一个小改进：必须在close（）之前调用allowToBeClosed（）的问题是它在处理异常棘手时正确关闭ZipInputStream并且会破坏Java 7的try-with-resources语句。

我建议按如下方式创建一个包装类：

public class UncloseableInputStream extends InputStream {
  private final InputStream input;

  public UncloseableInputStream(InputStream input) {
    this.input = input;
  }

  @Override
  public void close() throws IOException {} // do not close the wrapped stream

  @Override
  public int read() throws IOException {
    return input.read();
  }

  // delegate all other InputStream methods as with read above
}

然后可以安全地使用如下：

try (ZipInputStream zipIn = new ZipInputStream(...))
{
  DocumentBuilder db = DocumentBuilderFactory.newInstance().newDocumentBuilder();
  ZipEntry entry;
  while (null != (entry = zipIn.getNextEntry()))
  {
    if ("file.xml".equals(entry.getName())
    {
      Document doc = db.parse(new UncloseableInputStream(zipIn));
    }
  }
}

Java从ZipInputStream条目创建InputStream

3 个答案: