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

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

创建deco.py脚本，要求如下：

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

    
    

    
    
1
2
3
4

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

    
    

    
    
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

步骤二：测试脚本执行

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

    
    

    
    
1
2

                                </div>
            <link href="https://csdnimg.cn/release/phoenix/mdeditor/markdown_views-60ecaf1f42.css" rel="stylesheet">
                                            <div class="more-toolbox">
            <div class="left-toolbox">
                <ul class="toolbox-list">
                    
                    <li class="tool-item tool-active is-like "><a href="javascript:;"><svg class="icon" aria-hidden="true">
                        <use xlink:href="#csdnc-thumbsup"></use>
                    </svg><span class="name">点赞</span>
                    <span class="count"></span>
                    </a></li>
                    <li class="tool-item tool-active is-collection "><a href="javascript:;" data-report-click="{&quot;mod&quot;:&quot;popu_824&quot;}"><svg class="icon" aria-hidden="true">
                        <use xlink:href="#icon-csdnc-Collection-G"></use>
                    </svg><span class="name">收藏</span></a></li>
                    <li class="tool-item tool-active is-share"><a href="javascript:;" data-report-click="{&quot;mod&quot;:&quot;1582594662_002&quot;}"><svg class="icon" aria-hidden="true">
                        <use xlink:href="#icon-csdnc-fenxiang"></use>
                    </svg>分享</a></li>
                    <!--打赏开始-->
                                            <!--打赏结束-->
                                            <li class="tool-item tool-more">
                        <a>
                        <svg t="1575545411852" class="icon" viewBox="0 0 1024 1024" version="1.1" xmlns="http://www.w3.org/2000/svg" p-id="5717" xmlns:xlink="http://www.w3.org/1999/xlink" width="200" height="200"><defs><style type="text/css"></style></defs><path d="M179.176 499.222m-113.245 0a113.245 113.245 0 1 0 226.49 0 113.245 113.245 0 1 0-226.49 0Z" p-id="5718"></path><path d="M509.684 499.222m-113.245 0a113.245 113.245 0 1 0 226.49 0 113.245 113.245 0 1 0-226.49 0Z" p-id="5719"></path><path d="M846.175 499.222m-113.245 0a113.245 113.245 0 1 0 226.49 0 113.245 113.245 0 1 0-226.49 0Z" p-id="5720"></path></svg>
                        </a>
                        <ul class="more-box">
                            <li class="item"><a class="article-report">文章举报</a></li>
                        </ul>
                    </li>
                                        </ul>
            </div>
                        </div>
        <div class="person-messagebox">
            <div class="left-message"><a href="https://blog.csdn.net/xie_qi_chao">
                <img src="https://profile.csdnimg.cn/B/F/6/3_xie_qi_chao" class="avatar_pic" username="xie_qi_chao">
                                        <img src="https://g.csdnimg.cn/static/user-reg-year/1x/2.png" class="user-years">
                                </a></div>
            <div class="middle-message">
                                    <div class="title"><span class="tit"><a href="https://blog.csdn.net/xie_qi_chao" data-report-click="{&quot;mod&quot;:&quot;popu_379&quot;}" target="_blank">解启超</a></span>
                                        </div>
                <div class="text"><span>发布了404 篇原创文章</span> · <span>获赞 56</span> · <span>访问量 4万+</span></div>
            </div>
                            <div class="right-message">
                                        <a href="https://im.csdn.net/im/main.html?userName=xie_qi_chao" target="_blank" class="btn btn-sm btn-red-hollow bt-button personal-letter">私信
                    </a>
                                                        <a class="btn btn-sm attented bt-button personal-watch" data-report-click="{&quot;mod&quot;:&quot;popu_379&quot;}">已关注</a>
                                </div>
                        </div>
                </div>
</article>