首頁 > 後端開發 > Python教學 > 如何防止Shell腳本函數存取遠端資源逾時?

如何防止Shell腳本函數存取遠端資源逾時?

Patricia Arquette
發布: 2024-11-30 13:02:19
原創
249 人瀏覽過

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
作者最新文章
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板