34 lines
944 B
JavaScript
34 lines
944 B
JavaScript
'use strict';
|
|
|
|
var PSQL = require('cartodb-psql');
|
|
const dbParamsFromReqParams = require('../utils/database-params');
|
|
|
|
function PgQueryRunner(pgConnection) {
|
|
this.pgConnection = pgConnection;
|
|
}
|
|
|
|
module.exports = PgQueryRunner;
|
|
|
|
/**
|
|
* Runs `query` with `username`'s PostgreSQL role, callback receives error and rows array.
|
|
*
|
|
* @param {String} username
|
|
* @param {String} query
|
|
* @param {Function} callback function({Error}, {Array}) second argument is guaranteed to be an array
|
|
*/
|
|
PgQueryRunner.prototype.run = function(username, query, callback) {
|
|
|
|
this.pgConnection.getDatabaseParams(username, (err, databaseParams) => {
|
|
if (err) {
|
|
return callback(err);
|
|
}
|
|
|
|
const psql = new PSQL(dbParamsFromReqParams(databaseParams));
|
|
|
|
psql.query(query, function (err, resultSet) {
|
|
resultSet = resultSet || {};
|
|
return callback(err, resultSet.rows || []);
|
|
});
|
|
});
|
|
};
|