如何為Python中的函數設定逾時以防止死鎖？

使函數超時

在shell 腳本編寫中，經常會遇到某些任務（例如螢幕抓取網站）可能會變得無法執行的情況意外地耗時甚至無法完成。為了解決這個問題，我們可以實作一個超時機制，以便在超過預定時間限制時終止函數。

實作逾時的一種方法是使用 Python 的訊號模組。此模組允許我們設定計時器並在時間到期時引發異常。但需要注意的是，該解決方案僅適用於 UNIX 環境。

要實現超時功能，我們可以使用回應中提供的程式碼建立一個名為 @timeout 的裝飾器。此裝飾器可以應用於任何長時間運行的函數以強制執行時間限制。為了清楚起見，這裡是程式碼的修改版本：

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

@timeout
def my_long_running_function():
    ...

def my_main_function():
    try:
        my_long_running_function()
    except TimeoutError as e:
        print(e)

透過將@timeout 裝飾器套用到my_long_running_function ，它現在有10 秒的時間限制。如果函數超過這個截止時間，它將引發 TimeoutError 異常，可以在主函數中處理。此機制可確保您的腳本不會因任務緩慢或無回應而無限期地卡住。

以上是如何為Python中的函數設定逾時以防止死鎖？的詳細內容。更多資訊請關注PHP中文網其他相關文章！