70 lines
1.8 KiB
JavaScript
70 lines
1.8 KiB
JavaScript
// Copyright IBM Corp. 2015,2016. All Rights Reserved.
|
|
// Node module: loopback-connector-mysql
|
|
// This file is licensed under the MIT License.
|
|
// License text available at https://opensource.org/licenses/MIT
|
|
|
|
'use strict';
|
|
const debug = require('debug')('loopback:connector:mysql:transaction');
|
|
module.exports = mixinTransaction;
|
|
|
|
/*!
|
|
* @param {MySQL} MySQL connector class
|
|
* @param {Object} mysql mysql driver
|
|
*/
|
|
function mixinTransaction(MySQL, mysql) {
|
|
/**
|
|
* Begin a new transaction
|
|
* @param isolationLevel
|
|
* @param cb
|
|
*/
|
|
MySQL.prototype.beginTransaction = function(isolationLevel, cb) {
|
|
debug('Begin a transaction with isolation level: %s', isolationLevel);
|
|
this.client.getConnection(function(err, connection) {
|
|
if (err) return cb(err);
|
|
if (isolationLevel) {
|
|
connection.query(
|
|
'SET SESSION TRANSACTION ISOLATION LEVEL ' + isolationLevel,
|
|
function(err) {
|
|
if (err) return cb(err);
|
|
connection.beginTransaction(function(err) {
|
|
if (err) return cb(err);
|
|
return cb(null, connection);
|
|
});
|
|
},
|
|
);
|
|
} else {
|
|
connection.beginTransaction(function(err) {
|
|
if (err) return cb(err);
|
|
return cb(null, connection);
|
|
});
|
|
}
|
|
});
|
|
};
|
|
|
|
/**
|
|
*
|
|
* @param connection
|
|
* @param cb
|
|
*/
|
|
MySQL.prototype.commit = function(connection, cb) {
|
|
debug('Commit a transaction');
|
|
connection.commit(function(err) {
|
|
connection.release();
|
|
cb(err);
|
|
});
|
|
};
|
|
|
|
/**
|
|
*
|
|
* @param connection
|
|
* @param cb
|
|
*/
|
|
MySQL.prototype.rollback = function(connection, cb) {
|
|
debug('Rollback a transaction');
|
|
connection.rollback(function(err) {
|
|
connection.release();
|
|
cb(err);
|
|
});
|
|
};
|
|
}
|