20 lines
684 B
JavaScript
20 lines
684 B
JavaScript
|
import { Readable } from "stream";
|
||
|
|
export class ReadFromBuffers extends Readable {
|
||
|
|
constructor(options) {
|
||
|
|
super(options);
|
||
|
|
this.numBuffersRead = 0;
|
||
|
|
this.buffersToRead = options.buffers;
|
||
|
|
this.errorAfter = typeof options.errorAfter === "number" ? options.errorAfter : -1;
|
||
|
|
}
|
||
|
|
_read(size) {
|
||
|
|
if (this.errorAfter !== -1 && this.errorAfter === this.numBuffersRead) {
|
||
|
|
this.emit("error", new Error("Mock Error"));
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
if (this.numBuffersRead >= this.buffersToRead.length) {
|
||
|
|
return this.push(null);
|
||
|
|
}
|
||
|
|
return this.push(this.buffersToRead[this.numBuffersRead++]);
|
||
|
|
}
|
||
|
|
}
|