最近在看python的基础语法,发现里面有一个用于扩展函数功能的玩意,叫装饰器。在js的ES6语法中并不存在这个玩意,所以,我就尝试写了一个方法去实现类似的效果:
function Decorator(fnName, execute, obj=window) {
let oldFn = obj[fnName]
obj[fnName] = function() {
return execute(oldFn.bind(obj))
}
}
方法的第一个参数是要装饰的函数名,第二个参数是要自定义的功能,值是一个参数为被装饰函数的函数,第三个参数是被装饰函数所属的对象,我这里默认为window对象。
举个简单的例子,之前我在项目中,已经使用了一个fn1的函数,现在想为fn1增加功能,使其在运行功能之前先打印出时间,在运行之后也打印出时间,之前的代码是这样:
function fn1() {
console.log('running')
}
fn1()
现在我在fn1定义的地方,加入一个装饰器:
function fn1() {
console.log('running')
}
Decorator('fn1', (fn) => {
console.log('运行开始时间:', new Date())
fn()
console.log('运行结束时间:', new Date())
})
fn1()
调用的函数依然是fn1(),原来定义的函数没有修改,调用的方式也没有改,运行的结果变成:
我觉得这种方式可以运用在方法是由别人定义好的,然后在不动别人的方法的前提下去扩展方法的功能。