cls-mongoose index.js addition for Models

index.js

var wrapLastParamInCLSBind = function(clsns, container, functionName) {
    shimmer.wrap(container, functionName, function (original) {
        return function() {
            var lastArg = arguments[arguments.length - 1];
            if (lastArg && ('function' == typeof lastArg)) {
                var clsBoundLastArg = clsns.bind(lastArg);
                arguments[arguments.length - 1] = clsBoundLastArg;
            }
            return original.apply(this, arguments);
        };
    });
};



wrapLastParamInCLSBind(ns, mongoose.Model, 'find');
wrapLastParamInCLSBind(ns, mongoose.Model, 'update');
wrapLastParamInCLSBind(ns, mongoose.Model, 'distinct');