节点管道到标准输出 - 我如何判断是否已耗尽？

Question

确定是否需要等待process.stdout 上的drain 事件的标准建议是检查它在写入时是否返回false。

我应该如何检查我是否已将另一个流通过管道传输到它？在实际写入所有输出之前，该流似乎可以发出finish。我可以这样做吗？

upstreamOfStdout.on('finish', function(){
  if(!process.stdout.write('')) {
    process.stdout.on('drain', function() { done("I'm done"); });
  }
  else {
    done("I'm done"); 
  }
});
upstreamOfStdout.pipe(process.stdout);

我更喜欢不依赖于任何流内部的答案。刚刚考虑到流符合节点流接口，那么规范的方法是什么？

编辑：

更大的上下文是一个包装器：

new Promise(function(resolve, reject){
  stream.on(<some-event>, resolve);
  ... (perhaps something else here?)
});

其中stream 可以是process.stdout 或其他东西，其中有另一个直通流通过管道输送到其中。

每当调用resolve 时，我的程序都会退出——我假设Promise 代码会保持程序处于活动状态，直到所有承诺都得到解决。

我已经多次遇到这种情况，并且一直使用 hack 来解决问题（例如，process.stdout 中有几个私人成员很有用。）但我真的很想一劳永逸地解决这个问题（或知道这是一个错误，因此我至少可以跟踪问题并在解决问题时修复我的 hack）：我如何判断另一个流下游的流何时完成处理其输入？

Answer 1

不要直接写入process.stdout，而是创建一个自定义可写（如下所示），作为副作用写入标准输出。

const { Writable } = require('stream');
function writeStdoutAndFinish(){
  return new Writable({
    write(chunk, encoding, callback) {
      process.stdout.write(chunk,callback);
    },
  });
};

writeStdoutAndFinish() 的结果将发出一个finish 事件。

async function main(){
  ...
  await new Promise((resolve)=>{
    someReadableStream.pipe(writeStdoutAndFinish()).on('finish',()=>{
      console.log('finish received');
      resolve();
    }) 
  });
  ...
}

在实践中，我不认为上述方法的行为与

async function main(){
  ...
  await new Promise((resolve)=>{
    (someReadableStream.on('end',()=>{
      console.log('end received');
      resolve();
    })).pipe(process.stdout)
  });
  ...
}

Answer 2

首先，据我从documentation 看到的，该流永远不会发出finish 事件，因此您不太可能依赖它。

此外，从上面提到的文档中，drain 事件似乎用于在.write 方法返回false 时通知用户stream 何时准备好接受更多数据。在任何情况下，您都可以推断这意味着所有其他数据都已写入。从write 方法的文档中，我们确实推断出false 值（又名请停止推送数据）不是强制性的，您可以随意忽略它，但后续数据可能会填充在内存中，以便使用它长大了。

因此，基于我对唯一文档的假设，我想您可以依靠 drain 事件来了解所有数据何时已得到妥善处理或可能被清除。

也就是说，在我看来，还没有一种明确的方法可以明确知道所有数据何时已有效发送到控制台。

最后，您可以监听管道stream 的end 事件，以了解它何时已被完全消耗，无论它是否已写入控制台或数据仍缓冲在控制台流中。 当然，你也可以随意忽略这个问题，因为一个完全消耗的流应该由node.js很好地处理，因此被丢弃，一旦你将它传送到第二个流，你就不必再处理它了。