orange-orm/src/pglite/wrapCommand.js

Object Relational Mapper

var log = require('../table/log');
var replaceParamChar = require('../pg/replaceParamChar');

function wrapCommand(_context, connection) {
	var runOriginalQuery = connection.query;
	return runQuery;

	function runQuery(query, onCompleted) {
		var params = query.parameters;
		var sql = replaceParamChar(query, params);
		log.emitQuery({ sql, parameters: params });

		runOriginalQuery
			.call(connection, sql, params)
			.then(
				(result) => onInnerCompleted(null, result),
				(e) => onInnerCompleted(e)
			);

		function onInnerCompleted(err, result) {
			if (err) return onCompleted(err);

			if (Array.isArray(result)) result = result[result.length - 1];

			onCompleted(null, { rowsAffected: result.affectedRows });
		}
	}
}

module.exports = wrapCommand;