Attached queue to client.
This commit is contained in:
12
src/queue.rs
12
src/queue.rs
@ -1,10 +1,7 @@
|
||||
use crate::{client::Request, field::Field};
|
||||
use std::{
|
||||
collections::HashMap,
|
||||
sync::{
|
||||
mpsc::Sender,
|
||||
Arc, RwLock,
|
||||
},
|
||||
sync::{mpsc::Sender, Arc, RwLock},
|
||||
};
|
||||
use uuid::Uuid;
|
||||
|
||||
@ -12,6 +9,7 @@ use uuid::Uuid;
|
||||
pub enum MsgType {
|
||||
ClientRequest,
|
||||
NoOp,
|
||||
SessionValidate,
|
||||
}
|
||||
|
||||
#[derive(Clone)]
|
||||
@ -137,12 +135,12 @@ impl Queue {
|
||||
}
|
||||
}
|
||||
|
||||
fn add(&self, tx: Sender<Message>) {
|
||||
pub fn add(&self, tx: Sender<Message>) {
|
||||
let mut store = self.store.write().unwrap();
|
||||
store.push(tx);
|
||||
}
|
||||
|
||||
fn send(&self, msg: Message) {
|
||||
pub fn send(&self, msg: Message) {
|
||||
let store = self.store.read().unwrap();
|
||||
for sender in store.iter() {
|
||||
sender.send(msg.clone()).unwrap();
|
||||
@ -152,8 +150,8 @@ impl Queue {
|
||||
|
||||
#[cfg(test)]
|
||||
mod serviceredistries {
|
||||
use std::sync::mpsc::channel;
|
||||
use super::*;
|
||||
use std::sync::mpsc::channel;
|
||||
|
||||
#[test]
|
||||
fn create_queue() {
|
||||
|
Reference in New Issue
Block a user