Python实现自己的AOP

Java中的AOP可以用JDK的动态代理和cglib来实现,将需要拦截的方法前后可以额外添加些功能。

Python中有许多方法去实现AOP,现在先介绍第一种比较简单的:

1)利用with...as...

Python的with...as...子句是用于控制执行流程的语句,结构是with expression as variable,这里的expression里定义拦截的方法前后需要执行的逻辑,分别定义“__enter__”和“__exit__”方法,“__enter__”方法定义拦截方法执行前的逻辑,“__exit__”方法定义拦截方法执行后的逻辑,with...as...子句的主体是拦截的方法执行的逻辑,variable变量是用来接收“__enter__”方法返回的值,下面来看个示例:

class AOP: 
     def __enter__(self): 
          print("before action enter") 
          return 1 
 
     def __exit__(self, *args):
          print("after action enter")
          return 0
 
def Myfunction(): 
     print ('Myfunction() called and processed ')
     return 111 

def Myfunction1(): 
     print ('Myfunction1()  called and processed')
     return 222


def MyMain():
           with AOP() as AOPValue:
                      print ('AOPValue is %s'%AOPValue)
                      Myfunction()
                      Myfunction1()
          
if __name__ == '__main__':
           MyMain()

 
最后输出的结果是:

before action enter
AOPValue is 1
Myfunction() called and processed
Myfunction1()  called and processed
after action enter

PS:with子句中的variable接收的是__enter__方法的返回值。这是with子句的基本用法

2)利用contextlib实现AOP

Python核心库中的contextlib中的contextmanager来实现AOP就非常简单了,contextmanager是一个装饰器,具体用法如下示例:

from contextlib import contextmanager 

#define inteceptors

@contextmanager 
def AOPfunc(name): 
    print ("before:some actions done") 
    yield 
    print ("after:some actions done ") 

#normal service flow 
def BusinessProcess1(): 
     print ('BusinessProcess1()')
     return 111 

def BusinessProcess2(): 
     print ('BusinessProcess2()')
     return 111

def BusinessProcess3(): 
     print ('BusinessProcess3()')
     return 111

#main flow
def MyMain():
           with AOPfunc("aop"):
                      BusinessProcess1()
                      BusinessProcess2()
                     
           with AOPfunc("aop"):
                       BusinessProcess3()


if __name__ == '__main__':
           MyMain()

 
运行结果是:
 
before:some actions done
BusinessProcess1()
BusinessProcess2()
after:some actions done
before:some actions done
BusinessProcess3()
after:some actions done
 

你可能感兴趣的:(Python实现自己的AOP)