首页 > 后端开发 > Python教程 > 如何防止Shell脚本函数访问远程资源超时?

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

Patricia Arquette
发布: 2024-11-30 13:02:19
原创
264 人浏览过

How Can I Prevent Shell Script Functions from Timing Out When Accessing Remote Resources?

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

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

来源:php.cn
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
作者最新文章
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板