'use strict'; // Oracle Runner // ------ module.exports = function(client) { var _ = require('lodash'); var inherits = require('inherits'); var Promise = require('../../promise'); var Runner = require('../../runner'); var helpers = require('../../helpers'); var OracleQueryStream = require('./oracle-query-stream'); // Inherit from the `Runner` constructor's prototype, // so we can add the correct `then` method. function Runner_Oracle() { this.client = client; Runner.apply(this, arguments); } inherits(Runner_Oracle, Runner); Runner_Oracle.prototype._stream = Promise.method(function (obj, stream, options) { var self = this; obj.sql = this.client.positionBindings(obj.sql); if (this.isDebugging()) this.debug(obj); return new Promise(function (resolver, rejecter) { stream.on('error', rejecter); stream.on('end', resolver); var queryStream = new OracleQueryStream(self.connection, obj.sql, obj.bindings, options); queryStream.pipe(stream); }); }); // Runs the query on the specified connection, providing the bindings // and any other necessary prep work. Runner_Oracle.prototype._query = Promise.method(function(obj) { var connection = this.connection; // convert ? params into positional bindings (:1) obj.sql = this.client.positionBindings(obj.sql); // convert boolean parameters into 0 or 1 obj.bindings = this.client.preprocessBindings(obj.bindings) || []; if (!obj.sql) throw new Error('The query is empty'); if (this.isDebugging()) this.debug(obj); return new Promise(function(resolver, rejecter) { connection.execute(obj.sql, obj.bindings, function(err, response) { if (err) return rejecter(err); if (obj.returning) { var rowIds = obj.outParams.map(function (v, i) { return response['returnParam' + (i ? i : '')]; }); return connection.execute(obj.returningSql, rowIds, function (err, subres) { if (err) return rejecter(err); obj.response = subres; resolver(obj); }); } else { obj.response = response; resolver(obj); } }); }); }); // Process the response as returned from the query. Runner_Oracle.prototype.processResponse = function(obj) { var response = obj.response; var method = obj.method; if (obj.output) return obj.output.call(this, response); switch (method) { case 'select': case 'pluck': case 'first': response = helpers.skim(response); if (obj.method === 'pluck') response = _.pluck(response, obj.pluck); return obj.method === 'first' ? response[0] : response; case 'insert': case 'del': case 'update': case 'counter': if (obj.returning) { if (obj.returning.length > 1 || obj.returning[0] === '*') { return response; } // return an array with values if only one returning value was specified return _.flatten(_.map(response, _.values)); } return response.updateCount; default: return response; } }; function finishOracleTransaction(connection, finishFunc) { return new Promise(function (resolver, rejecter) { return finishFunc.bind(connection)(function (err, result) { if (err) { return rejecter(err); } // reset AutoCommit back to default to allow recycling in pool connection.setAutoCommit(true); resolver(result); }); }); } // disable autocommit to allow correct behavior (default is true) Runner_Oracle.prototype.beginTransaction = function() { return this.connection.setAutoCommit(false); }; Runner_Oracle.prototype.commitTransaction = function() { return finishOracleTransaction(this.connection, this.connection.commit); }; Runner_Oracle.prototype.rollbackTransaction = function() { return finishOracleTransaction(this.connection, this.connection.rollback); }; // Assign the newly extended `Runner` constructor to the client object. client.Runner = Runner_Oracle; };