《云计算全栈》-python篇:使用装饰器实现-测试程序运行效率

Agnes ·
更新时间:2024-05-16
· 794 次阅读

4 案例4:测试程序运行效率
4.1 问题

创建deco.py脚本,要求如下:

有个程序包含多个函数 程序运行耗时较长 为了确定哪个函数是瓶颈,需要计算出每个函数运行时间 要求使用装饰器实现 1234

4.2 方案

如果一个程序有多个函数,查看每个函数运行耗时时长,这时如果修改每个函数,为它加上计时的功能,我们需要耗时去了解每个函数的功能,思考如何修改去增加计时功能,这样会相当繁琐,为避免这种现象出现,我们利用装饰器函数在不变动其他函数基础上,新增计时功能,在每个函数前添加语法糖(调用装饰器函数)即可,实施方法如下:

1.首先,定义一个函数loop(),该函数可以拥有任何功能,这里将for循环输出的数字添加到空列表中,并让进程挂起0.3s的时间

2.定义装饰器函数计算loop()函数运行耗时时长,装饰器函数deco,该函数的传入参数是loop (即被装饰函数),返回参数是内层函数。这里的内层函数-timeit,其实就相当于闭包函数,它起到装饰给定函数的作用

3.@ deco这个语法相当于执行loop = deco(loop),为loop函数装饰并返回

4.装饰器函数在调用loop函数前后各返回一个当前时间,返回两个时间差即计算出运行耗时。

5.装饰器函数参数是你要装饰的函数名,装饰器函数返回是装饰完的函数名

需要要注意的是:为了不破坏原函数的逻辑,我们要保证内层函数timeit和被装饰函数loop的传入参数和返回值类型必须保持一致。
4.3 步骤

实现此案例需要按照如下步骤进行。

步骤一:编写脚本

[root@localhost day05]# vim deco.py #!/usr/bin/env python3 import time def deco(func): def timeit(): start = time.time() result = func() end = time.time() return end - start, result # 将会返回元组 return timeit @deco def loop(): result = [] for i in range(1, 6): result.append(i) time.sleep(0.3) return result if __name__ == '__main__': # loop = deco(loop) print(loop()) 1234567891011121314151617181920

步骤二:测试脚本执行

[root@localhost day06]# python3 deco.py (1.50368070602417, [1, 2, 3, 4, 5]) 12



运行 装饰器 程序 效率 测试 云计算 Python

需要 登录 后方可回复, 如果你还没有账号请 注册新账号