Current File : /home/opti/workflow/node_modules/cls-bluebird/lib/shimMethod.js
'use strict';
/*
* cls-bluebird
* Function to shim an object method to retain CLS context
*/
// Modules
var shimmer = require('shimmer');
// Exports
/**
* Patch method to run callbacks in current CLS context.
*
* @param {Object} obj - Object on which to find method
* @param {string} methodName - method name
* @param {Array} args - Array of indexes of arguments which are callbacks
* (negative numbers count from end e.g. -1 is last argument, -2 is penultimate)
* @param {Object} ns - CLS namespace to bind callbacks to
* @returns {undefined}
*/
module.exports = function(obj, methodName, args, ns) {
// Skip non-existent methods
if (!obj[methodName]) return;
// Patch method
shimmer.wrap(obj, methodName, function(original) {
return function() {
for (var i = 0; i < args.length; i++) {
var argIndex = args[i];
if (argIndex < 0) argIndex += arguments.length;
var callback = arguments[argIndex];
if (typeof callback === 'function') arguments[argIndex] = ns.bind(callback);
}
return original.apply(this, arguments);
};
});
};
Mini Shell