Node.js：如何在写入流'finish'事件上写入()

Question

我正在使用 Node.js 流逐行浏览文本文件，进行一些转换并输出到 SVG 文件。

我正在尝试在处理完成后写入最后一条数据（</svg>），但是当写入流发出finish事件时，尝试write()将抛出Error: write after end。

有没有一种优雅的方法可以解决这个问题？

注意：输入文件很大（大约 1GB），因此由于其 I/O 和内存管理，无法绕过 pipe() 方法。

var fs = require('fs');
var split2 = require('split2');
var through2 = require('through2');
var read_stream = fs.createReadStream('input.txt');
var write_stream = fs.createWriteStream('output.svg');
write_stream.write('<svg>');
write_stream.on('finish', function() {
  this.write('</svg>'); // doesn't work
});
read_stream
  .pipe(split2())
  .pipe(through2.obj(function(line, encoding, next) {
     this.push(line);
     next();
  }))
  .pipe(write_stream);

解决 方案

谢谢Jordan&pNre帮助我解决这个问题。

解决方案 1（通用）

使用 end:false 选项pipe()写入流，然后手动end()流。

var fs = require('fs');
var split2 = require('split2');
var through2 = require('through2');
var read_stream = fs.createReadStream('input.txt');
var write_stream = fs.createWriteStream('output.svg');
write_stream.write('<svg>');
read_stream
  .pipe(split2())
  .pipe(through2.obj(function(line, encoding, next) {
     this.push(line);
     next();
  }))
  .pipe(write_stream, { end: false });
read_stream.on('end', function() {
  write_stream.end('</svg>');
});

解决方案 2（特定于through/through2转换流）

through2具有可用于写入最终数据的刷新功能。

var fs = require('fs');
var split2 = require('split2');
var through2 = require('through2');
var read_stream = fs.createReadStream('input.txt');
var write_stream = fs.createWriteStream('output.svg');
write_stream.write('<svg>');
read_stream
  .pipe(split2())
  .pipe(through2.obj(function(line, encoding, next) {
    this.push(line);
    next();
  }, function(flush) {
    this.push('</svg>');
    flush();
  }))
  .pipe(write_stream);

Answer 1

管道似乎在完成后关闭了流。

http://nodejs.org/api/stream.html 的文档指出：

默认情况下，当源流发出 end 时，将在目标上调用 end（），因此该目标不再可写。传递 { end： false } 作为选项以保持目标流打开。

这使作家保持开放，以便可以在最后写下"再见"。

reader.pipe(writer, { end: false });
reader.on('end', function() {
  writer.end('Goodbyen');
});

Answer 2

您是否考虑过创建一个新流来附加</svg>标签？ through可以帮助您：

var fs = require('fs');
var split2 = require('split2');
var through = require('through');
var read_stream = fs.createReadStream('input.txt');
var write_stream = fs.createWriteStream('output.svg');
write_stream.write('<svg>');
var tag = through(function write(data) {
    this.queue(data);
}, function end() {
    this.queue('</svg>');
});
read_stream.pipe(split2()).pipe(some_transform).pipe(tag).pipe(write_stream);

Answer 3

似乎有一个未记录的事件叫做 'prefinish' .不过我没有用过。

Answer 4

最近遇到了这个问题，并找到了一个更优雅的解决方案。本机转换流上有一个（良好）记录的_flush方法。

https://nodejs.org/api/stream.html#stream_transform_flush_callback

解决方案如下所示：

const fs = require('fs')
const split2 = require('split2')
const { Transform } = require('stream')
const input = fs.createReadStream('input.txt')
const output = fs.createWriteStream('output.svg')
class SVGWrapper extends Transform {
    constructor(){ this.push('<svg>') }
    _flush(done){ this.push('</svg>') }
    _transform(line, enc, next){
        this.push(line)
        next()
    }
}
input
    .pipe(split2())
    .pipe(new SVGWrapper)
    .pipe(output)