javascript - Node.js 可写流完成未在类中触发

Question

我正在尝试创建一个可在我的应用程序中的其他脚本中使用的 CSV 类。当 CSV 类实例化时，它会为用户指定的文件创建一个可写流。 write 和 destroy 方法似乎有效，但我不知道如何在数据完成写入文件后执行类上的“writeEnd”成员。

“writeEnd”成员变量是一个应该由用户覆盖的函数。例如，这是一个文件，我正在其中测试类功能，并将“writeEnd”重写为我自己选择的函数。

附注请参阅底部粗体字的问题!

const CSV = require('./shared/classes/csv');
const csv = new CSV(__dirname);

csv.writeEnd = () => {
  console.log('Finished!');
};

for (let i = 0; i < 1000000; i++) {
  csv.write('Hello World.');
}

我希望“完成!”记录到控制台，但该函数根本不触发。我希望我做错了一些别人可以很容易发现的错误。

以下是未修改的类文件，供您引用:

const { createWriteStream } = require('fs');
const { Readable } = require('stream');

/**
 * @class CSV
 */
module.exports = class CSV {
  constructor(path) {
    this.readStream = new Readable({ read() {} });
    this.writeStream = createWriteStream(`${path}/csv/data.csv`);
    this.readStream.pipe(this.writeStream);

    this.writeEnd = () => {};
  }

  /**
   * @method write
   * @param {any} data
   */
  write(data) {
    this.readStream.push(`${data}\n`);
  }

  /**
   * @method destroy
   */
  destroy() {
    this.readStream.destroy();
    this.writeStream.destroy();
  }
};

下面是我失败的尝试之一:

/**
 * @class CSV
 */
module.exports = class CSV {
  constructor(path) {
    this.readStream = new Readable({ read() {} });
    this.writeStream = createWriteStream(`${path}/csv/data.csv`);
    this.readStream.pipe(this.writeStream);

    // I'm wondering if this executes immediately because no writing is taking place
    // during instantiation
    this.writeStream.on('finish', this.writeEnd);
    this.writeEnd = () => {};
  }

  /**
   * @method write
   * @param {any} data
   */
  write(data) {
    this.readStream.push(`${data}\n`);
  }

  /**
   * @method destroy
   */
  destroy() {
    this.readStream.destroy();
    this.writeStream.destroy();
  }
};

我想知道我是否需要第一次实际监听 readStream 获取推送到它的数据，然后设置“完成”回调？

Answer 1

问题是自定义/重写的 writeEnd 方法永远不会被调用，因为事件发射器保留对原始处理程序的引用，即您在构造函数中设置的函数: this.writeEnd = () = > {};

最简单的方法是允许将回调函数传递给 CSV 类的构造函数，并将其用作 finish 处理程序。考虑这个简单的例子:

const EventEmitter = require("events").EventEmitter;

class CSV {

    constructor(customWriteEndCb) {
        this.writeEnd = () => {
            console.log("in original writeEnd");
        };
        this.writeEnd = customWriteEndCb || this.writeEnd;
        this.writeStream = new EventEmitter();
        this.writeStream.on('finished', this.writeEnd);
    }

    testMe() {
        this.writeStream.emit('finished');
    }

}

const customWriteEnd = () => {
    console.log("in custom writeEnd")
}

const csv = new CSV(customWriteEnd);
csv.testMe(); // will print "in custom writeEnd"