javascript代理类实现

 

       前几天写单元测试，遇到了需要模拟被调函数的功能，还需要统计被调函数的调用次数，当时随便写了一个代理类，周末有新的想法，优化了一把，这里把自己的实现写下来。

 

       代理类参考了spring的aop的通知，代理被调函数时可以注册四个函数：

              before：在被调函数调用前运行

              after：被调函数调用后运行

              throwsException：被调用函数抛出异常时执行

              around：如果存在，将替代调用函数，里面可以决定是否执行被调用函数

 

 

具体代码如下：

 

 

---

 

 

 

/** * @author hz */ window.hzrui = window.hzrui||{}; window.hzrui.proxyUtil = window.hzrui.proxyUtil || {}; (function(context){ context._guid = 1; context._Fns = []; context._callCount = []; /** * * @param {Object} fn * @param {Object} opt * @example * var proxyFn = proxy(fn, { * argumentLen：3, * before:function(caller, fn, args){}, * around:function(invocation){return invocation.process();}, * after:function(caller, fn, args, result){}, * throwsException:function(caller, fn, args, e){} * }); */ context.proxy = function(fn, opt){ var _guid = context._guid++; opt = opt || {}; //代理函数 var _newFn = function(){ //调用次数加1 context._callCount[_guid]++; //判断参数个数 if(opt.argumentLen && opt.argumentLen != arguments.length){ return; } //这里可以判断参数类型 if(opt.before){ opt.before(this, fn, arguments); } // var _result; try{ if(opt.around){ _result = opt.around({ caller:this, arguments:arguments, fn:fn, process:function(){ return fn.apply(this.caller, this.arguments); } }); }else{ _result = fn.apply(this, arguments); } }catch(e){ if(opt.throwsException){ _result = opt.throwsException(this, fn, arguments, e); }else{ throw e; } } if(opt.after){ opt.after(this, fn, arguments, _result); } return _result; }; _newFn.guid = _guid; context._Fns[_guid] = fn; context._callCount[_guid] = 0; return _newFn; }; })(window.hzrui.proxyUtil);  

 

 

 

---

 

 

 

 

 

在代理类里面，原始函数和调用次数都保存了下来，可以用fn的guid查找得到，可以加多两个函数来还原和取得调用次数。

 

 

写了段测试代码，也作为demo吧，用了console，在ie下的话要修改输出方式

 

 

---

 

var addValue = function(a, b){ return a+b; }; var newFn = hzrui.proxyUtil.proxy(addValue, { argumentLen :2, before:function(caller, fn , args){ console.log("caller: " + caller); console.log("fn: " + fn); console.log("arguments: " + args.length); console.log("arguments: " + args[0] + "," + args[1]); console.log("您在使用加法计算 " + args[0] + "与" + args[1] + "的和"); }, around:function(invocation){ console.log("这里是【around】 "); console.log("caller: " + invocation.caller); console.log("fn: " + invocation.fn); console.log("arguments: " + invocation.arguments.length); var _result = invocation.process(); console.log("结果本来应该是" + _result + ",修改为0"); return 0; }, after:function(caller, fn, args, result){ console.log("这里是【after】 "); console.log("caller: " + caller); console.log("fn: " + fn); console.log("arguments: " + args.length); console.log("result: " + result); }, throwsException:function(caller, fn, args, e){ console.log("这里是【throwsException】 "); console.log("caller: " + caller); console.log("fn: " + fn); console.log("arguments: " + args.length); console.log("e: " + e); return "异常"; } }); console.log("运行函数的结果为："　+ newFn(3, 4));