2018-06-07 15:06:32 -07:00
|
|
|
//! The `ncp` module implements the network control plane.
|
2018-06-06 16:36:54 -07:00
|
|
|
|
2018-06-27 11:33:56 -07:00
|
|
|
use crdt::Crdt;
|
|
|
|
use packet::{BlobRecycler, SharedBlob};
|
2018-05-27 18:21:39 -07:00
|
|
|
use result::Result;
|
|
|
|
use std::net::UdpSocket;
|
|
|
|
use std::sync::atomic::AtomicBool;
|
|
|
|
use std::sync::mpsc::channel;
|
|
|
|
use std::sync::{Arc, RwLock};
|
|
|
|
use std::thread::JoinHandle;
|
|
|
|
use streamer;
|
|
|
|
|
2018-06-07 15:06:32 -07:00
|
|
|
pub struct Ncp {
|
2018-05-27 18:21:39 -07:00
|
|
|
pub thread_hdls: Vec<JoinHandle<()>>,
|
|
|
|
}
|
|
|
|
|
2018-06-07 15:06:32 -07:00
|
|
|
impl Ncp {
|
2018-05-27 18:21:39 -07:00
|
|
|
pub fn new(
|
2018-06-27 11:33:56 -07:00
|
|
|
crdt: Arc<RwLock<Crdt>>,
|
|
|
|
window: Arc<RwLock<Vec<Option<SharedBlob>>>>,
|
2018-05-27 18:21:39 -07:00
|
|
|
gossip_listen_socket: UdpSocket,
|
|
|
|
gossip_send_socket: UdpSocket,
|
|
|
|
exit: Arc<AtomicBool>,
|
2018-06-07 15:06:32 -07:00
|
|
|
) -> Result<Ncp> {
|
2018-06-27 11:33:56 -07:00
|
|
|
let blob_recycler = BlobRecycler::default();
|
2018-05-27 18:21:39 -07:00
|
|
|
let (request_sender, request_receiver) = channel();
|
|
|
|
trace!(
|
2018-06-07 15:06:32 -07:00
|
|
|
"Ncp: id: {:?}, listening on: {:?}",
|
2018-05-27 18:21:39 -07:00
|
|
|
&crdt.read().unwrap().me[..4],
|
|
|
|
gossip_listen_socket.local_addr().unwrap()
|
|
|
|
);
|
|
|
|
let t_receiver = streamer::blob_receiver(
|
|
|
|
exit.clone(),
|
|
|
|
blob_recycler.clone(),
|
|
|
|
gossip_listen_socket,
|
|
|
|
request_sender,
|
|
|
|
)?;
|
|
|
|
let (response_sender, response_receiver) = channel();
|
|
|
|
let t_responder = streamer::responder(
|
|
|
|
gossip_send_socket,
|
|
|
|
exit.clone(),
|
|
|
|
blob_recycler.clone(),
|
|
|
|
response_receiver,
|
|
|
|
);
|
2018-06-27 11:33:56 -07:00
|
|
|
let t_listen = Crdt::listen(
|
2018-05-27 18:21:39 -07:00
|
|
|
crdt.clone(),
|
|
|
|
window,
|
|
|
|
blob_recycler.clone(),
|
|
|
|
request_receiver,
|
|
|
|
response_sender.clone(),
|
|
|
|
exit.clone(),
|
|
|
|
);
|
2018-06-27 11:33:56 -07:00
|
|
|
let t_gossip = Crdt::gossip(crdt.clone(), blob_recycler, response_sender, exit);
|
2018-05-27 18:21:39 -07:00
|
|
|
let thread_hdls = vec![t_receiver, t_responder, t_listen, t_gossip];
|
2018-06-07 15:06:32 -07:00
|
|
|
Ok(Ncp { thread_hdls })
|
2018-05-27 18:21:39 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
mod tests {
|
|
|
|
use crdt::{Crdt, TestNode};
|
2018-06-07 15:06:32 -07:00
|
|
|
use ncp::Ncp;
|
2018-05-27 18:21:39 -07:00
|
|
|
use std::sync::atomic::{AtomicBool, Ordering};
|
|
|
|
use std::sync::{Arc, RwLock};
|
|
|
|
|
2018-05-30 09:50:28 -07:00
|
|
|
#[test]
|
|
|
|
// test that stage will exit when flag is set
|
|
|
|
fn test_exit() {
|
|
|
|
let exit = Arc::new(AtomicBool::new(false));
|
2018-05-27 18:21:39 -07:00
|
|
|
let tn = TestNode::new();
|
|
|
|
let crdt = Crdt::new(tn.data.clone());
|
|
|
|
let c = Arc::new(RwLock::new(crdt));
|
|
|
|
let w = Arc::new(RwLock::new(vec![]));
|
2018-06-07 15:06:32 -07:00
|
|
|
let d = Ncp::new(
|
2018-05-27 18:21:39 -07:00
|
|
|
c.clone(),
|
|
|
|
w,
|
|
|
|
tn.sockets.gossip,
|
|
|
|
tn.sockets.gossip_send,
|
2018-05-30 09:50:28 -07:00
|
|
|
exit.clone(),
|
2018-05-27 18:21:39 -07:00
|
|
|
).unwrap();
|
|
|
|
exit.store(true, Ordering::Relaxed);
|
2018-05-30 09:50:28 -07:00
|
|
|
for t in d.thread_hdls {
|
|
|
|
t.join().expect("thread join");
|
2018-05-27 18:21:39 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|