const Bluebird = require('bluebird'); const Transaction = require('../../transaction'); const { isUndefined } = require('lodash'); const debug = require('debug')('knex:tx'); module.exports = class Transaction_MSSQL extends Transaction { begin(conn) { debug('%s: begin', this.txid); return conn.tx_.begin().then(this._resolver, this._rejecter); } savepoint(conn) { debug('%s: savepoint at', this.txid); return Bluebird.resolve().then(() => this.query(conn, `SAVE TRANSACTION ${this.txid}`) ); } commit(conn, value) { this._completed = true; debug('%s: commit', this.txid); return conn.tx_.commit().then(() => this._resolver(value), this._rejecter); } release(conn, value) { return this._resolver(value); } rollback(conn, error) { this._completed = true; debug('%s: rolling back', this.txid); return conn.tx_.rollback().then( () => { let err = error; if (isUndefined(error)) { if (this.doNotRejectOnRollback) { this._resolver(); return; } err = new Error(`Transaction rejected with non-error: ${error}`); } this._rejecter(err); }, (err) => { if (error) err.originalError = error; return this._rejecter(err); } ); } rollbackTo(conn, error) { debug('%s: rolling backTo', this.txid); return Bluebird.resolve() .then(() => this.query(conn, `ROLLBACK TRANSACTION ${this.txid}`, 2, error) ) .then(() => this._rejecter(error)); } // Acquire a connection and create a disposer - either using the one passed // via config or getting one off the client. The disposer will be called once // the original promise is marked completed. acquireConnection(config, cb) { const configConnection = config && config.connection; return new Bluebird((resolve, reject) => { try { resolve( (this.outerTx ? this.outerTx.conn : null) || configConnection || this.client.acquireConnection() ); } catch (e) { reject(e); } }) .then((conn) => { conn.__knexTxId = this.txid; if (!this.outerTx) { this.conn = conn; conn.tx_ = conn.transaction(); } return conn; }) .then(async (conn) => { try { return await cb(conn); } finally { if (!this.outerTx) { if (conn.tx_) { if (!this._completed) { debug('%s: unreleased transaction', this.txid); conn.tx_.rollback(); } conn.tx_ = null; } this.conn = null; if (!configConnection) { debug('%s: releasing connection', this.txid); this.client.releaseConnection(conn); } else { debug('%s: not releasing external connection', this.txid); } } } }); } };