teascade/teanet-rust
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add toggle to disable accepting new connections

Browse Source
This commit is contained in:
Sofia 2026-07-17 22:12:17 +03:00
parent 4d843d2ccb
commit a947f3a8cb
2 changed files with 20 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/connections.rs
View File

@ -49,6 +49,8 @@ pub(crate) struct ConnectionManager<
connections: HashMap<SocketAddr, Connection<T>>,
closing_since: Option<Instant>,
config: PeerConfig,
accepting_connections: bool,
}
impl<T: std::fmt::Debug + Clone + Serialize + DeserializeOwned + Send + Sync + 'static>
@ -61,6 +63,7 @@ impl<T: std::fmt::Debug + Clone + Serialize + DeserializeOwned + Send + Sync + '
connections: HashMap::new(),
closing_since: None,
config,
accepting_connections: true,
}
}
@ -258,10 +261,15 @@ impl<T: std::fmt::Debug + Clone + Serialize + DeserializeOwned + Send + Sync + '
conn.state = ConnectionState::ConnectingNearlyReady;
}
} else {
self.connections.insert(
*addr,
Connection::from(*addr, ConnectionState::ReceivingConnection),
);
if !self.accepting_connections {
self.connections
.insert(*addr, Connection::from(*addr, ConnectionState::Closing));
} else {
self.connections.insert(
*addr,
Connection::from(*addr, ConnectionState::ReceivingConnection),
);
}
}
}
Package::Ping => {
@ -413,6 +421,10 @@ impl<T: std::fmt::Debug + Clone + Serialize + DeserializeOwned + Send + Sync + '
pub fn connections(&self) -> Vec<Connection<T>> {
self.connections.values().cloned().collect()
}
pub fn set_accepting_connections(&mut self, accepting: bool) {
self.accepting_connections = accepting;
}
}
/// Wrap UdpSocket with some helper methods

4
src/lib.rs
View File

@ -238,6 +238,10 @@ impl<T: std::fmt::Debug + Clone + Serialize + DeserializeOwned + Send + Sync + '
self.connection_mgr.connections()
}
pub fn set_accepting_connections(&mut self, accepting: bool) {
self.connection_mgr.set_accepting_connections(accepting);
}
pub fn statistics(&self) -> NetStats {
let conns = self.connection_mgr.connections();
let all_stats = conns.iter().map(|c| c.statistics());