首頁 >後端開發 >Python教學 >如何在 Shell 腳本中實現具有預定義持續時間的逾時功能?

如何在 Shell 腳本中實現具有預定義持續時間的逾時功能?

Barbara Streisand
Barbara Streisand原創
2024-12-07 11:25:15330瀏覽

How to Implement a Timeout Function in Shell Scripting with a Predefined Duration?

具有預定義持續時間的超時函數

在shell 腳本中,有一種機制可以終止長時間運行的任務,以避免過度等待。這裡面臨的挑戰是將函數包裝在逾時腳本中,如果超過指定的時間限制,則傳回 False。

一種方法是設定一個非同步計時器,在預先定義的時間間隔後觸發 False 回應。幸運的是,可以使用 signal 函式庫(在基於 UNIX 的系統上可用)中的訊號處理程序來實現這樣的解決方案。

這個過程涉及建立利用訊號的自訂裝飾器 (@timeout)。 Alarm() 設定所需時間間隔的鬧鐘。在裝飾函數中,當警報到期時,會引發 TimeoutError 異常,從而有效地中斷操作。

要將此解決方案合併到您的程式碼中,請將以下程式碼儲存為timeout.py 並匯入它:

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

現在,在您的應用程式程式碼中,使用@timeout 裝飾器註解任何可能長時間運行的函數。例如:

from timeout import timeout

# Short timeout
@timeout(5)
def slow_function():
    # ...

# Default timeout
@timeout
def another_slow_function():
    # ...

# Customize timeout and error message
@timeout(30, error_message="Task timed out")
def yet_another_slow_function():
    # ...

這種方法確保如果函數花費的時間超過指定的超時間隔,它將引發 TimeoutError 並返回 False,從而允許您優雅地處理意外延遲。

以上是如何在 Shell 腳本中實現具有預定義持續時間的逾時功能?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn