Skip to content

Commit b2cdd2e

Browse files
fix: update qos when a request is running on a killed worker
1 parent 2906793 commit b2cdd2e

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

cads_broker/dispatcher.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -316,8 +316,12 @@ def set_request_error_status(
316316
< CONFIG.broker_requeue_limit
317317
):
318318
logger.info("worker killed: re-queueing", job_id=request_uid)
319-
db.requeue_request(request=request, session=session)
320-
self.queue.add(request_uid, request)
319+
queued_request = db.requeue_request(request=request, session=session)
320+
if queued_request:
321+
self.queue.add(request_uid, request)
322+
self.qos.notify_end_of_request(
323+
request, session, scheduler=self.internal_scheduler
324+
)
321325
else:
322326
request = db.set_request_status(
323327
request_uid,

0 commit comments

Comments
 (0)