使函數超時
在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中文網其他相關文章！
