Node.js Stream

Stream 是一個(gè)抽象接口，Node 中有很多對(duì)象實(shí)現(xiàn)了這個(gè)接口。例如，對(duì)http 服務(wù)器發(fā)起請(qǐng)求的request 對(duì)象就是一個(gè) Stream，還有stdout（標(biāo)準(zhǔn)輸出）。

Node.js，Stream 有四種流類型：

• Readable - 可讀操作。

• Writable - 可寫(xiě)操作。

• Duplex - 可讀可寫(xiě)操作.

• Transform - 操作被寫(xiě)入數(shù)據(jù)，然后讀出結(jié)果。

所有的 Stream 對(duì)象都是 EventEmitter 的實(shí)例。常用的事件有：

• data - 當(dāng)有數(shù)據(jù)可讀時(shí)觸發(fā)。

• end - 沒(méi)有更多的數(shù)據(jù)可讀時(shí)觸發(fā)。

• error - 在接收和寫(xiě)入過(guò)程中發(fā)生錯(cuò)誤時(shí)觸發(fā)。

• finish - 所有數(shù)據(jù)已被寫(xiě)入到底層系統(tǒng)時(shí)觸發(fā)。

本教程會(huì)為大家介紹常用的流操作。

從流中讀取數(shù)據(jù)

創(chuàng)建 input.txt 文件，內(nèi)容如下：

小白教程官網(wǎng)地址：

創(chuàng)建 main.js 文件, 代碼如下：

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

// 創(chuàng)建可讀流
var readerStream = fs.createReadStream('input.txt');

// 設(shè)置編碼為 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("程序執(zhí)行完畢");

以上代碼執(zhí)行結(jié)果如下：

程序執(zhí)行完畢
小白教程官網(wǎng)地址：

寫(xiě)入流

創(chuàng)建 main.js 文件, 代碼如下：

var fs = require("fs");
var data = '小白教程官網(wǎng)地址：';

// 創(chuàng)建一個(gè)可以寫(xiě)入的流，寫(xiě)入到文件 output.txt 中
var writerStream = fs.createWriteStream('output.txt');

// 使用 utf8 編碼寫(xiě)入數(shù)據(jù)
writerStream.write(data,'UTF8');

// 標(biāo)記文件末尾
writerStream.end();

// 處理流事件 --> data, end, and error
writerStream.on('finish', function() {
    console.log("寫(xiě)入完成。");
});

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

console.log("程序執(zhí)行完畢");

以上程序會(huì)將 data 變量的數(shù)據(jù)寫(xiě)入到 output.txt 文件中。代碼執(zhí)行結(jié)果如下：

$ node main.js 
程序執(zhí)行完畢
寫(xiě)入完成。

查看 output.txt 文件的內(nèi)容：

$ cat output.txt 
小白教程官網(wǎng)地址：

管道流

如上面的圖片所示，我們把文件比作裝水的桶，而水就是文件里的內(nèi)容，我們用一根管子(pipe)連接兩個(gè)桶使得水從一個(gè)桶流入另一個(gè)桶，這樣就慢慢的實(shí)現(xiàn)了大文件的復(fù)制過(guò)程。

以下實(shí)例我們通過(guò)讀取一個(gè)文件內(nèi)容并將內(nèi)容寫(xiě)入到另外一個(gè)文件中。

設(shè)置 input.txt 文件內(nèi)容如下：

小白教程官網(wǎng)地址：
管道流操作實(shí)例

創(chuàng)建 main.js 文件, 代碼如下：

var fs = require("fs");

// 創(chuàng)建一個(gè)可讀流
var readerStream = fs.createReadStream('input.txt');

// 創(chuàng)建一個(gè)可寫(xiě)流
var writerStream = fs.createWriteStream('output.txt');

// 管道讀寫(xiě)操作
// 讀取 input.txt 文件內(nèi)容，并將內(nèi)容寫(xiě)入到 output.txt 文件中
readerStream.pipe(writerStream);

console.log("程序執(zhí)行完畢");

代碼執(zhí)行結(jié)果如下：

$ node main.js 
程序執(zhí)行完畢

查看 output.txt 文件的內(nèi)容：

$ cat output.txt 
小白教程官網(wǎng)地址：
管道流操作實(shí)例

鏈?zhǔn)搅?/h2>

鏈?zhǔn)绞峭ㄟ^(guò)連接輸出流到另外一個(gè)流并創(chuàng)建多個(gè)流操作鏈的機(jī)制。鏈?zhǔn)搅饕话阌糜诠艿啦僮鳌?/p>

接下來(lái)我們就是用管道和鏈?zhǔn)絹?lái)壓縮和解壓文件。

創(chuàng)建 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("文件壓縮完成。");

代碼執(zhí)行結(jié)果如下：

$ node compress.js 
文件壓縮完成。

執(zhí)行完以上操作后，我們可以看到當(dāng)前目錄下生成了 input.txt 的壓縮文件 input.txt.gz。

接下來(lái)，讓我們來(lái)解壓該文件，創(chuàng)建 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("文件解壓完成。");

代碼執(zhí)行結(jié)果如下：

$ node decompress.js 
文件解壓完成。