Node.js Stream(流)

Stream 是 Node.js 中非常重要的一个模块,应用广泛。

Stream 是一个抽象接口,Node 中有很多对象实现了这个接口。例如,对http 服务器发起请求的request 对象就是一个 Stream,还有stdout(标准输出)。

该抽象接口是可读、可写或是既可读又可写的,通过这些接口,我们可以和磁盘文件、套接字、HTTP请求来交互,实现数据从一个地方流动到另一个地方的功能。

Node.js,Stream 有四种流类型:

  • Readable - 可读操作。

  • Writable - 可写操作。

  • Duplex - 可读可写操作.

  • Transform - 操作被写入数据,然后读出结果。

所有的 Stream 对象都是 EventEmitter 的实例。常用的事件有:

  • data - 当有数据可读时触发。

  • end - 没有更多的数据可读时触发。

  • error - 在接收和写入过程中发生错误时触发。

  • finish - 所有数据已被写入到底层系统时触发。

本教程会为大家介绍常用的流操作。


从流中读取数据

创建 input.txt 文件,内容如下:

  1. nodejs官网地址:www.nodejs.org

创建 main.js 文件, 代码如下:

  1. var fs = require("fs");
  2. var data = '';
  3.  
  4. // 创建可读流
  5. var readerStream = fs.createReadStream('input.txt');
  6.  
  7. // 设置编码为 utf8。
  8. readerStream.setEncoding('UTF8');
  9.  
  10. // 处理流事件 --> data, end, and error
  11. readerStream.on('data', function(chunk) {
  12. data += chunk;
  13. });
  14.  
  15. readerStream.on('end',function(){
  16. console.log(data);
  17. });
  18.  
  19. readerStream.on('error', function(err){
  20. console.log(err.stack);
  21. });
  22.  
  23. console.log("程序执行完毕");

以上代码执行结果如下:

  1. 程序执行完毕
  2. nodejs官网地址:www.nodejs.org

写入流

创建 main.js 文件, 代码如下:

  1. var fs = require("fs");
  2. var data = 'nodejs官网地址:www.nodejs.org';
  3.  
  4. // 创建一个可以写入的流,写入到文件 output.txt 中
  5. var writerStream = fs.createWriteStream('output.txt');
  6.  
  7. // 使用 utf8 编码写入数据
  8. writerStream.write(data,'UTF8');
  9.  
  10. // 标记文件末尾
  11. writerStream.end();
  12.  
  13. // 处理流事件 --> data, end, and error
  14. writerStream.on('finish', function() {
  15. console.log("写入完成。");
  16. });
  17.  
  18. writerStream.on('error', function(err){
  19. console.log(err.stack);
  20. });
  21.  
  22. console.log("程序执行完毕");

以上程序会将 data 变量的数据写入到 output.txt 文件中。代码执行结果如下:

  1. $ node main.js
  2. 程序执行完毕
  3. 写入完成。

查看 output.txt 文件的内容:

  1. $ cat output.txt
  2. nodejs官网地址:www.nodejs.org

管道流

管道提供了一个输出流到输入流的机制。通常我们用于从一个流中获取数据并将数据传递到另外一个流中。

Node.js Stream(流) - 图1

如上面的图片所示,我们把文件比作装水的桶,而水就是文件里的内容,我们用一根管子(pipe)连接两个桶使得水从一个桶流入另一个桶,这样就慢慢的实现了大文件的复制过程。

以下实例我们通过读取一个文件内容并将内容写入到另外一个文件中。

设置 input.txt 文件内容如下:

  1. nodejs官网地址:www.nodejs.org
  2. 管道流操作实例

创建 main.js 文件, 代码如下:

  1. var fs = require("fs");
  2.  
  3. // 创建一个可读流
  4. var readerStream = fs.createReadStream('input.txt');
  5.  
  6. // 创建一个可写流
  7. var writerStream = fs.createWriteStream('output.txt');
  8.  
  9. // 管道读写操作
  10. // 读取 input.txt 文件内容,并将内容写入到 output.txt 文件中
  11. readerStream.pipe(writerStream);
  12.  
  13. console.log("程序执行完毕");

代码执行结果如下:

  1. $ node main.js
  2. 程序执行完毕

查看 output.txt 文件的内容:

  1. $ cat output.txt
  2. nodejs官网地址:www.nodejs.org
  3. 管道流操作实例

链式流

链式是通过连接输出流到另外一个流并创建多个对个流操作链的机制。链式流一般用于管道操作。

接下来我们就是用管道和链式来压缩和解压文件。

创建 compress.js 文件, 代码如下:

  1. var fs = require("fs");
  2. var zlib = require('zlib');
  3.  
  4. // 压缩 input.txt 文件为 input.txt.gz
  5. fs.createReadStream('input.txt')
  6. .pipe(zlib.createGzip())
  7. .pipe(fs.createWriteStream('input.txt.gz'));
  8.  
  9. console.log("文件压缩完成。");

代码执行结果如下:

  1. $ node compress.js
  2. 文件压缩完成。

执行完以上操作后,我们可以看到当前目录下生成了 input.txt 的压缩文件 input.txt.gz。

接下来,让我们来解压该文件,创建 decompress.js 文件,代码如下:

  1. var fs = require("fs");
  2. var zlib = require('zlib');
  3.  
  4. // 解压 input.txt.gz 文件为 input.txt
  5. fs.createReadStream('input.txt.gz')
  6. .pipe(zlib.createGunzip())
  7. .pipe(fs.createWriteStream('input.txt'));
  8.  
  9. console.log("文件解压完成。");

代码执行结果如下:

  1. $ node decompress.js
  2. 文件解压完成。