gaspersic/solana
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

replace nested for_each with flat_map

Browse Source
This commit is contained in:
Tao Zhu
2022-03-18 09:44:42 -05:00
committed by Tao Zhu
parent 1c369fb55f
commit 71ea05c176
1 changed files with 17 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
core/src/unprocessed_packet_batches.rs
View File

@ -107,32 +107,32 @@ impl UnprocessedPacketBatches {
.sum() .sum()
} }
/// Iterates packets in buffered packet_batches, returns all unprocessed packet's stake, /// Iterates the inner `Vec<DeserializedPacketBatch>`.
/// and its locator /// Returns the flattened result of mapping each
/// `DeserializedPacketBatch` to a list the batch's inner
/// packets' sender's stake and their `PacketLocator`'s within the
/// `Vec<DeserializedPacketBatch>`.
#[allow(dead_code)] #[allow(dead_code)]
fn get_stakes_and_locators(&self) -> (Vec<u64>, Vec<PacketLocator>) { fn get_stakes_and_locators(&self) -> (Vec<u64>, Vec<PacketLocator>) {
let num_unprocessed_packets = self.get_unprocessed_packets_count();
let mut stakes = Vec::<u64>::with_capacity(num_unprocessed_packets);
let mut locators = Vec::<PacketLocator>::with_capacity(num_unprocessed_packets);
self.iter() self.iter()
.enumerate() .enumerate()
.for_each(|(batch_index, deserialized_packet_batch)| { .flat_map(|(batch_index, deserialized_packet_batch)| {
let packet_batch = &deserialized_packet_batch.packet_batch; let packet_batch = &deserialized_packet_batch.packet_batch;
deserialized_packet_batch deserialized_packet_batch
.unprocessed_packets .unprocessed_packets
.keys() .keys()
.for_each(|packet_index| { .map(move |packet_index| {
let p = &packet_batch.packets[*packet_index]; let p = &packet_batch.packets[*packet_index];
stakes.push(p.meta.sender_stake); (
locators.push(PacketLocator { p.meta.sender_stake,
PacketLocator {
batch_index, batch_index,
packet_index: *packet_index, packet_index: *packet_index,
}); },
}); )
}); })
})
(stakes, locators) .unzip()
} }
} }