prevent nested monitoring

Author: Lendemor

reflex/utils/monitoring.py

@@ -4,6 +4,7 @@
 import contextlib
 import functools
 import inspect
+import threading
 from collections.abc import AsyncGenerator, Awaitable, Callable, Generator
 from typing import TypeVar, overload
 
@@ -20,6 +21,10 @@
     LeakAction = None  # pyright: ignore[reportAssignmentType]
 
 
+# Thread-local storage to track if monitoring is already active
+_thread_local = threading.local()
+
+
 def is_pyleak_enabled() -> bool:
     """Check if PyLeak monitoring is enabled and available.
 
@@ -43,18 +48,28 @@ def monitor_sync():
         yield
         return
 
+    # Check if monitoring is already active in this thread
+    if getattr(_thread_local, "monitoring_active", False):
+        yield
+        return
+
     config = get_config()
     action = config.pyleak_action or LeakAction.WARN  # pyright: ignore[reportOptionalMemberAccess]
 
-    with contextlib.ExitStack() as stack:
-        # Thread leak detection has issues with background tasks (no_thread_leaks)
-        stack.enter_context(
-            no_event_loop_blocking(  # pyright: ignore[reportOptionalCall]
-                action=action,
-                threshold=config.pyleak_blocking_threshold,
+    # Mark monitoring as active
+    _thread_local.monitoring_active = True
+    try:
+        with contextlib.ExitStack() as stack:
+            # Thread leak detection has issues with background tasks (no_thread_leaks)
+            stack.enter_context(
+                no_event_loop_blocking(  # pyright: ignore[reportOptionalCall]
+                    action=action,
+                    threshold=config.pyleak_blocking_threshold,
+                )
             )
-        )
-        yield
+            yield
+    finally:
+        _thread_local.monitoring_active = False
 
 
 @contextlib.asynccontextmanager
@@ -68,23 +83,33 @@ async def monitor_async():
         yield
         return
 
+    # Check if monitoring is already active in this thread
+    if getattr(_thread_local, "monitoring_active", False):
+        yield
+        return
+
     config = get_config()
     action = config.pyleak_action or LeakAction.WARN  # pyright: ignore[reportOptionalMemberAccess]
 
-    async with contextlib.AsyncExitStack() as stack:
-        # Thread leak detection has issues with background tasks (no_thread_leaks)
-        # Re-add thread leak later.
-
-        # Block detection for event loops
-        stack.enter_context(
-            no_event_loop_blocking(  # pyright: ignore[reportOptionalCall]
-                action=action,
-                threshold=config.pyleak_blocking_threshold,
+    # Mark monitoring as active
+    _thread_local.monitoring_active = True
+    try:
+        async with contextlib.AsyncExitStack() as stack:
+            # Thread leak detection has issues with background tasks (no_thread_leaks)
+            # Re-add thread leak later.
+
+            # Block detection for event loops
+            stack.enter_context(
+                no_event_loop_blocking(  # pyright: ignore[reportOptionalCall]
+                    action=action,
+                    threshold=config.pyleak_blocking_threshold,
+                )
             )
-        )
-        # Task leak detection has issues with background tasks (no_task_leaks)
+            # Task leak detection has issues with background tasks (no_task_leaks)
 
-        yield
+            yield
+    finally:
+        _thread_local.monitoring_active = False
 
 
 YieldType = TypeVar("YieldType")