chik-sdk-test 0.20.0

A wallet simulator and related tooling for testing Chik wallet code.
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

use std::{collections::HashMap, net::SocketAddr, sync::Arc};

use futures_channel::mpsc::UnboundedSender;
use tokio::sync::Mutex;
use tokio_tungstenite::tungstenite::Message;

pub(crate) type Ws = UnboundedSender<Message>;
type Peers = HashMap<SocketAddr, Ws>;

#[derive(Default, Clone)]
pub(crate) struct PeerMap(Arc<Mutex<Peers>>);

impl PeerMap {
    pub(crate) async fn insert(&self, addr: SocketAddr, ws: Ws) {
        self.0.lock().await.insert(addr, ws);
    }

    pub(crate) async fn remove(&self, addr: SocketAddr) {
        self.0.lock().await.remove(&addr);
    }

    pub(crate) async fn peers(&self) -> Vec<(SocketAddr, Ws)> {
        self.0
            .lock()
            .await
            .iter()
            .map(|(addr, ws)| (*addr, ws.clone()))
            .collect()
    }
}