装饰器模式

什么是装饰器模式

  所谓装饰器模式就是一种在不修改原来代码的前提下，动态的扩展代码功能的一种设计模式。当然了，这种方法有很多，比如spring的AOP，Java中的继承等，装饰器只是其中的一种。那装饰器到底是怎么样的呢？
  举个现实生活中的例子好了：
  首先，爱斯坦同学通过艰苦的实验发明了灯泡，为人类带来了光明，一开始大家都很happy，但是使用一段时间之后发现仅仅只是把灯泡挂在墙上太局限了。张三想要一个放在桌子上的灯泡，可以在看书的时候照明，李四想要那在手里的灯泡，晚上走路的时候可以用来照明。针对新的需求，我们需要发明新的灯泡吗？当然不需要啊，我们只要做到让灯泡可以在桌子上使用，可以在手里使用就可以了，这就是台灯和手电。
  台灯和手电都只是给灯泡做了个包装而已，这个包装的过程就是装饰器。

Python 装饰器

在python中函数至上，同样的装饰器在python也是通过函数调用来实现的。

def func():
      print("function called!")

def wrapper(func):
      print("wrappered function called!")
      func()      

  上面定义了两个函数，func，wrapper,wrapper其实就是func的一种包装，在func的基础上添加了新的功能。
  python针对装饰器做了优化，程序员可以通过@很简单的就写出装饰器代码。

def decorate(func):
    def wrapper(*args,**kwargs):
        print("Wrapped Decorate Method! with args:%s" %(str(a)))
        return func(*args,**kwargs)
    return wrapper

@decorate
def aMethod():
    pass

@decorate
def bMethod():
    pass

aMethod() //调用装饰器代码
bMethod() //调用装饰器代码

上面使用python的语法糖简单的实现了装饰器模式。转换成实现代码如下：

def decorate(func):
    def wrapper(*args,**kwargs):
        print("Wrapped Decorate Method! with args:%s" %(str(a)))
        return func(*args,**kwargs)
    return wrapper

def aMethod():
    pass

def bMethod():
    pass

aMethod = decorate(aMethod)  #python 中 @ 实现了此处的功能
bMethod = decorate(bMethod)  #python 中 @ 实现了此处的功能

aMethod()
bMethod()

  此外，还可以给装饰器传递参数。

def decorate_with_args(a):
    def decorate(func):
        def wrapper(*args,**kwargs):
            print("Wrapped Decorate Method! with args:%s" %(str(a)))
            return func(*args,**kwargs)
        return wrapper
    return decorate

@decorate_with_args(1)
def aMethod(a,b):
    print("A Method(%s,%s)" %(a,b))

def bMethod(a,b):
    print("B Method(%s,%s)" %(str(a),str(b)))

bMethod = decorate_with_args(2)(bMethod)

aMethod()
bMethod()