hyperlane_broadcast/broadcast_map/
impl.rs

1use crate::*;
2
3impl<T: Clone + Debug> BroadcastMapTrait for T {}
4
5impl<T: BroadcastMapTrait> Default for BroadcastMap<T> {
6    fn default() -> Self {
7        BroadcastMap {
8            broadcast: DashMap::new(),
9        }
10    }
11}
12
13impl<T: BroadcastMapTrait> BroadcastMap<T> {
14    pub fn new() -> Self {
15        Self::default()
16    }
17
18    pub fn insert<K>(&self, key: K, capacity: usize) -> OptionBroadcast<T>
19    where
20        K: ToString,
21    {
22        let key_string: String = key.to_string();
23        let broadcast: Broadcast<T> = Broadcast::new(capacity);
24        self.broadcast.insert(key_string, broadcast)
25    }
26
27    pub fn receiver_count(&self, key: &str) -> OptionReceiverCount {
28        self.broadcast
29            .get(key)
30            .map(|receiver| receiver.receiver_count())
31    }
32
33    pub fn subscribe(&self, key: &str) -> OptionBroadcastMapReceiver<T> {
34        self.broadcast.get(key).map(|receiver| receiver.subscribe())
35    }
36
37    pub fn subscribe_unwrap_or_insert(&self, key: &str) -> BroadcastMapReceiver<T> {
38        match self.broadcast.get(key) {
39            Some(sender) => sender.subscribe(),
40            None => {
41                self.insert(key, DEFAULT_BROADCAST_SENDER_CAPACITY);
42                self.subscribe_unwrap_or_insert(key)
43            }
44        }
45    }
46
47    pub fn send(&self, key: &str, data: T) -> BroadcastMapSendResult<T> {
48        match self.broadcast.get(key) {
49            Some(sender) => sender.send(data).map(|result| Some(result)),
50            None => Ok(None),
51        }
52    }
53}
hyperlane_broadcast/broadcast_map/impl.rs

hyperlane_broadcast/broadcast_map/
impl.rs
