From b38bd902d7e10b3e944f613823a3d4d8000030be Mon Sep 17 00:00:00 2001 From: Matt Corallo Date: Tue, 3 Dec 2019 01:46:35 -0500 Subject: [PATCH] Impose a max conns/sec --- src/datastore.rs | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/datastore.rs b/src/datastore.rs index 5f9874f..1ae9dd9 100644 --- a/src/datastore.rs +++ b/src/datastore.rs @@ -19,6 +19,7 @@ use regex::Regex; use crate::bgp_client::BGPClient; pub const SECS_PER_SCAN_RESULTS: u64 = 15; +const MAX_CONNS_PER_SEC_PER_STATUS: u64 = 30; #[derive(Clone, Copy, Hash, PartialEq, Eq)] pub enum AddressState { @@ -610,7 +611,8 @@ impl Store { for (idx, state_nodes) in nodes.state_next_scan.iter_mut().enumerate() { let rescan_interval = cmp::max(self.get_u64(U64Setting::RescanInterval(AddressState::from_num(idx as u8).unwrap())), 1); let cmp_time = cur_time - Duration::from_secs(rescan_interval); - let split_point = cmp::min(SECS_PER_SCAN_RESULTS * state_nodes.len() as u64 / rescan_interval, + let split_point = cmp::min(cmp::min(SECS_PER_SCAN_RESULTS * state_nodes.len() as u64 / rescan_interval, + SECS_PER_SCAN_RESULTS * MAX_CONNS_PER_SEC_PER_STATUS), state_nodes.binary_search_by(|a| a.0.cmp(&cmp_time)).unwrap_or_else(|idx| idx) as u64); let mut new_nodes = state_nodes.split_off(split_point as usize); mem::swap(&mut new_nodes, state_nodes); -- 2.30.2