python,限制任意函数,线程的执行时间或根据条件终止.

2019独角兽企业重金招聘Python工程师标准>>>

想写一个通用的decorator限制一个函数的执行时间.

首先想到的就是把函数执行封在一个线程了,执行结果写到线程的属性里,join指定时间后读线程的属性并返回.问题在于若函数没有执行完,读不到属性,返回后.这个线程将继续存活.在python中,无法直接杀死一个线程.这个线程也就一直占用着资源直到线程结束.

一个方法是,让那个函数或线程周期性的检测一个标志位_stopped,若标志位_stopped被外部赋值True,则不再执行(直接返回或抛出异常终止).但并不是所有函数都能这样,这个方法略戳.需要改被修饰函数的代码,不通用.

类似如下函数,在不改动他的前提下限制他的执行时间.

def forever():
    while True:
        print('forever')
        time.sleep(1)

这个问题的解如下:

参考:http://stackoverflow.com/questions/323972/is-there-any-way-to-kill-a-thread-in-python

原理是,从外部让该线程产生一个异常.详见该网页.

-.-

当然限制任意函数执行时间也可以通过把函数封在一个进程里面执行,py的multiprocess中,Process可以被terminate.也可以手动给信号发TERM,KILL信号.不过process太重量级了.

另外看了下gevent,gevent的greenlet有kill方法.未测试是否能限制任意函数的执行时间(如上所示那种).目测可行.原理大概是可以给指定的greenlet生成一个子greenlet,然后在子greenlet中抛出这个异常,这个异常会沿着他的父greenlet一直向上传.就达到了终止这个greenlet的效果.

试了gevent和greenlet.实际是无法通过协程(greenlet)来实现限制函数执行时间.目测可以的,从原理上来说不行,实际也不行.

在gevent中有.
gevent.with_timeout(seconds, function, *args, **kwds)
实测该方法无法限制函数执行时间,目测该方法只能用于非阻塞方法限制执行时间
在greenlet中,尝试在另一个线程中对greenlet进行switch会报错说无法切换线程.

从原理上来说,greenlet只有一个执行流,切换全是手动的.没法从外部干预这个执行流.

转载于:https://my.oschina.net/u/1024140/blog/122778