61 lines
1.4 KiB
JavaScript
61 lines
1.4 KiB
JavaScript
|
// PostgreSQL Native Driver (pg-native)
|
||
|
// -------
|
||
|
const Client_PG = require('../postgres');
|
||
|
|
||
|
class Client_PgNative extends Client_PG {
|
||
|
constructor(...args) {
|
||
|
super(...args);
|
||
|
this.driverName = 'pgnative';
|
||
|
this.canCancelQuery = true;
|
||
|
}
|
||
|
|
||
|
_driver() {
|
||
|
return require('pg').native;
|
||
|
}
|
||
|
|
||
|
_stream(connection, obj, stream, options) {
|
||
|
if (!obj.sql) throw new Error('The query is empty');
|
||
|
|
||
|
const client = this;
|
||
|
return new Promise((resolver, rejecter) => {
|
||
|
stream.on('error', rejecter);
|
||
|
stream.on('end', resolver);
|
||
|
|
||
|
return client
|
||
|
._query(connection, obj)
|
||
|
.then((obj) => obj.response)
|
||
|
.then(({ rows }) => rows.forEach((row) => stream.write(row)))
|
||
|
.catch(function (err) {
|
||
|
stream.emit('error', err);
|
||
|
})
|
||
|
.then(function () {
|
||
|
stream.end();
|
||
|
});
|
||
|
});
|
||
|
}
|
||
|
|
||
|
async cancelQuery(connectionToKill) {
|
||
|
try {
|
||
|
return await this._wrappedCancelQueryCall(null, connectionToKill);
|
||
|
} catch (err) {
|
||
|
this.logger.warn(`Connection Error: ${err}`);
|
||
|
throw err;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
_wrappedCancelQueryCall(emptyConnection, connectionToKill) {
|
||
|
return new Promise(function (resolve, reject) {
|
||
|
connectionToKill.native.cancel(function (err) {
|
||
|
if (err) {
|
||
|
reject(err);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
resolve(true);
|
||
|
});
|
||
|
});
|
||
|
}
|
||
|
}
|
||
|
|
||
|
module.exports = Client_PgNative;
|