Tracking current stage on pipeline unwind (#6558)

This commit is contained in:
Luca Provini
2024-04-29 13:15:38 +02:00
committed by GitHub
parent 1f6753b84a
commit 1fbcdeb065

View File

@ -217,6 +217,17 @@ impl<DB> NodeState<DB> {
self.current_stage = None;
}
}
PipelineEvent::Unwind { stage_id, input } => {
let current_stage = CurrentStage {
stage_id,
eta: Eta::default(),
checkpoint: input.checkpoint,
target: Some(input.unwind_to),
entities_checkpoint: input.checkpoint.entities(),
};
self.current_stage = Some(current_stage);
}
_ => (),
}
}