超時函數:如何限制執行時間
在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中文網其他相關文章!