add distributed lookahead

This commit is contained in:
Tan Chee Keong
2025-03-16 21:27:07 +08:00
parent 0c360c893e
commit 50ec5f17a1

View File

@@ -1137,8 +1137,17 @@ async fn fill_in_selection_proofs<T: SlotClock + 'static, E: EthSpec>(
let lookahead_slot = current_slot + selection_lookahead;
let mut relevant_duties = duties_by_slot.split_off(&lookahead_slot);
std::mem::swap(&mut relevant_duties, &mut duties_by_slot);
let relevant_duties = if duties_service.distributed {
if let Some(duties) = duties_by_slot.remove(&lookahead_slot) {
BTreeMap::from([(lookahead_slot, duties)])
} else {
BTreeMap::new()
}
} else {
let mut duties = duties_by_slot.split_off(&lookahead_slot);
std::mem::swap(&mut duties, &mut duties_by_slot);
duties
};
let batch_size = relevant_duties.values().map(Vec::len).sum::<usize>();