ankurah-websocket-client 0.9.0

Ankurah WebSocket Client - Native WebSocket client for Ankurah
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

use ankurah_core::connector::{PeerSender, SendError};
use ankurah_proto as proto;
use async_trait::async_trait;
use tokio::sync::mpsc;
use tracing::{debug, warn};

/// PeerSender implementation for websocket connections
#[derive(Clone)]
pub struct WebsocketPeerSender {
    tx: mpsc::UnboundedSender<proto::NodeMessage>,
    recipient_node_id: proto::EntityId,
}

impl WebsocketPeerSender {
    pub fn new(recipient_node_id: proto::EntityId) -> (Self, mpsc::UnboundedReceiver<proto::NodeMessage>) {
        let (tx, rx) = mpsc::unbounded_channel();
        (Self { tx, recipient_node_id }, rx)
    }
}

#[async_trait]
impl PeerSender for WebsocketPeerSender {
    fn send_message(&self, message: proto::NodeMessage) -> Result<(), SendError> {
        debug!("Queuing message for peer {}", self.recipient_node_id);

        self.tx.send(message).map_err(|_| {
            warn!("Failed to send message to peer {} - channel closed", self.recipient_node_id);
            SendError::ConnectionClosed
        })
    }

    fn recipient_node_id(&self) -> proto::EntityId { self.recipient_node_id }

    fn cloned(&self) -> Box<dyn PeerSender> { Box::new(self.clone()) }
}