超时函数:如何限制执行时间
在 shell 脚本中,防止某些函数无休止地执行可能是有益的,尤其是当涉及访问 URL 等远程资源时。通过设置超时,您可以确保函数不会由于服务器缓慢或无响应而冻结脚本。
实现超时的一种方法是利用信号模块的超时装饰器。信号文档中描述的这个装饰器使用信号处理程序来设置警报。一旦指定的时间间隔过去,就会引发异常,从而有效地停止函数。
要使用超时装饰器,请从 timeout.py 模块导入它并指定超时持续时间(以秒为单位)作为参数。下面是一个示例片段,它创建一个默认超时为 10 秒的装饰器和自定义错误消息:
import errno import os import signal import functools class TimeoutError(Exception): pass def timeout(seconds=10, error_message=os.strerror(errno.ETIME)): def decorator(func): def _handle_timeout(signum, frame): raise TimeoutError(error_message) @functools.wraps(func) def wrapper(*args, **kwargs): signal.signal(signal.SIGALRM, _handle_timeout) signal.alarm(seconds) try: result = func(*args, **kwargs) finally: signal.alarm(0) return result return wrapper return decorator
要使用装饰器,只需将其应用于长时间运行的函数,如下所示示例:
# Default timeout of 10 seconds @timeout def long_running_function1(): ... # Timeout of 5 seconds @timeout(5) def long_running_function2(): ... # Timeout of 30 seconds, with custom error message 'Connection timed out' @timeout(30, os.strerror(errno.ETIMEDOUT)) def long_running_function3(): ...
请记住,此实现仅适用于 UNIX 系统。通过使用信号模块的装饰器实现超时,您现在可以控制函数的执行持续时间,并防止它们不必要地拖延您的 shell 脚本。
以上是如何防止Shell脚本函数访问远程资源超时?的详细内容。更多信息请关注PHP中文网其他相关文章!