fix: Produce SessionCancelledEvent for pending-timeout sessions

lablup · Mar 30, 2022 · 70b78f5 · 70b78f5
1 parent 6edf9ba
commit 70b78f5
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/ai/backend/manager/scheduler/dispatcher.py b/src/ai/backend/manager/scheduler/dispatcher.py
@@ -294,6 +294,14 @@ async def _apply_cancellation():
                     await db_conn.execute(query)
 
             await execute_with_retry(_apply_cancellation)
+            for item in cancelled_session_rows:
+                await self.event_producer.produce_event(
+                    SessionCancelledEvent(
+                        item['session_id'],
+                        item['session_creation_id'],
+                        reason="pending timeout",
+                    ),
+                )
 
         log.debug(
             "running scheduler (sgroup:{}, pending:{}, existing:{}, cancelled:{})",