Automatically call .recycle in Drop (#7429)

automerge
This commit is contained in:
anatoly yakovenko
2019-12-11 11:58:40 -08:00
committed by Grimes
parent 0aa4dc904e
commit bec5835289
5 changed files with 74 additions and 111 deletions

View File

@@ -1,11 +1,8 @@
//! The `packet` module defines data structures and methods to pull data from the network.
use crate::{
cuda_runtime::PinnedVec,
recycler::{Recycler, Reset},
};
use crate::{cuda_runtime::PinnedVec, recycler::Recycler};
use serde::Serialize;
pub use solana_sdk::packet::{Meta, Packet, PACKET_DATA_SIZE};
use std::{mem, net::SocketAddr};
use std::net::SocketAddr;
pub const NUM_PACKETS: usize = 1024 * 8;
@@ -16,38 +13,13 @@ pub const PACKETS_BATCH_SIZE: usize = (PACKETS_PER_BATCH * PACKET_DATA_SIZE);
#[derive(Debug, Clone)]
pub struct Packets {
pub packets: PinnedVec<Packet>,
recycler: Option<PacketsRecycler>,
}
impl Drop for Packets {
fn drop(&mut self) {
if let Some(ref recycler) = self.recycler {
let old = mem::replace(&mut self.packets, PinnedVec::default());
recycler.recycle(old)
}
}
}
impl Reset for Packets {
fn reset(&mut self) {
self.packets.resize(0, Packet::default());
}
fn warm(&mut self, size_hint: usize) {
self.packets.set_pinnable();
self.packets.resize(size_hint, Packet::default());
}
}
//auto derive doesn't support large arrays
impl Default for Packets {
fn default() -> Packets {
let packets = PinnedVec::with_capacity(NUM_RCVMMSGS);
Packets {
packets,
recycler: None,
}
Packets { packets }
}
}
@@ -56,19 +28,13 @@ pub type PacketsRecycler = Recycler<PinnedVec<Packet>>;
impl Packets {
pub fn new(packets: Vec<Packet>) -> Self {
let packets = PinnedVec::from_vec(packets);
Self {
packets,
recycler: None,
}
Self { packets }
}
pub fn new_with_recycler(recycler: PacketsRecycler, size: usize, name: &'static str) -> Self {
let mut packets = recycler.allocate(name);
packets.reserve_and_pin(size);
Packets {
packets,
recycler: Some(recycler),
}
Packets { packets }
}
pub fn new_with_recycler_data(
recycler: &PacketsRecycler,
@@ -142,15 +108,6 @@ mod tests {
use solana_sdk::signature::{Keypair, KeypairUtil};
use solana_sdk::system_transaction;
#[test]
fn test_packets_reset() {
let mut packets = Packets::default();
packets.packets.resize(10, Packet::default());
assert_eq!(packets.packets.len(), 10);
packets.reset();
assert_eq!(packets.packets.len(), 0);
}
#[test]
fn test_to_packets() {
let keypair = Keypair::new();