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) -> &Self
19    where
20        K: ToString,
21    {
22        let key_string: String = key.to_string();
23        let broadcast: Broadcast<T> = Broadcast::new(capacity);
24        self.get_broadcast().insert(key_string, broadcast);
25        self
26    }
27
28    pub fn receiver_count(&self, key: &str) -> OptionReceiverCount {
29        self.get_broadcast()
30            .get(key)
31            .map(|receiver| receiver.receiver_count())
32    }
33
34    pub fn subscribe(&self, key: &str) -> OptionBroadcastMapReceiver<T> {
35        self.get_broadcast()
36            .get(key)
37            .map(|receiver| receiver.subscribe())
38    }
39
40    pub fn subscribe_unwrap_or_insert(&self, key: &str) -> BroadcastMapReceiver<T> {
41        match self.get_broadcast().get(key) {
42            Some(sender) => sender.subscribe(),
43            None => {
44                self.insert(key, DEFAULT_BROADCAST_SENDER_CAPACITY);
45                self.subscribe_unwrap_or_insert(key)
46            }
47        }
48    }
49
50    pub fn send(&self, key: &str, data: T) -> BroadcastMapSendResult<T> {
51        match self.get_broadcast().get(key) {
52            Some(sender) => sender.send(data).map(|result| Some(result)),
53            None => Ok(None),
54        }
55    }
56}
hyperlane_broadcast/broadcast_map/impl.rs

hyperlane_broadcast/broadcast_map/
impl.rs
