creating StreamCopy class

This commit is contained in:
Simon Martín 2018-06-08 11:09:51 +02:00
parent 2bcea0484a
commit 732c229e63

View File

@ -5,9 +5,13 @@ const copyFrom = require('pg-copy-streams').from;
const StreamCopyMetrics = require('./stream_copy_metrics');
const { Client } = require('pg');
module.exports = {
to (res, sql, userDbParams, user, logger, cb) {
let metrics = new StreamCopyMetrics(logger, 'copyto', sql, user);
module.exports = class StreamCopy {
constructor(logger) {
this.logger = logger;
}
to(res, sql, userDbParams, user, cb) {
let metrics = new StreamCopyMetrics(this.logger, 'copyto', sql, user);
const pg = new PSQL(userDbParams);
pg.connect(function (err, client, done) {
@ -63,10 +67,10 @@ module.exports = {
})
.pipe(res);
});
},
}
from (req, sql, userDbParams, user, gzip, logger, cb) {
let metrics = new StreamCopyMetrics(logger, 'copyfrom', sql, user, gzip);
from(req, sql, userDbParams, user, gzip, cb) {
let metrics = new StreamCopyMetrics(this.logger, 'copyfrom', sql, user, gzip);
const pg = new PSQL(userDbParams);
pg.connect(function (err, client, done) {