Python設置函數調用超時

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