node中Stream分為Readable(可讀流)、Writable(可寫流)、Duplex(可讀寫流)、Transform(讀寫過程中可以修改和變換數據的 Duplex 流)。
為了實現可寫流,我們需要使用流模塊中的Writable構造函數。 我們只需給Writable構造函數傳遞一些選項并創建一個對象。唯一需要的選項是write函數,該函數揭露數據塊要往哪里寫。
chunk通常是一個buffer,除非我們配置不同的流。
encoding是在特定情況下需要的參數,通常我們可以忽略它。
callback是在完成處理數據塊后需要調用的函數。這是寫數據成功與否的標志。若要發出故障信號,請用錯誤對象調用回調函數
下面代碼用ES6對可寫流進行代碼簡要的實現:
let fs = require('fs');
let EventEmitter = require('events');
class WriteStream extends EventEmitter {
constructor(path, options) {
super();
let self = this;
Object.assign(self, options); //還需設置默認值
self.path = path; //文件路經
self.isWriting = false;
self.Buffer = []; //緩沖區
self.len = null;
self.pos = self.start; //初始化寫入位置
self.fd = null;
self.open();
}
open() {//首先打開文件
let self = this;
fs.open(self.path, self.flags, self.mode, (err, fd) => {
self.fd = fd;
if (err) return self.destroy(err);
self.emit('open');
});
}
destroy(err) {
fs.close(this.fd, () => {
this.emit('error', err);
});
}
write(chunk, encoding, cb) {
let self = this
, ret = null;
encoding = encoding?encoding:self.encoding; //優先使用write傳入的編碼方式
chunk = Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk, encoding);
self.len += chunk.length;
ret = self.highWaterMark > self.len; //判斷當前最新的緩沖區是否已達到最高水位線
if (self.isWriting) { //說明正在調用底層方法真正寫入文件,先寫入Buffer
self.Buffer.push({
chunk
, cb
});
} else {
self.isWriting = true;
self._write(chunk, cb, () => self.clearBuffer());
}
return ret;
}
_write(chunk, cb, clear) {
let self = this;
if (!self.fd) return self.once('open', () => {
self._write(chunk, cb, clear)
});
fs.write(self.fd, chunk, 0, chunk.length, self.pos, (err, bytesWritten) => {
if (err) {
if (self.autoClose) {
self.destroy();
self.emit('error', err);
}
}
self.len -= bytesWritten;
self.pos += bytesWritten;
cb && cb();
clear && clear();
});
}
clearBuffer() {
let self = this
, data = null;
data = self.Buffer.shift();
if (data) {
self._write(data.chunk, data.cb, () => self.clearBuffer());
} else { //此時說明緩沖區已無數據
self.isWriting = false;
self.emit('drain');
}
}
}