108 lines
2.9 KiB
JavaScript
108 lines
2.9 KiB
JavaScript
|
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);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}
|
||
|
|
};
|