Python设置函数调用超时

发布时间：2019-09-18 07:29:46编辑：auto阅读（2106）

    http://blog.sina.com.cn/s/blog_63041bb80102uy5o.html

    背景：

           最近写的Python代码不知为何，总是执行到一半卡住不动，为了使程序能够继续运行，设置了函数调用超时机制。

    代码：
    import time
    import signal

    def test(i):
        time.sleep(i%4)
        print "%d within time"%(i)
        return i

    if __name__ == '__main__':
        def handler(signum, frame):
            raise AssertionError

        i = 0
        for i in range(1,10):
            try:
                signal.signal(signal.SIGALRM, handler)
                signal.alarm(3)
                test(i)
                i = i + 1
                signal.alarm(0)
            except AssertionError:
                print "%d timeout"%(i)

    说明：
            1、调用test函数超时监控，使用sleep模拟函数执行超时
            2、引入signal模块，设置handler捕获超时信息，返回断言错误
            3、alarm(3)，设置3秒闹钟，函数调用超时3秒则直接返回
            4、捕获异常，打印超时信息

    程序执行结果：
    1 within time
    2 within time
    3 timeout
    4 within time
    5 within time
    6 within time
    7 timeout
    8 within time
    9 within time

关键字：

上一篇： 教你阅读Python开源项目代码

下一篇： ubuntu sublime text3