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

Timeout is passed into query runner

Browse Source
This commit is contained in:
Raul Ochoa 2016-10-10 12:00:54 +02:00
parent 51d4ff0698
commit 5401a7edff
2 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
batch/job_runner.js
View File

@ -23,6 +23,10 @@ JobRunner.prototype.run = function (job_id, callback) {
}
var query = job.getNextQuery();
var timeout = 12 * 3600 * 1000;
if (Number.isFinite(global.settings.batch_query_timeout)) {
timeout = global.settings.batch_query_timeout;
}
try {
job.setStatus(jobStatus.RUNNING);
@ -37,15 +41,15 @@ JobRunner.prototype.run = function (job_id, callback) {
profiler.done('running');
self._run(job, query, profiler, callback);
self._run(job, query, timeout, profiler, callback);
});
});
};
JobRunner.prototype._run = function (job, query, profiler, callback) {
JobRunner.prototype._run = function (job, query, timeout, profiler, callback) {
var self = this;
self.queryRunner.run(job.data.job_id, query, job.data.user, function (err /*, result */) {
self.queryRunner.run(job.data.job_id, query, job.data.user, timeout, function (err /*, result */) {
if (err) {
if (!err.code) {
return callback(err);

7
batch/query_runner.js
View File

@ -1,7 +1,6 @@
'use strict';
var PSQL = require('cartodb-psql');
var BATCH_QUERY_TIMEOUT = global.settings.batch_query_timeout || 12 * 3600 * 1000; // 12 hours in millisecond
var debug = require('./util/debug')('query-runner');
function QueryRunner(userDatabaseMetadataService) {
@ -10,7 +9,7 @@ function QueryRunner(userDatabaseMetadataService) {
module.exports = QueryRunner;
QueryRunner.prototype.run = function (job_id, sql, user, callback) {
QueryRunner.prototype.run = function (job_id, sql, user, timeout, callback) {
this.userDatabaseMetadataService.getUserMetadata(user, function (err, userDatabaseMetadata) {
if (err) {
return callback(err);
@ -18,7 +17,7 @@ QueryRunner.prototype.run = function (job_id, sql, user, callback) {
var pg = new PSQL(userDatabaseMetadata, {}, { destroyOnError: true });
pg.query('SET statement_timeout=' + BATCH_QUERY_TIMEOUT, function (err) {
pg.query('SET statement_timeout=' + timeout, function (err) {
if(err) {
return callback(err);
}
@ -26,7 +25,7 @@ QueryRunner.prototype.run = function (job_id, sql, user, callback) {
// mark query to allow to users cancel their queries
sql = '/* ' + job_id + ' */ ' + sql;
debug('Running query %s', sql);
debug('Running query [timeout=%d] %s', timeout, sql);
pg.eventedQuery(sql, function (err, query) {
if (err) {
return callback(err);