stream-size
Version:
Get the size of a stream and abort it if threshold is reached
23 lines • 1.16 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
var stream_1 = require("stream");
/**
* @param maximumSizeInBytesAllowed Optional. Set this parameter to the maximum amount of bytes before throwing an error and closing the stream. Leave unset to allow any amount of bytes and not apply any limit.
*/
function getSizeTransform(maximumSizeInBytesAllowed) {
var transformedStream = new stream_1.Transform({
transform: function (chunk, encoding, callback) {
var sizeInBytesAfterWritingChunk = this.sizeInBytes + Buffer.byteLength(chunk, encoding);
if (maximumSizeInBytesAllowed &&
sizeInBytesAfterWritingChunk > maximumSizeInBytesAllowed) {
return callback(new Error("Stream exceeded maximum size of " + maximumSizeInBytesAllowed + " bytes (" + sizeInBytesAfterWritingChunk + " bytes found)."));
}
this.sizeInBytes = sizeInBytesAfterWritingChunk;
callback(null, chunk);
},
});
transformedStream.sizeInBytes = 0;
return transformedStream;
}
exports.default = getSizeTransform;
//# sourceMappingURL=index.js.map