yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-09-04 06:33:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

MINOR - Only timeout on main threads (#18341)

Browse Source
This commit is contained in:
Pere Miquel Brull 2024-10-21 15:18:33 +02:00
parent 7d42926960
commit 9f60ba68f5
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
ingestion/src/metadata/utils/timeout.py
View File

@ -18,6 +18,7 @@ import inspect
import os import os
import platform import platform
import signal import signal
import threading
from typing import Callable from typing import Callable
from metadata.utils.constants import TEN_MIN from metadata.utils.constants import TEN_MIN
@ -47,7 +48,11 @@ def timeout(seconds: int = TEN_MIN) -> Callable:
def decorator(fn): def decorator(fn):
@functools.wraps(fn) @functools.wraps(fn)
def inner(*args, **kwargs): def inner(*args, **kwargs):
if platform.system() != "Windows": # SIGALRM not supported on Windows # SIGALRM is not supported on Windows or sub-threads
if (
platform.system() != "Windows"
and threading.current_thread() == threading.main_thread()
):
signal.signal(signal.SIGALRM, _handle_timeout) signal.signal(signal.SIGALRM, _handle_timeout)
signal.alarm(seconds) signal.alarm(seconds)
try: try: