aljaz/notedeck
1
0
Fork 0
mirror of https://github.com/aljazceru/notedeck.git synced 2025-12-24 03:24:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

integrate SubsDebug into RelayPool

Browse Source 
Signed-off-by: kernelkind <kernelkind@gmail.com>
This commit is contained in:
kernelkind
2024-12-29 20:45:38 -05:00
parent 612c89118f
commit 54dfa0c945
1 changed files with 35 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
crates/enostr/src/relay/pool.rs
View File

@@ -13,6 +13,8 @@ use ewebsock::{WsEvent, WsMessage};
#[cfg(not(target_arch = "wasm32"))]
use tracing::{debug, error};
use super::subs_debug::SubsDebug;
#[derive(Debug)]
pub struct PoolEvent<'a> {
pub relay: &'a str,
@@ -58,6 +60,7 @@ impl PoolRelay {
pub struct RelayPool {
pub relays: Vec<PoolRelay>,
pub ping_rate: Duration,
pub debug: Option<SubsDebug>,
}
impl Default for RelayPool {
@@ -72,9 +75,14 @@ impl RelayPool {
RelayPool {
relays: vec![],
ping_rate: Duration::from_secs(25),
debug: None,
}
}
pub fn use_debug(&mut self) {
self.debug = Some(SubsDebug::default());
}
pub fn ping_rate(&mut self, duration: Duration) -> &mut Self {
self.ping_rate = duration;
self
@@ -99,18 +107,32 @@ impl RelayPool {
pub fn send(&mut self, cmd: &ClientMessage) {
for relay in &mut self.relays {
if let Some(debug) = &mut self.debug {
debug.send_cmd(relay.relay.url.clone(), cmd);
}
relay.relay.send(cmd);
}
}
pub fn unsubscribe(&mut self, subid: String) {
for relay in &mut self.relays {
relay.relay.send(&ClientMessage::close(subid.clone()));
let cmd = &ClientMessage::close(subid.clone());
if let Some(debug) = &mut self.debug {
debug.send_cmd(relay.relay.url.clone(), cmd);
}
relay.relay.send(cmd);
}
}
pub fn subscribe(&mut self, subid: String, filter: Vec<Filter>) {
for relay in &mut self.relays {
if let Some(debug) = &mut self.debug {
debug.send_cmd(
relay.relay.url.clone(),
&ClientMessage::req(subid.clone(), filter.clone()),
);
}
relay.relay.subscribe(subid.clone(), filter.clone());
}
}
@@ -164,6 +186,9 @@ impl RelayPool {
for relay in &mut self.relays {
let relay = &mut relay.relay;
if relay.url == relay_url {
if let Some(debug) = &mut self.debug {
debug.send_cmd(relay.url.clone(), cmd);
}
relay.send(cmd);
return;
}
@@ -242,10 +267,17 @@ impl RelayPool {
}
}
}
return Some(PoolEvent {
if let Some(debug) = &mut self.debug {
debug.receive_cmd(relay.url.clone(), (&event).into());
}
let pool_event = PoolEvent {
event,
relay: &relay.url,
});
};
return Some(pool_event);
}
}