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

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

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

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

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()) 

步骤二:测试脚本执行

[root@localhost day06]# python3 deco.py 
(1.50368070602417, [1, 2, 3, 4, 5])
发布了275 篇原创文章 · 获赞 46 · 访问量 2万+

猜你喜欢

转载自blog.csdn.net/xie_qi_chao/article/details/104726225