Fix process restart after error exit (#14)

johnwason · Mar 5, 2024 · 84c80de · 84c80de
1 parent 6bffacb
commit 84c80de
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/drekar_launch.py b/drekar_launch.py
@@ -126,7 +126,8 @@ async def run(self):
                 if not s.restart:
                     break
                 if self._keep_going:
-                    await self.exit_event.wait(s.restart_backoff)
+                    with suppress(asyncio.TimeoutError):
+                        await asyncio.wait_for(self.parent.exit_event.wait(), timeout=s.restart_backoff)
 
     @property
     def process_state(self):