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

A cleaner approach to the cancel command

Browse Source
This commit is contained in:
Rafa de la Torre 2018-06-04 16:34:37 +02:00
parent 4022fb2967
commit 014ea8142b
1 changed files with 14 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
app/services/stream_copy.js
View File

@ -16,6 +16,9 @@ module.exports = {
} }
let responseEnded = false; let responseEnded = false;
let connectionClosedByClient = false;
const copyToStream = copyTo(sql);
const pgstream = client.query(copyToStream);
res res
.on('error', err => { .on('error', err => {
@ -25,28 +28,26 @@ module.exports = {
}) })
.on('close', () => { .on('close', () => {
if (!responseEnded) { if (!responseEnded) {
// Cancel the running COPY TO query. connectionClosedByClient = true;
// Cancel the running COPY TO query
// See https://www.postgresql.org/docs/9.5/static/protocol-flow.html#PROTOCOL-COPY // See https://www.postgresql.org/docs/9.5/static/protocol-flow.html#PROTOCOL-COPY
const runningClient = client; const runningClient = client;
const cancelingClient = new Client(runningClient.connectionParameters); const cancelingClient = new Client(runningClient.connectionParameters);
const connection = cancelingClient.connection; cancelingClient.cancel(runningClient, pgstream);
connection.on('connect', () => { pgstream.unpipe(res);
connection.cancel(runningClient.processID, runningClient.secretKey); done(new Error('Connection closed by client'));
done(); return cb(err);
});
connection.connect(runningClient.port, runningClient.host);
return cb(new Error('Connection closed by client'));
} }
}) })
.on('end', () => responseEnded = true); .on('end', () => responseEnded = true);
const copyToStream = copyTo(sql);
const pgstream = client.query(copyToStream);
pgstream pgstream
.on('error', err => { .on('error', err => {
if (!connectionClosedByClient) {
pgstream.unpipe(res); pgstream.unpipe(res);
done(); done(new Error('Connection closed by client'));
return cb(err); return cb(err);
}
}) })
.on('data', data => metrics.addSize(data.length)) .on('data', data => metrics.addSize(data.length))
.on('end', () => { .on('end', () => {