cartodb/CartoDB-SQL-API
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

managing gzip pipe with PassThrough

Browse Source
This commit is contained in:
Simon Martín 2018-06-21 15:54:42 +02:00
parent f5641a1873
commit 195a4866c3
1 changed files with 7 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
app/controllers/copy_controller.js
View File

@ -12,6 +12,7 @@ const errorHandlerFactory = require('../services/error_handler_factory');
const StreamCopy = require('../services/stream_copy');
const StreamCopyMetrics = require('../services/stream_copy_metrics');
const zlib = require('zlib');
const { PassThrough } = require('stream');
function CopyController(metadataBackend, userDatabaseService, userLimitsService, logger) {
this.metadataBackend = metadataBackend;
@ -121,13 +122,6 @@ function handleCopyFrom (logger) {
const err = new Error('Connection closed by client');
pgstream.emit('cancelQuery', err);
pgstream.emit('error', err);
})
.on('data', data => {
if (isGzip) {
metrics.addGzipSize(data.length);
} else {
metrics.addSize(data.length);
}
});
pgstream
@ -151,14 +145,12 @@ function handleCopyFrom (logger) {
});
});
if (isGzip) {
req
.pipe(zlib.createGunzip())
.on('data', data => metrics.addSize(data.length))
.pipe(pgstream);
} else {
req.pipe(pgstream);
}
const middleStream = isGzip ? zlib.createGunzip() : new PassThrough();
req
.on('data', data => isGzip ? metrics.addGzipSize(data.length) : undefined)
.pipe(middleStream)
.on('data', data => metrics.addSize(data.length))
.pipe(pgstream);
}
);
};