扩展本机函数以在函数调用上运行代码

时间:2013-08-15 12:57:09

标签: javascript function profiling interceptor

  • 无论如何在JavaScript中原型化或扩展原生Function对象以在函数调用上运行另一个函数?
Function.prototype.called = function() {
  var start = new Date().getTime();
  var returnValue = this();
  this.end = new Date().getTime() - start;
  this.calls += 1;
  return returnValue;
};

当然这段代码不起作用,但这里的目标是基准函数调用执行时间。

我知道我可以在函数本身之外执行此操作并且不扩展本机Function对象(这通常是一个坏主意),但这实际上只是一个实验而不是一个严肃的解决方案。

我想对多个功能进行基准测试,因此这将是一个不那么繁琐的解决方案。

这也是我没有对callapply属性进行原型化的原因(这意味着我必须将每个函数调用test()重写为test.call() })。

提前谢谢!

1 个答案:

答案 0 :(得分:3)

无法有效拦截JavaScript中的所有方法调用。您始终可以覆盖Function.prototype.callFunction.prototype.apply,但在正常调用函数时不会调用这些函数,例如someFunction();

但是,如果要从特定对象中定位方法,则可以始终对这些对象进行深度遍历,并将每个函数包装到拦截器函数中。但是,请注意,在流程运行后添加的每个方法都必须重新运行拦截过程。

我为你创建了一个例子:

/**
 * AOP utility function developed for learning purposes.
 *
 * @param {Object} o The object to traverse for overriding functions.
 * @param {RegExp} rx A regular expression for matching members.
 * @param {String} pointcut 'before' or 'after'.
 * @param {Function} advice The function to run when the pointcut is met. This function will be passed an {Options} object as last argument.
 * @param {maxDepth} maxDepth The maximum depth for deep-traversal. Defaults to 0.
 *
 * Options object
 *     overrideReturn {Boolean} - True to override the return value of the original function with the return value of the advice. Defaults to false. Pointcuts: before, after
 *     cancel {Boolean} - True to avoid calling the original function. Default to false. Pointcuts: before
 *     overrideArgs {Boolean} - True to override the arguments that will be passed to the original function with the result of the advice. Defaults to false. Pointcuts: before
 *     result {*} - The return value of the original function. Pointcuts: after
 */

function inject(o, rx, pointcut, advice, maxDepth) {
    var pointcuts = {
        before: function (fn1, fn2) {
            return function () {

                var options = injectNewOptions(arguments, BeforeOptions),
                    fn2Result = fn2.apply(this, arguments),
                    fn1Result;

                if (options.cancel) {
                    return fn2Result;
                }

                fn1Result = fn1.apply(this, (options.overrideArgs ? fn2Result : arguments));

                return options.overrideReturn ? fn2Result : fn1Result;
            };
        },
        after: function (fn1, fn2) {
            return function () {
                var fn1Result = fn1.apply(this, arguments),
                    options = injectNewOptions(arguments, Options),
                    fn2Result;

                options.result = fn1Result;

                fn2Result = fn2.apply(this, arguments);

                return options.overrideReturn ? fn2Result : fn1Result;
            };
        }
    },
        Options = {
            overrideReturn: false
        },
        BeforeOptions = Object.create(Options, {
            cancel: {
                enumerable: true,
                writable: true,
                value: false
            },
            overrideArgs: {
                enumerable: true,
                writable: true,
                value: false
            }
        });

    function injectNewOptions(args, baseOptions) {
        var options = Object.create(baseOptions);

        Array.prototype.push.call(args, options);

        return options;
    }

    inject = function (o, rx, pointcut, advice, maxDepth, depth) {
        var k, f;

        maxDepth = maxDepth || 0;
        depth = 0 || depth;

        for (k in o) {
            if (typeof (f = o[k]) === 'function' && rx.test(k)) {
                o[k] = pointcuts[pointcut](f, advice, pointcut);
            } else if (typeof f === 'object' && maxDepth <= depth) {
                inject(f, rx, pointcut, advice, maxDepth, ++depth);
            }
        }
    };

    inject.apply(this, arguments);
}

现在我们可以使用inject之类的:

var o = {
    sum: function (a, b) {
        return a + b;
    },
    product: function (a, b) {
        return a * b;
    }
};

inject(o, /^sum$/, 'before', function () {
    var options = arguments[arguments.length - 1]; //get options object

    //override the arguments passed to the intercepted method
    options.overrideArgs = true; 

    return [2, 2];
});


inject(o, /^product$/, 'after', function () {
    var options = arguments[arguments.length - 1]; //get options object

    //override the arguments passed to the intercepted method
    options.overrideReturn = true; 

    return options.result + 3;
});

o.sum(1, 2); //4 because we have overriden the args with [2, 2]
o.product(2, 2); //7 because we added 3 to the result and overrided the return value

编辑:

值得一提的是@Bergi在评论中所说的内容。

  

如果您想要分析所有功能而不是仅选择一些功能   其中,您最好使用浏览器的开发人员工具   一些剧本。