mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2026-04-23 00:17:25 +08:00
yzwu
26 lines
692 B
Python
26 lines
692 B
Python
import functools
|
|
import signal
|
|
|
|
|
|
def timeout(seconds):
|
|
def decorator(func):
|
|
def _handle_timeout(signum, frame):
|
|
raise TimeoutError(f"Function '{func.__name__}' timed out after {seconds} seconds")
|
|
|
|
@functools.wraps(func)
|
|
def wrapper(*args, **kwargs):
|
|
original_handler = signal.signal(signal.SIGALRM, _handle_timeout)
|
|
signal.alarm(seconds)
|
|
|
|
try:
|
|
result = func(*args, **kwargs)
|
|
signal.alarm(0)
|
|
return result
|
|
finally:
|
|
signal.signal(signal.SIGALRM, original_handler)
|
|
signal.alarm(0)
|
|
|
|
return wrapper
|
|
|
|
return decorator
|