nostr-rs-relay/src/conn.rs

107 lines
3.1 KiB
Rust
Raw Normal View History

2021-12-11 22:43:41 -05:00
//! Client connection state
use crate::close::Close;
use crate::error::Error;
use crate::error::Result;
use crate::event::Event;
use crate::subscription::Subscription;
2022-09-24 10:01:09 -04:00
use log::{debug, info};
use std::collections::HashMap;
use uuid::Uuid;
2021-12-11 22:43:41 -05:00
/// A subscription identifier has a maximum length
const MAX_SUBSCRIPTION_ID_LEN: usize = 256;
2021-12-11 22:43:41 -05:00
/// State for a client connection
pub struct ClientConn {
2021-12-11 22:43:41 -05:00
/// Unique client identifier generated at connection time
client_id: Uuid,
2021-12-11 22:43:41 -05:00
/// The current set of active client subscriptions
subscriptions: HashMap<String, Subscription>,
2021-12-11 22:43:41 -05:00
/// Per-connection maximum concurrent subscriptions
max_subs: usize,
}
2021-12-11 22:56:52 -05:00
impl Default for ClientConn {
fn default() -> Self {
Self::new()
}
}
impl ClientConn {
2021-12-11 22:43:41 -05:00
/// Create a new, empty connection state.
2022-09-24 10:01:09 -04:00
#[must_use]
pub fn new() -> Self {
let client_id = Uuid::new_v4();
ClientConn {
2021-12-11 22:43:41 -05:00
client_id,
subscriptions: HashMap::new(),
max_subs: 32,
}
}
2021-12-11 22:43:41 -05:00
/// Get a short prefix of the client's unique identifier, suitable
/// for logging.
2022-09-24 10:01:09 -04:00
#[must_use]
pub fn get_client_prefix(&self) -> String {
self.client_id.to_string().chars().take(8).collect()
}
/// Find all matching subscriptions.
2022-09-24 10:01:09 -04:00
#[must_use]
pub fn get_matching_subscriptions(&self, e: &Event) -> Vec<&str> {
let mut v: Vec<&str> = vec![];
2022-09-24 10:01:09 -04:00
for (id, sub) in &self.subscriptions {
if sub.interested_in_event(e) {
v.push(id);
}
}
v
}
2021-12-11 22:43:41 -05:00
/// 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();
2021-12-11 22:43:41 -05:00
// prevent arbitrarily long subscription identifiers from
// being used.
if sub_id_len > MAX_SUBSCRIPTION_ID_LEN {
2021-12-12 11:03:28 -05:00
info!(
"ignoring sub request with excessive length: ({})",
sub_id_len
);
return Err(Error::SubIdMaxLengthError);
}
// check if an existing subscription exists, and replace if so
if self.subscriptions.contains_key(&k) {
self.subscriptions.remove(&k);
self.subscriptions.insert(k, s);
2021-12-12 11:03:28 -05:00
debug!("replaced existing subscription");
return Ok(());
}
// check if there is room for another subscription.
if self.subscriptions.len() >= self.max_subs {
return Err(Error::SubMaxExceededError);
}
// add subscription
self.subscriptions.insert(k, s);
2021-12-12 11:03:28 -05:00
debug!(
"registered new subscription, currently have {} active subs",
self.subscriptions.len()
);
2021-12-11 22:56:52 -05:00
Ok(())
}
2021-12-11 22:43:41 -05:00
/// Remove the subscription for this connection.
pub fn unsubscribe(&mut self, c: Close) {
// TODO: return notice if subscription did not exist.
2021-12-11 22:43:41 -05:00
self.subscriptions.remove(&c.id);
2021-12-12 11:03:28 -05:00
debug!(
"removed subscription, currently have {} active subs (cid={})",
self.subscriptions.len(),
self.client_id
);
}
}