2018-05-12 09:53:25 -07:00
|
|
|
//! The `rpu` module implements the Request Processing Unit, a
|
2018-06-06 10:26:57 -07:00
|
|
|
//! 3-stage transaction processing pipeline in software. It listens
|
|
|
|
//! for `Request` messages from clients and replies with `Response`
|
|
|
|
//! messages.
|
2018-06-07 12:02:08 -07:00
|
|
|
//!
|
|
|
|
//! ```text
|
|
|
|
//! .------.
|
|
|
|
//! | Bank |
|
|
|
|
//! `---+--`
|
|
|
|
//! |
|
|
|
|
//! .------------------|-------------------.
|
|
|
|
//! | RPU | |
|
|
|
|
//! | v |
|
|
|
|
//! .---------. | .-------. .---------. .---------. | .---------.
|
|
|
|
//! | Alice |--->| | | | | +---->| Alice |
|
|
|
|
//! `---------` | | Fetch | | Request | | Respond | | `---------`
|
|
|
|
//! | | Stage |->| Stage |->| Stage | |
|
|
|
|
//! .---------. | | | | | | | | .---------.
|
|
|
|
//! | Bob |--->| | | | | +---->| Bob |
|
|
|
|
//! `---------` | `-------` `---------` `---------` | `---------`
|
|
|
|
//! | |
|
|
|
|
//! | |
|
|
|
|
//! `--------------------------------------`
|
|
|
|
//! ```
|
2018-03-29 11:20:54 -07:00
|
|
|
|
2018-05-14 14:33:11 -07:00
|
|
|
use bank::Bank;
|
2018-06-27 11:33:56 -07:00
|
|
|
use packet::{BlobRecycler, PacketRecycler};
|
2018-05-12 16:50:55 -07:00
|
|
|
use request_processor::RequestProcessor;
|
|
|
|
use request_stage::RequestStage;
|
2018-07-03 21:14:08 -07:00
|
|
|
use service::Service;
|
2018-05-10 14:30:18 -07:00
|
|
|
use std::net::UdpSocket;
|
2018-05-14 15:27:40 -07:00
|
|
|
use std::sync::atomic::AtomicBool;
|
|
|
|
use std::sync::mpsc::channel;
|
2018-05-29 20:20:28 -07:00
|
|
|
use std::sync::Arc;
|
2018-07-03 21:14:08 -07:00
|
|
|
use std::thread::{self, JoinHandle};
|
2018-03-26 21:03:26 -07:00
|
|
|
use streamer;
|
2018-02-28 09:07:54 -08:00
|
|
|
|
2018-05-12 09:53:25 -07:00
|
|
|
pub struct Rpu {
|
2018-07-03 21:14:08 -07:00
|
|
|
thread_hdls: Vec<JoinHandle<()>>,
|
2018-03-29 12:18:08 -07:00
|
|
|
}
|
|
|
|
|
2018-05-12 09:53:25 -07:00
|
|
|
impl Rpu {
|
2018-05-15 10:33:43 -07:00
|
|
|
pub fn new(
|
2018-07-11 20:10:25 -07:00
|
|
|
bank: &Arc<Bank>,
|
2018-05-15 09:27:18 -07:00
|
|
|
requests_socket: UdpSocket,
|
|
|
|
respond_socket: UdpSocket,
|
|
|
|
exit: Arc<AtomicBool>,
|
|
|
|
) -> Self {
|
2018-06-27 11:33:56 -07:00
|
|
|
let packet_recycler = PacketRecycler::default();
|
2018-04-02 19:32:58 -07:00
|
|
|
let (packet_sender, packet_receiver) = channel();
|
2018-05-11 15:41:35 -07:00
|
|
|
let t_receiver = streamer::receiver(
|
|
|
|
requests_socket,
|
2018-07-05 15:43:48 -07:00
|
|
|
exit,
|
2018-05-11 15:41:35 -07:00
|
|
|
packet_recycler.clone(),
|
|
|
|
packet_sender,
|
2018-05-15 08:53:51 -07:00
|
|
|
);
|
2018-03-26 21:07:11 -07:00
|
|
|
|
2018-06-27 11:33:56 -07:00
|
|
|
let blob_recycler = BlobRecycler::default();
|
2018-05-15 09:44:47 -07:00
|
|
|
let request_processor = RequestProcessor::new(bank.clone());
|
2018-07-02 14:45:16 -07:00
|
|
|
let (request_stage, blob_receiver) = RequestStage::new(
|
2018-05-12 10:39:24 -07:00
|
|
|
request_processor,
|
2018-05-15 11:15:29 -07:00
|
|
|
packet_receiver,
|
2018-05-11 19:50:50 -07:00
|
|
|
packet_recycler.clone(),
|
|
|
|
blob_recycler.clone(),
|
|
|
|
);
|
|
|
|
|
2018-07-11 07:38:57 -07:00
|
|
|
let t_responder =
|
|
|
|
streamer::responder("rpu", respond_socket, blob_recycler.clone(), blob_receiver);
|
2018-04-28 00:31:20 -07:00
|
|
|
|
2018-07-03 21:14:08 -07:00
|
|
|
let mut thread_hdls = vec![t_receiver, t_responder];
|
|
|
|
thread_hdls.extend(request_stage.thread_hdls().into_iter());
|
2018-05-15 09:44:47 -07:00
|
|
|
Rpu { thread_hdls }
|
2018-03-26 21:07:11 -07:00
|
|
|
}
|
2018-04-11 13:05:29 -07:00
|
|
|
}
|
2018-07-03 21:14:08 -07:00
|
|
|
|
|
|
|
impl Service for Rpu {
|
|
|
|
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(())
|
|
|
|
}
|
|
|
|
}
|