docs: add rustdoc comments

This commit is contained in:
Greg Heartsfield
2021-12-11 21:43:41 -06:00
parent 04850506a8
commit ca0f01c94b
8 changed files with 157 additions and 100 deletions
+18 -9
View File
@@ -1,3 +1,4 @@
//! Client connection state
use crate::close::Close;
use crate::error::Result;
use crate::event::Event;
@@ -6,34 +7,38 @@ use log::*;
use std::collections::HashMap;
use uuid::Uuid;
// subscription identifiers must be reasonably sized.
/// A subscription identifier has a maximum length
const MAX_SUBSCRIPTION_ID_LEN: usize = 256;
// state for a client connection
/// State for a client connection
pub struct ClientConn {
/// Unique client identifier generated at connection time
client_id: Uuid,
// current set of subscriptions
/// The current set of active client subscriptions
subscriptions: HashMap<String, Subscription>,
// websocket
//stream: WebSocketStream<TcpStream>,
/// Per-connection maximum concurrent subscriptions
max_subs: usize,
}
impl ClientConn {
/// Create a new, empty connection state.
pub fn new() -> Self {
let client_id = Uuid::new_v4();
ClientConn {
client_id: client_id,
client_id,
subscriptions: HashMap::new(),
max_subs: 128,
}
}
/// Get a short prefix of the client's unique identifier, suitable
/// for logging.
pub fn get_client_prefix(&self) -> String {
self.client_id.to_string().chars().take(8).collect()
}
// return the first subscription that matches the event.
/// Find the first subscription identifier that matches the event,
/// if any do.
pub fn get_matching_subscription(&self, e: &Event) -> Option<&str> {
for (id, sub) in self.subscriptions.iter() {
if sub.interested_in_event(e) {
@@ -43,9 +48,12 @@ impl ClientConn {
None
}
/// Add a new subscription for this connection.
pub fn subscribe(&mut self, s: Subscription) -> Result<()> {
let k = s.get_id();
let sub_id_len = k.len();
// prevent arbitrarily long subscription identifiers from
// being used.
if sub_id_len > MAX_SUBSCRIPTION_ID_LEN {
info!("Dropping subscription with huge ({}) length", sub_id_len);
return Ok(());
@@ -54,7 +62,7 @@ impl ClientConn {
if self.subscriptions.contains_key(&k) {
self.subscriptions.remove(&k);
self.subscriptions.insert(k, s);
info!("Replaced existing subscription");
debug!("Replaced existing subscription");
return Ok(());
}
@@ -72,9 +80,10 @@ impl ClientConn {
return Ok(());
}
/// Remove the subscription for this connection.
pub fn unsubscribe(&mut self, c: Close) {
// TODO: return notice if subscription did not exist.
self.subscriptions.remove(&c.get_id());
self.subscriptions.remove(&c.id);
info!(
"Removed subscription, currently have {} active subs",
self.subscriptions.len()