21 lines
393 B
JavaScript
21 lines
393 B
JavaScript
|
const { Transform } = require('stream');
|
||
|
|
|
||
|
|
class Limiter extends Transform {
|
||
|
|
constructor(limit) {
|
||
|
|
super();
|
||
|
|
this.limit = limit;
|
||
|
|
this.length = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
_transform(chunk, encoding, callback) {
|
||
|
|
this.length += chunk.length;
|
||
|
|
this.push(chunk);
|
||
|
|
if (this.length > this.limit) {
|
||
|
|
return callback(new Error('limit'));
|
||
|
|
}
|
||
|
|
callback();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
module.exports = Limiter;
|