88 lines
2.0 KiB
Rust
88 lines
2.0 KiB
Rust
use ewebsock::{WsMessage, WsReceiver, WsSender};
|
|
|
|
use crate::{ClientMessage, Filter, Result};
|
|
use log::info;
|
|
use std::fmt;
|
|
use std::hash::{Hash, Hasher};
|
|
|
|
pub mod message;
|
|
pub mod pool;
|
|
|
|
#[derive(Debug)]
|
|
pub enum RelayStatus {
|
|
Connected,
|
|
Connecting,
|
|
Disconnected,
|
|
}
|
|
|
|
pub struct Relay {
|
|
pub url: String,
|
|
pub status: RelayStatus,
|
|
pub sender: WsSender,
|
|
pub receiver: WsReceiver,
|
|
}
|
|
|
|
impl fmt::Debug for Relay {
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
|
f.debug_struct("Relay")
|
|
.field("url", &self.url)
|
|
.field("status", &self.status)
|
|
.finish()
|
|
}
|
|
}
|
|
|
|
impl Hash for Relay {
|
|
fn hash<H: Hasher>(&self, state: &mut H) {
|
|
// Hashes the Relay by hashing the URL
|
|
self.url.hash(state);
|
|
}
|
|
}
|
|
|
|
impl PartialEq for Relay {
|
|
fn eq(&self, other: &Self) -> bool {
|
|
self.url == other.url
|
|
}
|
|
}
|
|
|
|
impl Eq for Relay {}
|
|
|
|
impl Relay {
|
|
pub fn new(url: String, wakeup: impl Fn() + Send + Sync + 'static) -> Result<Self> {
|
|
let status = RelayStatus::Connecting;
|
|
let (sender, receiver) = ewebsock::connect_with_wakeup(&url, wakeup)?;
|
|
|
|
Ok(Self {
|
|
url,
|
|
sender,
|
|
receiver,
|
|
status,
|
|
})
|
|
}
|
|
|
|
pub fn send(&mut self, msg: &ClientMessage) {
|
|
let txt = WsMessage::Text(msg.to_json());
|
|
self.sender.send(txt);
|
|
}
|
|
|
|
pub fn connect(&mut self, wakeup: impl Fn() + Send + Sync + 'static) -> Result<()> {
|
|
let (sender, receiver) = ewebsock::connect_with_wakeup(&self.url, wakeup)?;
|
|
self.status = RelayStatus::Connecting;
|
|
self.sender = sender;
|
|
self.receiver = receiver;
|
|
Ok(())
|
|
}
|
|
|
|
pub fn ping(&mut self) {
|
|
let msg = WsMessage::Ping(vec![]);
|
|
self.sender.send(msg);
|
|
}
|
|
|
|
pub fn subscribe(&mut self, subid: String, filters: Vec<Filter>) {
|
|
info!(
|
|
"sending '{}' subscription to relay pool: {:?}",
|
|
subid, filters
|
|
);
|
|
self.send(&ClientMessage::req(subid, filters));
|
|
}
|
|
}
|