Skip to content

public / sequelize

Go to a project

不要怂，就是干，撸起袖子干！

You need to sign in or sign up before continuing.
Commit 000b28c7 by overlookmotel
Options

CLS support with `cls-bluebird` module

1 parent acce21d4
Expand all
Inline Side-by-side
Showing with 63 additions and 101 deletions
changelog.md
...@@ -2,6 +2,7 @@ ...@@ -2,6 +2,7 @@
- [ADDED] include now supports string as an argument (on top of model/association), string will expand into an association matched literally from Model.associations - [ADDED] include now supports string as an argument (on top of model/association), string will expand into an association matched literally from Model.associations
- [FIXED] Accept dates as string while using `typeValidation` [#6453](https://github.com/sequelize/sequelize/issues/6453) - [FIXED] Accept dates as string while using `typeValidation` [#6453](https://github.com/sequelize/sequelize/issues/6453)
- [FIXED] - ORDER clause was not included in subquery if `order` option value was provided as plain string (not as an array value) - [FIXED] - ORDER clause was not included in subquery if `order` option value was provided as plain string (not as an array value)
- [FIXED] support for CLS with `cls-bluebird` module
# 4.0.0-1 # 4.0.0-1
- [CHANGED] Removed `modelManager` parameter from `Model.init()` [#6437](https://github.com/sequelize/sequelize/issues/6437) - [CHANGED] Removed `modelManager` parameter from `Model.init()` [#6437](https://github.com/sequelize/sequelize/issues/6437)
......
lib/model.js
...@@ -1952,8 +1952,8 @@ class Model { ...@@ -1952,8 +1952,8 @@ class Model {
options = _.assign({}, options); options = _.assign({}, options);
if (options.transaction === undefined && this.sequelize.constructor.cls) { if (options.transaction === undefined && this.sequelize.constructor._cls) {
const t = this.sequelize.constructor.cls.get('transaction'); const t = this.sequelize.constructor._cls.get('transaction');
if (t) { if (t) {
options.transaction = t; options.transaction = t;
} }
......
lib/promise.js
'use strict'; 'use strict';
const Promise = require('bluebird').getNewLibraryCopy(); const Promise = require('bluebird').getNewLibraryCopy();
const shimmer = require('shimmer');
// functionName: The Promise function that should be shimmed
// fnArgs: The arguments index that should be CLS enabled (typically all callbacks). Offset from last if negative
function shimCLS(object, functionName, fnArgs){
shimmer.wrap(object, functionName, fn => {
return function() {
if (Promise.Sequelize && Promise.Sequelize.cls) {
const ns = Promise.Sequelize.cls;
for(let x = 0; x < fnArgs.length; x++) {
const argIndex = fnArgs[x] < 0 ? arguments.length + fnArgs[x] : fnArgs[x];
if ( argIndex < arguments.length && typeof arguments[argIndex] === 'function' ) {
arguments[argIndex] = ns.bind( arguments[argIndex] );
}
}
}
return fn.apply(this, arguments);
};
});
}
// Core
shimCLS(Promise, 'join', [-1]);
shimCLS(Promise.prototype, 'then', [0, 1, 2]);
shimCLS(Promise.prototype, 'spread', [0, 1]);
shimCLS(Promise.prototype, 'catch', [-1]);
shimCLS(Promise.prototype, 'error', [0]);
shimCLS(Promise.prototype, 'finally', [0]);
// Collections
shimCLS(Promise, 'map', [1]);
shimCLS(Promise, 'mapSeries', [1]);
shimCLS(Promise, 'reduce', [1]);
shimCLS(Promise, 'filter', [1]);
shimCLS(Promise, 'each', [1]);
shimCLS(Promise.prototype, 'map', [0]);
shimCLS(Promise.prototype, 'mapSeries', [0]);
shimCLS(Promise.prototype, 'reduce', [0]);
shimCLS(Promise.prototype, 'filter', [0]);
shimCLS(Promise.prototype, 'each', [0]);
// Promisification
shimCLS(Promise.prototype, 'nodeify', [0]);
// Utility
shimCLS(Promise.prototype, 'tap', [0]);
// Error management configuration
shimCLS(Promise.prototype, 'done', [0, 1]);
module.exports = Promise; module.exports = Promise;
module.exports.Promise = Promise; module.exports.Promise = Promise;
......
lib/sequelize.js
...@@ -3,6 +3,7 @@ ...@@ -3,6 +3,7 @@
const url = require('url'); const url = require('url');
const Path = require('path'); const Path = require('path');
const retry = require('retry-as-promised'); const retry = require('retry-as-promised');
const clsBluebird = require('cls-bluebird');
const Utils = require('./utils'); const Utils = require('./utils');
const Model = require('./model'); const Model = require('./model');
const DataTypes = require('./data-types'); const DataTypes = require('./data-types');
...@@ -522,8 +523,8 @@ class Sequelize { ...@@ -522,8 +523,8 @@ class Sequelize {
searchPath: this.options.hasOwnProperty('searchPath') ? this.options.searchPath : 'DEFAULT' searchPath: this.options.hasOwnProperty('searchPath') ? this.options.searchPath : 'DEFAULT'
}); });
if (options.transaction === undefined && Sequelize.cls) { if (options.transaction === undefined && Sequelize._cls) {
options.transaction = Sequelize.cls.get('transaction'); options.transaction = Sequelize._cls.get('transaction');
} }
if (!options.type) { if (!options.type) {
...@@ -952,7 +953,7 @@ class Sequelize { ...@@ -952,7 +953,7 @@ class Sequelize {
* const cls = require('continuation-local-storage'); * const cls = require('continuation-local-storage');
* const ns = cls.createNamespace('....'); * const ns = cls.createNamespace('....');
* const Sequelize = require('sequelize'); * const Sequelize = require('sequelize');
* Sequelize.cls = ns; * Sequelize.useCls(ns);
* ``` * ```
* Note, that CLS is enabled for all sequelize instances, and all instances will share the same namespace * Note, that CLS is enabled for all sequelize instances, and all instances will share the same namespace
* *
...@@ -974,41 +975,61 @@ class Sequelize { ...@@ -974,41 +975,61 @@ class Sequelize {
// testhint argsConform.end // testhint argsConform.end
const transaction = new Transaction(this, options); const transaction = new Transaction(this, options);
const ns = Sequelize.cls; const ns = Sequelize._cls;
if (autoCallback) { if (!autoCallback) return transaction.prepareEnvironment().return(transaction);
let transactionResolver = (resolve, reject) => {
transaction.prepareEnvironment().then(() => { // autoCallback provided
if (ns) { const wrapper = ns ? function(fn) {
autoCallback = ns.bind(autoCallback); var promise;
ns.run(() => promise = fn());
return promise;
} : function(fn) { return fn(); };
return wrapper(() =>
transaction.prepareEnvironment().then(() =>
autoCallback(transaction)
).tap(() =>
transaction.commit()
).catch(err =>
// Rollback transaction if not already finished (commit, rollback, etc) and reject with original error
Promise.try(() => {
// Rollback (ignore any error in rollback)
if (!transaction.finished) return transaction.rollback().catch(function() {});
}).throw(err)
)
);
} }
const returnValue = autoCallback(transaction); /**
if (!returnValue || !returnValue.then) throw new Error('You need to return a promise chain/thenable to the sequelize.transaction() callback'); * Use CLS with Sequelize.
* CLS namespace provided is stored as `Sequelize._cls`
* and bluebird Promise is patched to use the namespace, using `cls-bluebird` module.
*
* @param {Object} ns CLS namespace
* @returns {Object} Sequelize constructor
*/
static useCls(ns) {
// check `ns` is valid CLS namespace
if (!ns || typeof ns !== 'object' || typeof ns.bind !== 'function' || typeof ns.run !== 'function') throw new Error('Must provide CLS namespace');
return returnValue.then(result => transaction.commit()).then(() => { // save namespace as `Sequelize._cls`
resolve(returnValue); this._cls = ns;
});
}).catch(err => {
// If the transaction has already finished (commit, rollback, etc), reject with the original error
if (transaction.finished) {
reject(err);
} else {
return transaction.rollback().finally(() => {
reject(err);
});
}
});
};
if (ns) { // patch bluebird to bind all promise callbacks to CLS namespace
transactionResolver = ns.bind(transactionResolver, ns.createContext()); clsBluebird(ns, Promise);
// return Sequelize for chaining
return this;
} }
return new Promise(transactionResolver); static get cls() {
} else { return this._cls;
return transaction.prepareEnvironment().return(transaction);
} }
static set cls(ns) {
Utils.deprecate('Sequelize.cls should not be set directly. Use Sequelize.useCls().');
this.useCls(ns);
} }
log() { log() {
...@@ -1314,7 +1335,6 @@ Sequelize.prototype.InstanceError = Sequelize.InstanceError = ...@@ -1314,7 +1335,6 @@ Sequelize.prototype.InstanceError = Sequelize.InstanceError =
Sequelize.prototype.EmptyResultError = Sequelize.EmptyResultError = Sequelize.prototype.EmptyResultError = Sequelize.EmptyResultError =
sequelizeErrors.EmptyResultError; sequelizeErrors.EmptyResultError;
// Allows the promise to access cls namespaces module.exports = Sequelize;
module.exports = Promise.Sequelize = Sequelize;
module.exports.Sequelize = Sequelize; module.exports.Sequelize = Sequelize;
module.exports.default = Sequelize; module.exports.default = Sequelize;
lib/transaction.js
...@@ -111,8 +111,8 @@ class Transaction { ...@@ -111,8 +111,8 @@ class Transaction {
throw setupErr; throw setupErr;
})) }))
.tap(() => { .tap(() => {
if (this.sequelize.constructor.cls) { if (this.sequelize.constructor._cls) {
this.sequelize.constructor.cls.set('transaction', this); this.sequelize.constructor._cls.set('transaction', this);
} }
return null; return null;
}); });
...@@ -155,7 +155,7 @@ class Transaction { ...@@ -155,7 +155,7 @@ class Transaction {
} }
_clearCls() { _clearCls() {
const cls = this.sequelize.constructor.cls; const cls = this.sequelize.constructor._cls;
if (cls) { if (cls) {
if (cls.get('transaction') === this) { if (cls.get('transaction') === this) {
......
package.json
...@@ -37,6 +37,7 @@ ...@@ -37,6 +37,7 @@
}, },
"dependencies": { "dependencies": {
"bluebird": "^3.4.6", "bluebird": "^3.4.6",
"cls-bluebird": "^2.0.1",
"debug": "^2.2.0", "debug": "^2.2.0",
"depd": "^1.1.0", "depd": "^1.1.0",
"dottie": "^1.0.0", "dottie": "^1.0.0",
...@@ -48,7 +49,6 @@ ...@@ -48,7 +49,6 @@
"node-uuid": "~1.4.4", "node-uuid": "~1.4.4",
"retry-as-promised": "^2.0.0", "retry-as-promised": "^2.0.0",
"semver": "^5.0.1", "semver": "^5.0.1",
"shimmer": "1.1.0",
"terraformer-wkt-parser": "^1.1.2", "terraformer-wkt-parser": "^1.1.2",
"toposort-class": "^1.0.1", "toposort-class": "^1.0.1",
"validator": "^5.6.0", "validator": "^5.6.0",
......
test/integration/transaction.test.js
...@@ -72,15 +72,6 @@ describe(Support.getTestDialectTeaser('Transaction'), function() { ...@@ -72,15 +72,6 @@ describe(Support.getTestDialectTeaser('Transaction'), function() {
}); });
}); });
it('errors when no promise chain is returned', function() {
var t;
return (expect(this.sequelize.transaction(function(transaction) {
t = transaction;
})).to.eventually.be.rejected).then(function() {
expect(t.finished).to.be.equal('rollback');
});
});
if (dialect === 'postgres' || dialect === 'mssql') { if (dialect === 'postgres' || dialect === 'mssql') {
it('do not rollback if already committed', function() { it('do not rollback if already committed', function() {
var SumSumSum = this.sequelize.define('transaction', { var SumSumSum = this.sequelize.define('transaction', {
......
test/unit/model/find-or-create.test.js
...@@ -15,11 +15,11 @@ describe(Support.getTestDialectTeaser('Model'), function() { ...@@ -15,11 +15,11 @@ describe(Support.getTestDialectTeaser('Model'), function() {
describe('method findOrCreate', function () { describe('method findOrCreate', function () {
before(function () { before(function () {
current.constructor.cls = cls.createNamespace('sequelize'); current.constructor.useCls(cls.createNamespace('sequelize'));
}); });
after(function () { after(function () {
delete current.constructor.cls; delete current.constructor._cls;
}); });
beforeEach(function () { beforeEach(function () {
...@@ -30,7 +30,7 @@ describe(Support.getTestDialectTeaser('Model'), function() { ...@@ -30,7 +30,7 @@ describe(Support.getTestDialectTeaser('Model'), function() {
this.transactionStub = stub(this.User.sequelize, 'transaction'); this.transactionStub = stub(this.User.sequelize, 'transaction');
this.transactionStub.returns(new Promise(function () {})); this.transactionStub.returns(new Promise(function () {}));
this.clsStub = stub(current.constructor.cls, 'get'); this.clsStub = stub(current.constructor._cls, 'get');
this.clsStub.returns({ id: 123 }); this.clsStub.returns({ id: 123 });
}); });
......
Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!