SQ: Bulkheads: Protect Task.try_get_next_task_unit() and all overridi…

…ng implementations
davidfstr · Feb 3, 2024 · 57c0fba · 57c0fba
1 parent dbfd83d
commit 57c0fba
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/crystal/task.py b/src/crystal/task.py
@@ -421,6 +421,7 @@ def clear_completed_children(self) -> None:
 
     # === Public Operations ===
 
+    @capture_crashes_to(self)
     @fg_affinity
     def try_get_next_task_unit(self) -> Optional[Callable[[], None]]:
         """
@@ -1318,6 +1319,7 @@ def __init__(self, group: ResourceGroup) -> None:
         self._children_loaded = False
 
     @overrides
+    @capture_crashes_to(self)
     def try_get_next_task_unit(self) -> Optional[Callable[[], None]]:
         if not self._children_loaded:
             @capture_crashes_to(self)
@@ -1570,6 +1572,8 @@ def fg_task() -> None:
         # NOTE: Must synchronize access to RootTask.children with foreground thread
         fg_call_and_wait(fg_task)
 
+    @overrides
+    @capture_crashes_to(self)
     @fg_affinity
     def try_get_next_task_unit(self):
         if self.complete: