[node] Node.js 中Stream流_nodejs stream

什么是 Stream

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

Node.js，Stream 有四种流类型：

• Readable - 可读操作
• Writable - 可写操作
• Duplex - 可读可写操作
• Transform - 操作被写入数据，然后读出结果

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

• data - 当有数据可读时触发
• end - 没有更多的数据可读时触发
• error - 在接收和写入过程中发生错误时触发
• finish - 所有数据已被写入到底层系统时触发

流操作

从流中读取数据

input.txt

Test read by stream
1

main.js

var fs = require("fs");
var data = '';

// 创建可读流
var readerStream = fs.createReadStream('input.txt');
// 设置编码为 utf8
readerStream.setEncoding('UTF8');
// 处理流事件 --> data, end, and error
readerStream.on('data', function(chunk) {
   data += chunk;
});
readerStream.on('end',function(){
   console.log(data);
});
readerStream.on('error', function(err){
   console.log(err.stack);
});

console.log("程序执行完毕");
//输出：

//程序执行完毕
//Test read by stream
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

写入流

main.js

var fs = require("fs");
var data = 'Test for write ';

// 创建一个可以写入的流，写入到文件 output.txt 中
var writerStream = fs.createWriteStream('output.txt');

// 使用 utf8 编码写入数据
writerStream.write(data,'UTF8');

// 标记文件末尾
writerStream.end();

// 处理流事件 --> finish、error
writerStream.on('finish', function() {
    console.log("写入完成。");
});

writerStream.on('error', function(err){
   console.log(err.stack);
});

console.log("程序执行完毕");
//程序执行完毕
//写入完成。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

生成的output.txt,在执行命令的路径下，不在执行的文件夹所在路径下，内容为：

Test for write 
1

管道流

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

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

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

设置 input.txt 文件如下：

管道流操作实例
1

创建main.js 文件，并执行：

var fs = require("fs");
// 创建一个可读流
var readerStream = fs.createReadStream('input.txt');
// 创建一个可以写入的流，写入到文件 output.txt 中
var writerStream = fs.createWriteStream('output.txt');

// 管道读写操作
// 读取 input.txt 文件内容，写入到 output.txt 文件中
readerStream.pipe(writerStream);

console.log("程序执行完毕");
//程序执行完毕
1
2
3
4
5
6
7
8
9
10
11
12

查看output.txt文件：

管道流操作实例
1

链式流

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

下面是用管道和链式来压缩和解压文件的实例。

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

var fs = require("fs");
var zlib = require('zlib');

// 压缩 input.txt 文件为 input.txt.gz
fs.createReadStream('input.txt')
  .pipe(zlib.createGzip())
  .pipe(fs.createWriteStream('input.txt.gz'));
  
console.log("文件压缩完成。");
//执行后输出：
//文件压缩完成。
1
2
3
4
5
6
7
8
9
10
11

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

接下来，解压该文件，创建 decompress.js 文件：

var fs = require("fs");
var zlib = require('zlib');

// 解压 input.txt.gz 文件为 input.txt
fs.createReadStream('input.txt.gz')
  .pipe(zlib.createGunzip())
  .pipe(fs.createWriteStream('input.txt'));
  
console.log("文件解压完成。");
//执行后输出：
//文件解压完成。
1
2
3
4
5
6
7
8
9
10
11