1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
use {
crate::packet::PacketBatch,
rand::{thread_rng, Rng},
};
pub fn discard_batches_randomly(
batches: &mut Vec<PacketBatch>,
max_packets: usize,
mut total_packets: usize,
) -> usize {
while total_packets > max_packets {
let index = thread_rng().gen_range(0, batches.len());
let removed = batches.swap_remove(index);
total_packets = total_packets.saturating_sub(removed.len());
}
total_packets
}
#[cfg(test)]
mod tests {
use {super::*, crate::packet::Packet};
#[test]
fn test_batch_discard_random() {
solana_logger::setup();
let mut batch = PacketBatch::default();
batch.resize(1, Packet::default());
let num_batches = 100;
let mut batches = vec![batch; num_batches];
let max = 5;
discard_batches_randomly(&mut batches, max, num_batches);
assert_eq!(batches.len(), max);
}
}