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

unify streams

Browse Source
This commit is contained in:
Simon Martín 2018-06-21 14:53:07 +02:00
parent f4651cadae
commit 3ede1ea9f1
1 changed files with 7 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
app/services/stream_copy.js
View File

@ -12,8 +12,7 @@ module.exports = class StreamCopy {
this.sql = sql;
this.connectionClosedByClient = false;
this.copyToStream = null;
this.copyFromStream = null;
this.stream = null;
}
static get ACTION_TO() {
@ -30,8 +29,8 @@ module.exports = class StreamCopy {
return cb(err);
}
this.copyToStream = copyTo(this.sql);
const pgstream = client.query(this.copyToStream);
this.stream = copyTo(this.sql);
const pgstream = client.query(this.stream);
pgstream
.on('end', () => done())
@ -55,8 +54,8 @@ module.exports = class StreamCopy {
return cb(err);
}
this.copyFromStream = copyFrom(this.sql);
const pgstream = client.query(this.copyFromStream);
this.stream = copyFrom(this.sql);
const pgstream = client.query(this.stream);
pgstream
.on('end', () => done())
@ -69,13 +68,7 @@ module.exports = class StreamCopy {
});
}
getRowCount(action = ACTION_TO) {
if (action === ACTION_TO && this.copyToStream) {
return this.copyToStream.rowCount;
}
if (action === ACTION_FROM && this.copyFromStream) {
return this.copyFromStream.rowCount;
}
getRowCount() {
return this.stream.rowCount;
}
};