Python实现自己的AOP

2023-01-31 02:01:52 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

相关文章