NodeJS实现自定义流的方法

前言

常见的自定义流有四种，Readable（可读流）、Writable（可写流）、Duplex（双工流）和 Transform（转换流），常见的自定义流应用有 HTTP 请求、响应， crypto 加密，进程 stdin 通信等等。

stream 模块介绍

在 NodeJS 中要想实现自定义流，需要依赖模块 stream ，直接引入，不需下载，所有种类的流都是继承这个模块内部提供的对应不同种类的类来实现的。

实现一个自定义可读流 Readable

1、创建自定义可读流的类 MyRead

实现自定义可读流需创建一个类为 MyRead ，并继承 stream 中的 Readable 类，重写 _read 方法，这是所有自定义流的固定套路。

let { Readable } = require("stream");
// 创建自定义可读流的类
class MyRead extends Readable {
 constructor() {
 super();
 this.index = 0;
 }
 // 重写自定义的可读流的 _read 方法
 _read() {
 this.index++;
 this.push(this.index + "");
 if (this.index === 3) {
  this.push(null);
 }
 }
}

我们自己写的 _read 方法会先查找并执行，在读取时使用 push 方法将数据读取出来，直到 push 的值为 null 才会停止，否则会认为没有读取完成，会继续调用 _read 。

2、验证自定义可读流

let myRead = new MyRead();
myRead.on("data", data => {
 console.log(data);
});
myRead.on("end", function() {
 console.log("读取完成");
});
// <Buffer 31>
// <Buffer 32>
// <Buffer 33>
// 读取完成

实现一个自定义可写流 Writable

1、创建自定义可写流的类 MyWrite

创建一个类名为 MyWrite ，并继承 stream 中的 Writable 类，重写 _write 方法。

let { Writable } = require("stream");
// 创建自定义可写流的类
class MyWrite extends Writable {
 // 重写自定义的可写流的 _write 方法
 _write(chunk, encoding, callback)) {
 callback(); // 将缓存区写入文件
 }
}

写入内容时默认第一次写入直接写入文件，后面的写入都写入缓存区，如果不调用 callback 只能默认第一次写入文件，调用 callback 会将缓存区清空并写入文件。

2、验证自定义可写流

let myWrite = new MyWrite();
myWrite.write("hello", "utf8", () => {
 console.log("hello ok");
});
myWrite.write("world", "utf8", () => {
 console.log("world ok");
});
// hello ok
// world ok