moirai-core 0.6.0

Core abstractions and traits for the Moirai concurrency library
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

#![expect(
    clippy::unwrap_used,
    reason = "ratchet MOIRAI-UNWRAP-1: pre-existing debt"
)]

use crate::channel::{ChannelError, MpmcSender};
use std::collections::HashMap;
use std::hash::Hash;
use std::sync::{Arc, RwLock};

/// Router for message-based communication patterns
pub struct MessageRouter<K: Hash + Eq + Clone, V: Send + 'static> {
    /// Routes mapped by key
    routes: Arc<RwLock<HashMap<K, MpmcSender<V>>>>,
}

impl<K: Hash + Eq + Clone, V: Send + 'static> MessageRouter<K, V> {
    /// Create a new message router
    pub fn new() -> Self {
        Self {
            routes: Arc::new(RwLock::new(HashMap::new())),
        }
    }

    /// Register a route
    pub fn register(&self, key: K, sender: MpmcSender<V>) {
        let mut routes = self.routes.write().unwrap();
        routes.insert(key, sender);
    }

    /// Route a message to the appropriate channel
    pub fn route(&self, key: &K, message: V) -> Result<(), ChannelError> {
        let routes = self.routes.read().unwrap();

        if let Some(sender) = routes.get(key) {
            sender.try_send(message)
        } else {
            Err(ChannelError::Closed)
        }
    }

    /// Remove a route
    pub fn unregister(&self, key: &K) -> bool {
        let mut routes = self.routes.write().unwrap();
        routes.remove(key).is_some()
    }
}

impl<K: Hash + Eq + Clone, V: Send + 'static> Default for MessageRouter<K, V> {
    fn default() -> Self {
        Self::new()
    }
}