Attached queue to client.
This commit is contained in:
@ -194,21 +194,23 @@ mod clientlinks {
|
||||
}
|
||||
|
||||
pub struct Client {
|
||||
queue: Queue,
|
||||
registry: ClientRegistry,
|
||||
rx: Receiver<Message>,
|
||||
}
|
||||
|
||||
impl Client {
|
||||
fn new(rx: Receiver<Message>) -> Self {
|
||||
fn new(rx: Receiver<Message>, queue: Queue) -> Self {
|
||||
Self {
|
||||
queue: queue,
|
||||
registry: ClientRegistry::new(),
|
||||
rx: rx,
|
||||
}
|
||||
}
|
||||
|
||||
pub fn start(_queue: Queue) -> ClientLink {
|
||||
pub fn start(queue: Queue) -> ClientLink {
|
||||
let (tx, rx) = channel();
|
||||
let mut client = Client::new(rx);
|
||||
let mut client = Client::new(rx, queue);
|
||||
let link = ClientLink::new(tx, client.get_registry());
|
||||
spawn(move || {
|
||||
client.listen();
|
||||
@ -219,6 +221,7 @@ impl Client {
|
||||
fn listen(&mut self) {
|
||||
loop {
|
||||
let msg = self.rx.recv().unwrap();
|
||||
self.queue.send(Message::new(MsgType::SessionValidate));
|
||||
let id = msg.get_data().get("tx_id").unwrap().to_uuid().unwrap();
|
||||
let reply = Reply {};
|
||||
self.registry.send(&id, reply);
|
||||
@ -234,12 +237,22 @@ impl Client {
|
||||
mod clients {
|
||||
use super::*;
|
||||
use requests::create_request;
|
||||
use std::time::Duration;
|
||||
|
||||
static TIMEOUT: Duration = Duration::from_millis(500);
|
||||
|
||||
#[test]
|
||||
fn start_client() {
|
||||
let queue = Queue::new();
|
||||
let (tx, rx) = channel();
|
||||
let mut queue = Queue::new();
|
||||
queue.add(tx);
|
||||
let mut link = Client::start(queue.clone());
|
||||
let req = create_request();
|
||||
link.send(req);
|
||||
let sess = rx.recv_timeout(TIMEOUT).unwrap();
|
||||
match sess.get_class() {
|
||||
MsgType::SessionValidate => {},
|
||||
_ => unreachable!("should request session validation"),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
Reference in New Issue
Block a user