如何使用 NIO 将 InputStream 写入 File？

Question

我正在使用以下方式将InputStream 写入File：

private void writeToFile(InputStream stream) throws IOException {
    String filePath = "C:\\Test.jpg";
    FileChannel outChannel = new FileOutputStream(filePath).getChannel();       
    ReadableByteChannel inChannel = Channels.newChannel(stream);
    ByteBuffer buffer = ByteBuffer.allocate(1024);
    
    while(true) {
        if(inChannel.read(buffer) == -1) {
            break;
        }
        
        buffer.flip();
        outChannel.write(buffer);
        buffer.clear();
    }
    
    inChannel.close();
    outChannel.close();
}

我想知道这是否是使用 NIO 的正确方法。我看过一个方法FileChannel.transferFrom，它需要三个参数：

1. ReadableByteChannel src
2. 多头
3. 长计数

就我而言，我只有src，我没有position 和count，有什么方法可以使用这种方法来创建文件吗？

对于 Image 还有没有更好的方法来仅从 InputStream 和 NIO 创建图像？

任何信息都会对我非常有用。这里有类似的问题，在 SO 中，但我找不到适合我的情况的任何特定解决方案。

Answer 1

我会使用 Files.copy

Files.copy(is, Paths.get(filePath));

至于你的版本

1. ByteBuffer.allocateDirect 更快 - Java 将尽最大努力直接在其上执行本机 I/O 操作。

2. 关闭是不可靠的，如果第一次失败，第二次将永远不会执行。请改用 try-with-resources，Channel 也是 AutoCloseable。

Answer 2

不，这是不正确的。您冒着丢失数据的风险。规范的 NIO 复制循环如下：

while (in.read(buffer) >= 0 || buffer.position() > 0)
{
  buffer.flip();
  out.write(buffer);
  buffer.compact();
}

注意更改的循环条件，它负责在 EOS 上刷新输出，以及使用 compact() 而不是 clear(), 来处理短写入的可能性。

同样，规范的transferTo()/transferFrom() 循环如下：

long offset = 0;
long quantum = 1024*1024; // or however much you want to transfer at a time
long count;
while ((count = out.transferFrom(in, offset, quantum)) > 0)
{
    offset += count;
}

它必须在循环中调用，因为它不能保证传输整个量程。