59 lines
1.7 KiB
JavaScript
59 lines
1.7 KiB
JavaScript
'use strict';
|
|
|
|
var PSQL = require('cartodb-psql');
|
|
var debug = require('./util/debug')('query-runner');
|
|
|
|
function QueryRunner(userDatabaseMetadataService) {
|
|
this.userDatabaseMetadataService = userDatabaseMetadataService;
|
|
}
|
|
|
|
module.exports = QueryRunner;
|
|
|
|
function hasDBParams (dbparams) {
|
|
return (dbparams.user && dbparams.host && dbparams.port && dbparams.dbname && dbparams.pass);
|
|
}
|
|
|
|
QueryRunner.prototype.run = function (job_id, sql, user, timeout, dbparams, callback) {
|
|
if (hasDBParams(dbparams)) {
|
|
return this._run(dbparams, job_id, sql, timeout, callback);
|
|
}
|
|
|
|
this.userDatabaseMetadataService.getUserMetadata(user, (err, userDBParams) => {
|
|
if (err) {
|
|
return callback(err);
|
|
}
|
|
|
|
this._run(userDBParams, job_id, sql, timeout, callback);
|
|
});
|
|
};
|
|
|
|
QueryRunner.prototype._run = function (dbparams, job_id, sql, timeout, callback) {
|
|
var pg = new PSQL(dbparams);
|
|
|
|
pg.query('SET statement_timeout=' + timeout, function (err) {
|
|
if(err) {
|
|
return callback(err);
|
|
}
|
|
|
|
// mark query to allow to users cancel their queries
|
|
sql = '/* ' + job_id + ' */ ' + sql;
|
|
|
|
debug('Running query [timeout=%d] %s', timeout, sql);
|
|
pg.eventedQuery(sql, function (err, query) {
|
|
if (err) {
|
|
return callback(err);
|
|
}
|
|
|
|
query.on('error', callback);
|
|
|
|
query.on('end', function (result) {
|
|
// only if result is present then query is done sucessfully otherwise an error has happened
|
|
// and it was handled by error listener
|
|
if (result) {
|
|
callback(null, result);
|
|
}
|
|
});
|
|
});
|
|
});
|
|
};
|