如何为长时间运行的 Shell 脚本函数实现超时？

长时间 Shell 脚本函数的超时实现

在 shell 脚本编写中，您可能会遇到某些函数执行时间过长，导致延迟的情况甚至无限循环。为了防止这种情况，如果这些函数未能在指定的时间范围内完成，通常会实现超时来终止这些函数。

一种有效的方法是使用异步计时器来强制超时。通过设置计时器，如果超过指定的持续时间，您可以终止函数，确保脚本继续按预期执行。

要实现这一点，您可以利用 Python 提供的信号模块。以下是定义超时装饰器来包装函数的方法：

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

一旦有了超时装饰器，您就可以将其应用于您想要在设置的超时后终止的任何函数。例如，如果你有一个名为 long_running_function 的函数，你想在 10 秒后超时，你可以按如下方式装饰它：

from timeout import timeout

@timeout
def long_running_function():
    ...

通过使用超时装饰器，你可以有效地防止你的 shell 脚本函数因长期执行而被无限期绞死。这种方法可确保脚本继续执行，而不会出现不必要的延迟或中断。

以上是如何为长时间运行的 Shell 脚本函数实现超时？的详细内容。更多信息请关注PHP中文网其他相关文章！