Python中利用signal模块实现函数调用超时控制

Python函数超时控制背景

背景介绍

近期在编写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
 
 
for i in range(1, 10):
 try:
  signal.signal(signal.SIGALRM, handler)
  signal.alarm(3)
  test(i)
 except AssertionError:
  print "%d timeout" % (i)
 finally:
  signal.alarm(0)
  signal.signal(signal.SIGALRM, signal.SIG_DFL)

代码逻辑与执行流程解析

代码逻辑说明

首先定义一个测试函数test，通过time.sleep模拟不同耗时的执行场景。

引入Python标准库中的signal模块，定义一个信号处理函数handler，用于捕获超时信号并抛出AssertionError异常。

使用signal.alarm(3)设置3秒的闹钟，这意味着如果test函数执行时间超过3秒，系统将触发SIGALRM信号，进而执行handler返回错误。

通过try-except块捕获可能发生的超时异常，并在except分支中打印超时提示信息，确保程序流程不被中断。

程序执行结果分析

程序运行输出结果

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中通过signal模块有效处理函数调用超时及进程卡住的问题，希望能为遇到类似问题的开发者提供参考。