首頁 > web前端 > js教程 > 了解NodeJS中的可讀流

了解NodeJS中的可讀流

青灯夜游
發布: 2020-11-20 17:45:57
轉載
9463 人瀏覽過

了解NodeJS中的可讀流

相關推薦:《node js教學

什麼是可讀流

可讀流是生產資料用來供程式消費的流。我們常見的資料生產方式有讀取磁碟檔案、讀取網路請求內容等,看一下前面介紹什麼是流用的例子:

const rs = fs.createReadStream(filePath);
登入後複製

rs 就是一個可讀流,其生產資料的方式是讀取磁碟的文件,我們常見的控制台process.stdin 也是一個可讀流:

process.stdin.pipe(process.stdout);
登入後複製

透過簡單的一句話可以把控制台的輸入打印出來,process.stdin 生產資料的方式是讀取使用者在控制台的輸入。

回頭再看一下我們對可讀流的定義:可讀流是生產資料用來供程式消費的流。

自訂可讀流

除了系統提供給我們的fs.CreateReadStream 我們也常使用gulp 或vinyl-fs 提供的src方法

gulp.src(['*.js', 'dist/**/*.scss'])
登入後複製

如果我們想自己以某種特定的方式生產數據,交給程式消費,那麼改該如何開始呢?

簡單兩步驟即可

  1. 繼承sream 模組的Readable 類別
  2. 重寫_read 方法,調用this.push 將生產的資料放入待讀取佇列

Readable 類別已經把可讀流要做的大部分工作完成,我們只需要繼承它,然後把生產資料的方式寫在_read 方法裡就可以實作一個自訂的可讀流。

如果我們想實現一個每100 毫秒生產一個隨機數的流(沒什麼用處)

const Readable = require('stream').Readable;

class RandomNumberStream extends Readable {
    constructor(max) {
        super()
    }

    _read() {
        const ctx = this;
        setTimeout(() => {
            const randomNumber = parseInt(Math.random() * 10000);

            // 只能 push 字符串或 Buffer,为了方便显示打一个回车
            ctx.push(`${randomNumber}\n`);
        }, 100);
    }
}

module.exports = RandomNumberStream;
登入後複製

類別繼承部分程式碼很簡單,主要看一下_read 方法的實現,有幾個值得注意的地方

  1. Readable 類別中預設有_read 方法的實現,不過什麼都沒有做,我們做的是覆蓋重寫
  2. _read 方法有一個參數size,用來向read 方法指定應該讀取多少資料返回,不過只是一個參考資料,很多實作忽略此參數,我們這裡也忽略了,後面會詳細提到
  3. 透過this.push 向緩衝區推送數據,緩衝區概念後面會提到,暫時理解為擠到了水管中可消費了
  4. push 的內容只能是字串或Buffer,不能是數字
  5. push 方法有第二個參數encoding,用於第一個參數是字串時指定encoding

執行一下看看效果

const RandomNumberStream = require('./RandomNumberStream');

const rns = new RandomNumberStream();

rns.pipe(process.stdout);
登入後複製

這樣可以看到數字源源不斷的顯示到了控制台上,我們實現了一個產生隨機數的可讀流,還有幾個小問題待解決

如何停下來

我們每隔100 毫秒向緩衝區推送一個數字,那麼就像讀取本地文件總有讀完的時候,如何停下來標識資料讀取完畢?

向緩衝區 push 一個 null 就可以。我們修改一下程式碼,允許消費者定義需要多少個隨機數字:

const Readable = require('stream').Readable;

class RandomNumberStream extends Readable {
    constructor(max) {
        super()
        this.max = max;
    }

    _read() {
        const ctx = this;

        setTimeout(() => {
            if (ctx.max) {
                const randomNumber = parseInt(Math.random() * 10000);

                // 只能 push 字符串或 Buffer,为了方便显示打一个回车
                ctx.push(`${randomNumber}\n`);
                ctx.max -= 1;
            } else {
                ctx.push(null);
            }
        }, 100);
    }
}

module.exports = RandomNumberStream;
登入後複製

我們使用了一個max 的標識,允許消費者指定需要的字元數,在實例化的時候指定即可

const RandomNumberStream = require('./RandomNumberStream');

const rns = new RandomNumberStream(5);

rns.pipe(process.stdout);
登入後複製

這樣可以看到控制台只印了5 個字元

為什麼是setTimeout 而不是setInterval

細心的同學可能注意到,我們每隔100 毫秒生產一個隨機數字並不是調用的setInterval,而是使用的setTimeout,為什麼只是延時了一下並沒有重複生產,結果卻是正確的呢?

這就需要了解流的兩種工作方式

  1. 流動模式:資料由底層系統讀出,並儘可能快地提供給應用程式
  2. #暫停模式:必須顯示地呼叫read() 方法來讀取若干資料塊

流在預設狀態下是處於暫停模式的,也就是需要程式明確的呼叫read() 方法,但我們的例子中並沒有調用就可以得到數據,因為我們的流透過pipe() 方法切換成了流動模式,這樣我們的_read() 方法會自動被反覆調用,直到數據讀取完畢,所以我們每次_read() 方法裡面只需要讀取一次資料即可。

流動模式和暫停模式切換

流從預設的暫停模式切換到流動模式可以使用以下幾種方式:

  1. 透過新增data 事件監聽器來啟動資料監聽
  2. 呼叫resume() 方法啟動資料流
  3. 呼叫pipe() 方法將資料轉接到另一個可寫流

從流動模式切換為暫停模式又兩種方法:

  1. 在流没有 pipe() 时,调用 pause() 方法可以将流暂停
  2. pipe() 时,需要移除所有 data 事件的监听,再调用 unpipe() 方法

data 事件

使用了 pipe() 方法后数据就从可读流进入了可写流,但对我们好像是个黑盒,数据究竟是怎么流向的呢?我们看到切换流动模式和暂停模式的时候有两个重要的名词

  1. 流动模式对应的 data 事件
  2. 暂停模式对应的 read() 方法

这两个机制是我们能够驱动数据流动的原因,先来看一下流动模式 data 事件,一旦我们监听了可读流的 data 时、事件,流就进入了流动模式,我们可以改写一下上面调用流的代码

const RandomNumberStream = require('./RandomNumberStream');

const rns = new RandomNumberStream(5);

rns.on('data', chunk => {
  console.log(chunk);
});
登入後複製

这样我们可以看到控制台打印出了类似下面的结果

<Buffer 39 35 37 0a>
<Buffer 31 30 35 37 0a>
<Buffer 38 35 31 30 0a>
<Buffer 33 30 35 35 0a>
<Buffer 34 36 34 32 0a>
登入後複製

当可读流生产出可供消费的数据后就会触发 data 事件,data 事件监听器绑定后,数据会被尽可能地传递。data 事件的监听器可以在第一个参数收到可读流传递过来的 Buffer 数据,这也就是我们打印的 chunk,如果想显示为数字,可以调用 Buffer 的 toString() 方法。

当数据处理完成后还会触发一个 end 事件,应为流的处理不是同步调用,所以如果我们希望完事后做一些事情就需要监听这个事件,我们在代码最后追加一句:

rns.on('end', () => {
  console.log('done');
});
登入後複製

这样可以在数据接收完了显示 'done'

当然数据处理过程中出现了错误会触发 error 事件,我们同样可以监听,做异常处理:

rns.on('error', (err) => {
  console.log(err);
});
登入後複製

read(size)

流在暂停模式下需要程序显式调用 read() 方法才能得到数据。read() 方法会从内部缓冲区中拉取并返回若干数据,当没有更多可用数据时,会返回null。

使用 read() 方法读取数据时,如果传入了 size 参数,那么它会返回指定字节的数据;当指定的size字节不可用时,则返回null。如果没有指定size参数,那么会返回内部缓冲区中的所有数据。

现在有一个矛盾了,在流动模式下流生产出了数据,然后触发 data 事件通知给程序,这样很方便。在暂停模式下需要程序去读取,那么就有一种可能是读取的时候还没生产好,如果我们才用轮询的方式未免效率有些低。

NodeJS 为我们提供了一个 readable 的事件,事件在可读流准备好数据的时候触发,也就是先监听这个事件,收到通知又数据了我们再去读取就好了:

const rns = new RandomNumberStream(5);

rns.on('readable', () => {
  let chunk;
  while((chunk = rns.read()) !== null){
    console.log(chunk);
  }
});
登入後複製

这样我们同样可以读取到数据,值得注意的一点是并不是每次调用 read() 方法都可以返回数据,前面提到了如果可用的数据没有达到 size 那么返回 null,所以我们在程序中加了个判断。

数据会不会漏掉

开始使用流动模式的时候我经常会担心一个问题,上面代码中可读流在创建好的时候就生产数据了,那么会不会在我们绑定 readable 事件之前就生产了某些数据,触发了 readable 事件,我们还没有绑定,这样不是极端情况下会造成开头数据的丢失嘛

可事实并不会,按照 NodeJS event loop 我们创建流和调用事件监听在一个事件队列里面,儿生产数据由于涉及到异步操作,已经处于了下一个事件队列,我们监听事件再慢也会比数据生产块,数据不会丢失。

看到这里,大家其实对 data事件、readable事件触发时机, read() 方法每次读多少数据,什么时候返回 null 还有又一定的疑问,因为到现在为止我们接触到的仍然是一个黑盒,后面我们介绍了可写流后会在 back pressure 机制部分对这些内部细节结合源码详细讲解,且听下回分解吧。

更多编程相关知识,请访问:编程入门!!

以上是了解NodeJS中的可讀流的詳細內容。更多資訊請關注PHP中文網其他相關文章!

相關標籤:
來源:cnblogs.com
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板