2021-05-10 python 进程超时自动退出 装饰器

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

用于pytest数据驱动，一个用例超时退出会导致剩下用例无法运行，目前没想到好的解法

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

思路
利用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() #满足第3点
        return wrapper
    return decorator

   

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

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

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

————————————————
版权声明：本文为CSDN博主「xjc_glenn1995」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/Jcggnice/article/details/108524617