Skip to content

Commit b9d007e

Browse files
committed
Save new client_operation_id to operation_id mapping in Redis.
1 parent 03d1bfa commit b9d007e

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

nativelink-scheduler/src/store_awaited_action_db.rs

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -533,9 +533,17 @@ where
533533
return Ok(None);
534534
}
535535

536+
self.store
537+
.update_data(UpdateClientIdToOperationId {
538+
client_operation_id: client_operation_id.clone(),
539+
operation_id: operation_id.clone(),
540+
})
541+
.await
542+
.err_tip(|| "In RedisAwaitedActionDb::try_subscribe")?;
543+
536544
Ok(Some(OperationSubscriber::new(
537545
Some(client_operation_id.clone()),
538-
OperationIdToAwaitedAction(Cow::Owned(operation_id)),
546+
OperationIdToAwaitedAction(Cow::Owned(operation_id.clone())),
539547
Arc::downgrade(&self.store),
540548
self.now_fn.clone(),
541549
)))

0 commit comments

Comments
 (0)