fix a qos error when workers are killed

francesconazzaro · francesconazzaro · commit a8fa3387d53a · 2024-05-06T09:38:34.000+02:00
diff --git a/cads_broker/dispatcher.py b/cads_broker/dispatcher.py
@@ -246,8 +246,7 @@ def on_future_done(self, future: distributed.Future) -> None:
                     job_status={future.status},
                     job_id=request.request_uid,
                 )
-                return
-            self.futures.pop(future.key)
+            self.futures.pop(future.key, None)
             self.qos.notify_end_of_request(
                 request, session, scheduler=self.internal_scheduler
             )

Original file line number	Diff line number	Diff line change
`@@ -246,8 +246,7 @@ def on_future_done(self, future: distributed.Future) -> None:`
`246`	`246`	`job_status={future.status},`
`247`	`247`	`job_id=request.request_uid,`
`248`	`248`	`)`
`249`		`- return`
`250`		`- self.futures.pop(future.key)`
	`249`	`+ self.futures.pop(future.key, None)`
`251`	`250`	`self.qos.notify_end_of_request(`
`252`	`251`	`request, session, scheduler=self.internal_scheduler`
`253`	`252`	`)`