python 进程超时自动退出 装饰器

时间:2020-9-12 作者:admin

python 进程超时自动退出 装饰器


为啥写这个装饰器

之前做用例测试的时候,单进程测试,发现有些用例执行时间特别长,卡死,影响下面用例的执行。刚开始在主函数内添加了超时退出,但是感觉用的不方便,虽然也能实现功能,但是主函数变得特别累赘,所以还是选择写一个装饰器。

思路
利用python的多线程一个线程控制主进程运行时间,时间到则利用线程os._exit()迫使主进程退出。另一个线程调用主进程函数执行脚本。

代码

from functools import wraps
import time
import os
from threading import Thread

def set_time_limit(t):
    def auto_quit(t1):
        '''此为控制进程超时退出的线程函数'''
        time.sleep(t1)
        print("time out {}".format(t1))
        os._exit(1) #此函数专门用于线程控制主进程退出,有兴趣的可以看一下和sys.exit()的区别
    def decorator(f):
        '''此函数用于传入被装饰函数f'''
        @wraps(f)
        def wrapper(*args,**kwargs):
            '''装饰器内部遵循的逻辑是:
            1.auto_quit先执行完,进程结束
            2.被修饰函数f先执行完,auto_quit函数停止执行
            3.被修饰函数执行完,下面的代码才能运行
            '''
            t1=Thread(target=auto_quit,args=(t,))  #此处的t是set_time_limit函数的形参,是auto_quit函数的实参
            t2=Thread(target=f,args=args,kwargs=kwargs)
            t1.setDaemon(True) #满足第2点
            t1.start()
            t2.start()
            t2.join() #满足第3return wrapper
    return decorator

测试代码
设置超时时间为3s,主函数执行时间为2s

@set_time_limit(3)
def a(t):
    time.sleep(t)
    print("execuate finished")

if __name__=="__main__":
    a(2)
    print("执行完毕")

执行完结果
执行完成,auto_quit函数没有调用
python 进程超时自动退出 装饰器
测试代码
设置超时时间为2s,主函数执行时间为3s

@set_time_limit(2)
def a(t):
    time.sleep(t)
    print("execuate finished")

if __name__=="__main__":
    a(3)
    print("执行完毕")

执行结果
追进程超时退出
python 进程超时自动退出 装饰器

你们也可以把t1.setDaemon(True)注释掉(超时时间设长一些,执行时间短一些)执行一下看一下这样就会理解啥是守护线程
你会发现执行一直不结束,在等待auto_quit执行结束

你们也可以把t2.join()注释掉
会发现无论程序是否超时退出都先打印:执行完毕

声明:本文内容由互联网用户自发贡献自行上传,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任。如果您发现有涉嫌版权的内容,欢迎进行举报,并提供相关证据,工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。