node.js使用stream模塊實現(xiàn)自定義流示例

更新時間：2020年02月13日 08:30:57 作者：懷素真

這篇文章主要介紹了node.js使用stream模塊實現(xiàn)自定義流,結合實例形式詳細分析了node.js基于stream模塊實現(xiàn)自定義的可讀流、可寫流、可讀寫流等相關操作技巧,需要的朋友可以參考下

本文實例講述了node.js使用stream模塊實現(xiàn)自定義流。分享給大家供大家參考，具體如下：

有些時候我們需要自定義一些流，來操作特殊對象，node.js中為我們提供了一些基本流類。

我們新創(chuàng)建的流類需要繼承四個基本流類之一(stream.Writeable，stream.Readable，stream.Duplex，stream.Transform)，并確保調用了父類構造函數(shù)。

一、實現(xiàn)自定義的可讀流

實現(xiàn)可讀流需繼承 stream.Readable，并實現(xiàn) readable._read() 方法。

下面的代碼我們實現(xiàn)了一個從數(shù)組中讀取數(shù)據(jù)的流

const {Readable} = require('stream');
//這里我們自定義了一個用來讀取數(shù)組的流
class ArrRead extends Readable {
  constructor(arr, opt) {
    //注意這里，需調用父類的構造函數(shù)
    super(opt);
    this.arr = arr;
    this.index = 0;
  }
  //實現(xiàn) _read() 方法
  _read(size) {
    //如果當前下標等于數(shù)組長度，說明數(shù)據(jù)已經(jīng)讀完
    if (this.index == this.arr.length) {
      this.push(null);
    } else {
      this.arr.slice(this.index, this.index + size).forEach((value) => {
        this.push(value.toString());
      });
      this.index += size;
    }
  }
}
let arr = new ArrRead([1, 2, 3, 4, 5, 6, 7, 8, 9, 0], {
  highWaterMark: 2
});
//這樣當我們監(jiān)聽 'data' 事件時，流會調用我們實現(xiàn)的 _read() 方法往緩沖區(qū)中讀取數(shù)據(jù)
//然后提供給消費者
arr.on('data', function (data) {
  console.log(data.toString());
});

二、實現(xiàn)自定義的可寫流

實現(xiàn)可寫流必須繼承 stream.Writeable ，并實現(xiàn) writeable._write() 方法。writable._writev() 方法是可選的。

const {Writable} = require('stream');
//這里我們自定義了一個用來寫入數(shù)組的流
class ArrWrite extends Writable {
  constructor(arr, opt) {
    super(opt);
    this.arr = arr;
  }
  //實現(xiàn) _write() 方法
  _write(chunk, encoding, callback) {
    this.arr.push(chunk.toString());
    callback();
  }
}
let data = [];
let arr = new ArrWrite(data, {
  highWaterMark: 3
});
arr.write('1');
arr.write('2');
arr.write('3');
console.log(data);

三、實現(xiàn)自定義的可讀可寫流

可讀可寫流必須繼承 stream.Duplex，并實現(xiàn) readable._read() 和 writable._write() 方法。

const {Duplex} = require('stream');
//這里我們自定義了一個用來寫讀可寫數(shù)組的流
class ArrReadWrite extends Duplex {
  constructor(arr, opt) {
    super(opt);
    this.arr = arr;
    this.index = 0;
  }
  //實現(xiàn) _write() 方法
  _write(chunk, encoding, callback) {
    this.arr.push(chunk.toString());
    callback();
  }
  //實現(xiàn) _read() 方法
  _read(size) {
    //如果當前下標等于數(shù)組長度，說明數(shù)據(jù)已經(jīng)讀完
    if (this.index == this.arr.length) {
      this.push(null);
    } else {
      this.arr.slice(this.index, this.index + size).forEach((value) => {
        this.push(value.toString());
      });
      this.index += size;
    }
  }
}
let data = [];
let arrWR = new ArrReadWrite(data, {
  highWaterMark: 3
});
//往流中寫入數(shù)據(jù)
arrWR.write('1');
arrWR.write('2');
arrWR.write('3');
console.log(data);
//往流中讀取數(shù)據(jù)
console.log(arrWR.read(2).toString());
console.log(arrWR.read(2).toString());

四、自定義的轉換流

轉換流必須繼承 stream.Transform，需實現(xiàn) transform._transform() 方法。

const {Transform} = require('stream');
//這里我們自定義了一個用來轉換數(shù)組的流
class Trans extends Transform {
  constructor(opt) {
    super(opt);
  }
  _transform(chunk, encoding, callback) {
    //將轉換后的數(shù)據(jù)輸出到可讀流
    this.push(chunk.toString().toUpperCase());
    //參數(shù)一是Error對象
    //參數(shù)二如果傳入，會被轉發(fā)到 readable.push()
    callback();
  }
}
let t = new Trans({
  highWaterMark: 3
});
t.on('data', function (data) {
  console.log(data.toString());
});
t.write('a');
t.write('b');
t.write('c');

轉換流就是將讀取到的數(shù)據(jù)做些計算然后輸出。轉換流既可以作為可讀流，又可以作為可寫流。

const {Transform} = require('stream');
//這里我們自定義了一個用來轉換數(shù)組的流
class Trans extends Transform {
  constructor(opt) {
    super(opt);
  }
  _transform(chunk, encoding, callback) {
    //將轉換后的數(shù)據(jù)輸出到可讀流
    this.push(chunk.toString().toUpperCase());
    //參數(shù)一是Error對象
    //參數(shù)二如果傳入，會被轉發(fā)到 readable.push()
    callback();
  }
}
let t = new Trans({
  highWaterMark: 3
});
t.on('data', function (data) {
  console.log('data', data.toString());
});
//stdin.pipe(t) 表示將我們的標準輸入寫入到我的轉換流 t 中，此時 t 是可寫流。
//pipe(process.stdout) 表示將轉換流 t 中的數(shù)據(jù)讀取到標準輸出中，此時 t 是可讀流。
process.stdin.pipe(t).pipe(process.stdout);

希望本文所述對大家node.js程序設計有所幫助。

您可能感興趣的文章: