CartoDB-SQL-API/batch/query_runner.js

'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) {
    this.userDatabaseMetadataService = userDatabaseMetadataService;
}

module.exports = QueryRunner;

QueryRunner.prototype.run = function (job_id, sql, user, callback) {
    this.userDatabaseMetadataService.getUserMetadata(user, function (err, userDatabaseMetadata) {
        if (err) {
            return callback(err);
        }

        var pg = new PSQL(userDatabaseMetadata, {}, { destroyOnError: true });

        pg.query('SET statement_timeout=' + BATCH_QUERY_TIMEOUT, function (err) {
            if(err) {
                return callback(err);
            }

            // mark query to allow to users cancel their queries
            sql = '/* ' + job_id + ' */ ' + sql;

            debug('Running query %s', 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);
                    }
                });
            });
        });
    });
};