Added routing to queue.

src/queue.rs

@@ -5,11 +5,12 @@ use std::{
 };
 use uuid::Uuid;
 
-#[derive(Clone)]
+#[derive(Clone, Debug, Eq, Hash, PartialEq)]
 pub enum MsgType {
     ClientRequest,
     NoOp,
     SessionValidate,
+    Session,
 }
 
 #[derive(Clone)]
@@ -125,43 +126,109 @@ mod messages {
 
 #[derive(Clone)]
 pub struct Queue {
-    store: Arc<RwLock<Vec<Sender<Message>>>>,
+    store: Arc<RwLock<HashMap<MsgType, Vec<Sender<Message>>>>>,
 }
 
 impl Queue {
     pub fn new() -> Self {
         Self {
-            store: Arc::new(RwLock::new(Vec::new())),
+            store: Arc::new(RwLock::new(HashMap::new())),
         }
     }
 
-    pub fn add(&self, tx: Sender<Message>) {
+    pub fn add(&self, tx: Sender<Message>, msg_types: Vec<MsgType>) {
         let mut store = self.store.write().unwrap();
-        store.push(tx);
+        for msg_type in msg_types.into_iter() {
+            if !store.contains_key(&msg_type) {
+                store.insert(msg_type.clone(), Vec::new());
+            }
+            let senders = store.get_mut(&msg_type).unwrap();
+            senders.push(tx.clone());
+        }
     }
 
     pub fn send(&self, msg: Message) {
         let store = self.store.read().unwrap();
-        for sender in store.iter() {
+        let senders = store.get(&msg.get_class()).unwrap();
+        for sender in senders.into_iter() {
             sender.send(msg.clone()).unwrap();
         }
     }
 }
 
 #[cfg(test)]
-mod serviceredistries {
+mod queues {
     use super::*;
-    use std::sync::mpsc::channel;
+    use std::{
+        sync::mpsc::{channel, RecvTimeoutError},
+        time::Duration,
+    };
+
+    static TIMEOUT: Duration = Duration::from_millis(500);
 
     #[test]
     fn create_queue() {
         let queue = Queue::new();
         let (tx1, rx1) = channel();
         let (tx2, rx2) = channel();
-        queue.add(tx1);
-        queue.add(tx2);
+        queue.add(tx1, [MsgType::NoOp].to_vec());
+        queue.add(tx2, [MsgType::NoOp].to_vec());
         queue.send(Message::new(MsgType::NoOp));
         rx1.recv().unwrap();
         rx2.recv().unwrap();
     }
+
+    #[test]
+    fn messages_are_routed() {
+        let queue = Queue::new();
+        let (tx1, rx1) = channel();
+        let (tx2, rx2) = channel();
+        queue.add(tx1, [MsgType::SessionValidate].to_vec());
+        queue.add(tx2, [MsgType::Session].to_vec());
+        queue.send(Message::new(MsgType::SessionValidate));
+        let result = rx1.recv().unwrap();
+        match result.get_class() {
+            MsgType::SessionValidate => {}
+            _ => unreachable!(
+                "received {:?}, should have been session vvalidate",
+                result.get_class()
+            ),
+        }
+        match rx2.recv_timeout(TIMEOUT) {
+            Ok(_) => unreachable!("should not have received anything"),
+            Err(err) => match err {
+                RecvTimeoutError::Timeout => {}
+                _ => unreachable!("{:?}", err),
+            },
+        }
+        queue.send(Message::new(MsgType::Session));
+        let result = rx2.recv().unwrap();
+        match result.get_class() {
+            MsgType::Session => {}
+            _ => unreachable!(
+                "received {:?}, should have been session vvalidate",
+                result.get_class()
+            ),
+        }
+        match rx1.recv_timeout(TIMEOUT) {
+            Ok(_) => unreachable!("should not have received anything"),
+            Err(err) => match err {
+                RecvTimeoutError::Timeout => {}
+                _ => unreachable!("{:?}", err),
+            },
+        }
+    }
+
+    #[test]
+    fn assign_sender_multiple_message_types() {
+        let queue = Queue::new();
+        let (tx, rx) = channel();
+        queue.add(tx, [MsgType::Session, MsgType::SessionValidate].to_vec());
+        queue.send(Message::new(MsgType::SessionValidate));
+        let msg = rx.recv().unwrap();
+        assert_eq!(msg.get_class(), &MsgType::SessionValidate);
+        queue.send(Message::new(MsgType::Session));
+        let msg = rx.recv().unwrap();
+        assert_eq!(msg.get_class(), &MsgType::Session);
+    }
 }