如何防止Shell脚本函数访问远程资源超时？

超时函数：如何限制执行时间

在 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中文网其他相关文章！