solana-perf 4.0.0

Solana Performance APIs
Documentation 
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
35
36
37
38
39

use {
    crate::packet::PacketBatch,
    rand::{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 = rng().random_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::{BytesPacket, BytesPacketBatch, Meta},
        bytes::Bytes,
    };

    #[test]
    fn test_batch_discard_random() {
        agave_logger::setup();
        let mut batch = BytesPacketBatch::new();
        batch.resize(1, BytesPacket::new(Bytes::new(), Meta::default()));
        let batch = PacketBatch::from(batch);
        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);
    }
}