add notify end of request after requeue

francesconazzaro · francesconazzaro · commit 2ef80ca2a61c · 2024-05-03T11:28:29.000+02:00
diff --git a/cads_broker/dispatcher.py b/cads_broker/dispatcher.py
@@ -194,6 +194,9 @@ def sync_database(self, session: sa.orm.Session) -> None:
                     "Request not found: re-queueing", job_id={request.request_uid}
                 )
                 db.requeue_request(request_uid=request.request_uid, session=session)
+                self.qos.notify_end_of_request(
+                    request, session, scheduler=self.internal_scheduler
+                )
 
     def on_future_done(self, future: distributed.Future) -> None:
         job_status = DASK_STATUS_TO_STATUS.get(future.status, "accepted")

Original file line number	Diff line number	Diff line change
`@@ -194,6 +194,9 @@ def sync_database(self, session: sa.orm.Session) -> None:`
`194`	`194`	`"Request not found: re-queueing", job_id={request.request_uid}`
`195`	`195`	`)`
`196`	`196`	`db.requeue_request(request_uid=request.request_uid, session=session)`
	`197`	`+ self.qos.notify_end_of_request(`
	`198`	`+ request, session, scheduler=self.internal_scheduler`
	`199`	`+ )`
`197`	`200`
`198`	`201`	`def on_future_done(self, future: distributed.Future) -> None:`
`199`	`202`	`job_status = DASK_STATUS_TO_STATUS.get(future.status, "accepted")`