2018-05-12 09:53:25 -07:00
|
|
|
//! The `rpu` module implements the Request Processing Unit, a
|
2018-05-08 16:32:50 -07:00
|
|
|
//! 5-stage transaction processing pipeline in software.
|
2018-03-29 11:20:54 -07:00
|
|
|
|
2018-05-14 14:33:11 -07:00
|
|
|
use bank::Bank;
|
2018-03-26 21:07:11 -07:00
|
|
|
use packet;
|
2018-05-12 16:50:55 -07:00
|
|
|
use request_processor::RequestProcessor;
|
|
|
|
use request_stage::RequestStage;
|
2018-05-10 14:30:18 -07:00
|
|
|
use std::net::UdpSocket;
|
2018-05-15 10:33:43 -07:00
|
|
|
use std::sync::Arc;
|
2018-05-14 15:27:40 -07:00
|
|
|
use std::sync::atomic::AtomicBool;
|
|
|
|
use std::sync::mpsc::channel;
|
|
|
|
use std::thread::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-05-15 09:27:18 -07:00
|
|
|
pub 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-05-15 10:21:40 -07:00
|
|
|
bank: Arc<Bank>,
|
2018-05-15 09:27:18 -07:00
|
|
|
requests_socket: UdpSocket,
|
|
|
|
respond_socket: UdpSocket,
|
|
|
|
exit: Arc<AtomicBool>,
|
|
|
|
) -> Self {
|
2018-04-02 19:32:58 -07:00
|
|
|
let packet_recycler = packet::PacketRecycler::default();
|
|
|
|
let (packet_sender, packet_receiver) = channel();
|
2018-05-11 15:41:35 -07:00
|
|
|
let t_receiver = streamer::receiver(
|
|
|
|
requests_socket,
|
|
|
|
exit.clone(),
|
|
|
|
packet_recycler.clone(),
|
|
|
|
packet_sender,
|
2018-05-15 08:53:51 -07:00
|
|
|
);
|
2018-03-26 21:07:11 -07:00
|
|
|
|
2018-05-11 19:50:50 -07:00
|
|
|
let blob_recycler = packet::BlobRecycler::default();
|
2018-05-15 09:44:47 -07:00
|
|
|
let request_processor = RequestProcessor::new(bank.clone());
|
2018-05-12 09:31:28 -07:00
|
|
|
let request_stage = RequestStage::new(
|
2018-05-12 10:39:24 -07:00
|
|
|
request_processor,
|
2018-05-11 19:50:50 -07:00
|
|
|
exit.clone(),
|
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-05-11 19:11:25 -07:00
|
|
|
let t_responder = streamer::responder(
|
|
|
|
respond_socket,
|
2018-04-28 00:31:20 -07:00
|
|
|
exit.clone(),
|
|
|
|
blob_recycler.clone(),
|
2018-05-12 14:24:17 -07:00
|
|
|
request_stage.blob_receiver,
|
2018-04-28 00:31:20 -07:00
|
|
|
);
|
|
|
|
|
2018-05-15 11:15:29 -07:00
|
|
|
let thread_hdls = vec![t_receiver, t_responder, request_stage.thread_hdl];
|
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
|
|
|
}
|