This commit is contained in:
parent
55ea1fb8a2
commit
07d9efb4f9
@ -7,7 +7,7 @@ use std::{
|
|||||||
mpsc::{channel, Receiver, Sender},
|
mpsc::{channel, Receiver, Sender},
|
||||||
Arc, RwLock,
|
Arc, RwLock,
|
||||||
},
|
},
|
||||||
thread::spawn,
|
thread::{sleep, spawn},
|
||||||
time::Duration,
|
time::Duration,
|
||||||
};
|
};
|
||||||
use uuid::Uuid;
|
use uuid::Uuid;
|
||||||
@ -5736,3 +5736,75 @@ mod createdocs {
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
struct Clock {
|
||||||
|
queue: Queue,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Clock {
|
||||||
|
fn new(queue: Queue) -> Self {
|
||||||
|
Self { queue: queue }
|
||||||
|
}
|
||||||
|
|
||||||
|
fn start(mut queue: Queue) {
|
||||||
|
let clock = Clock::new(queue.clone());
|
||||||
|
let (tx, rx) = channel();
|
||||||
|
let id = queue.add_sender(tx);
|
||||||
|
let reg_msg = Register::new(id, RegMsg::AddDocName(Name::english("clock")));
|
||||||
|
let msg = Message::new(NameType::None, reg_msg.clone());
|
||||||
|
queue.send(msg).unwrap();
|
||||||
|
rx.recv().unwrap();
|
||||||
|
spawn(move || {
|
||||||
|
clock.listen();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
fn listen(&self) {
|
||||||
|
loop {
|
||||||
|
self.queue
|
||||||
|
.send(Message::new(Name::english("clock"), RecordIter::new()));
|
||||||
|
sleep(Duration::from_secs(1));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod clocks {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
static TIMEOUT: Duration = Duration::from_millis(1500);
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn does_clock_send_reply_every_second() {
|
||||||
|
let mut queue = Queue::new();
|
||||||
|
let (tx, rx) = channel();
|
||||||
|
let id = queue.add_sender(tx);
|
||||||
|
let request = Register::new(
|
||||||
|
id,
|
||||||
|
RegMsg::AddRoute(Path::new(Include::All, Include::All, Include::All)),
|
||||||
|
);
|
||||||
|
queue.send(Message::new(NameType::None, request)).unwrap();
|
||||||
|
rx.recv_timeout(TIMEOUT).unwrap();
|
||||||
|
let mut holder: Vec<Message> = Vec::new();
|
||||||
|
let start = Utc::now();
|
||||||
|
Clock::start(queue.clone());
|
||||||
|
while holder.len() < 2 {
|
||||||
|
holder.push(rx.recv_timeout(TIMEOUT).unwrap());
|
||||||
|
}
|
||||||
|
let end = Utc::now();
|
||||||
|
assert!((end - start) > chrono::TimeDelta::seconds(1));
|
||||||
|
assert!((end - start) < chrono::TimeDelta::seconds(2));
|
||||||
|
for msg in holder.iter() {
|
||||||
|
let name_id = msg.get_document_id();
|
||||||
|
match name_id {
|
||||||
|
NameType::Name(data) => assert_eq!(data, &Name::english("clock")),
|
||||||
|
_ => unreachable!("got {:?}, should have been clock", name_id),
|
||||||
|
}
|
||||||
|
let action = msg.get_action();
|
||||||
|
match action {
|
||||||
|
MsgAction::Records(result) => assert_eq!(result.len(), 0),
|
||||||
|
_ => unreachable!("got {:?}, should have been empty record", action),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user