2018-08-09 12:03:34 -07:00
|
|
|
//! The `retransmit_stage` retransmits blobs between validators
|
2018-06-13 21:52:23 -07:00
|
|
|
|
2018-08-09 11:54:23 -07:00
|
|
|
use broadcaster;
|
2018-06-13 21:52:23 -07:00
|
|
|
use crdt::Crdt;
|
2018-06-27 11:33:56 -07:00
|
|
|
use packet::BlobRecycler;
|
2018-07-03 21:14:08 -07:00
|
|
|
use service::Service;
|
2018-06-13 21:52:23 -07:00
|
|
|
use std::net::UdpSocket;
|
|
|
|
use std::sync::mpsc::channel;
|
|
|
|
use std::sync::{Arc, RwLock};
|
2018-07-03 21:14:08 -07:00
|
|
|
use std::thread::{self, JoinHandle};
|
2018-07-30 16:56:01 -07:00
|
|
|
use streamer::{self, BlobReceiver, SharedWindow};
|
2018-06-13 21:52:23 -07:00
|
|
|
|
2018-08-09 12:03:34 -07:00
|
|
|
pub struct RetransmitStage {
|
2018-07-03 21:14:08 -07:00
|
|
|
thread_hdls: Vec<JoinHandle<()>>,
|
2018-06-13 21:52:23 -07:00
|
|
|
}
|
|
|
|
|
2018-08-09 12:03:34 -07:00
|
|
|
impl RetransmitStage {
|
2018-06-13 21:52:23 -07:00
|
|
|
pub fn new(
|
2018-07-11 13:40:46 -07:00
|
|
|
crdt: &Arc<RwLock<Crdt>>,
|
2018-07-30 16:56:01 -07:00
|
|
|
window: SharedWindow,
|
2018-06-27 12:35:58 -07:00
|
|
|
entry_height: u64,
|
2018-06-13 21:52:23 -07:00
|
|
|
retransmit_socket: UdpSocket,
|
2018-07-11 13:40:46 -07:00
|
|
|
blob_recycler: &BlobRecycler,
|
2018-06-27 11:33:56 -07:00
|
|
|
fetch_stage_receiver: BlobReceiver,
|
2018-07-02 14:45:16 -07:00
|
|
|
) -> (Self, BlobReceiver) {
|
2018-06-13 21:52:23 -07:00
|
|
|
let (retransmit_sender, retransmit_receiver) = channel();
|
|
|
|
|
2018-08-09 11:54:23 -07:00
|
|
|
let t_retransmit = broadcaster::retransmitter(
|
2018-06-13 21:52:23 -07:00
|
|
|
retransmit_socket,
|
|
|
|
crdt.clone(),
|
|
|
|
blob_recycler.clone(),
|
|
|
|
retransmit_receiver,
|
|
|
|
);
|
|
|
|
let (blob_sender, blob_receiver) = channel();
|
|
|
|
let t_window = streamer::window(
|
|
|
|
crdt.clone(),
|
|
|
|
window,
|
2018-06-27 12:35:58 -07:00
|
|
|
entry_height,
|
2018-06-13 21:52:23 -07:00
|
|
|
blob_recycler.clone(),
|
|
|
|
fetch_stage_receiver,
|
|
|
|
blob_sender,
|
|
|
|
retransmit_sender,
|
|
|
|
);
|
|
|
|
let thread_hdls = vec![t_retransmit, t_window];
|
|
|
|
|
2018-08-09 12:03:34 -07:00
|
|
|
(RetransmitStage { thread_hdls }, blob_receiver)
|
2018-06-13 21:52:23 -07:00
|
|
|
}
|
|
|
|
}
|
2018-07-03 21:14:08 -07:00
|
|
|
|
2018-08-09 12:03:34 -07:00
|
|
|
impl Service for RetransmitStage {
|
2018-07-03 21:14:08 -07:00
|
|
|
fn thread_hdls(self) -> Vec<JoinHandle<()>> {
|
|
|
|
self.thread_hdls
|
|
|
|
}
|
|
|
|
|
|
|
|
fn join(self) -> thread::Result<()> {
|
|
|
|
for thread_hdl in self.thread_hdls() {
|
|
|
|
thread_hdl.join()?;
|
|
|
|
}
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
}
|