2018-05-14 16:36:19 -07:00
|
|
|
//! The `tpu` module implements the Transaction Processing Unit, a
|
|
|
|
//! 5-stage transaction processing pipeline in software.
|
|
|
|
|
|
|
|
use bank::Bank;
|
2018-10-25 14:56:21 -07:00
|
|
|
use banking_stage::{BankingStage, BankingStageReturnType};
|
2018-09-18 13:49:10 -07:00
|
|
|
use entry::Entry;
|
2018-05-29 10:18:12 -07:00
|
|
|
use fetch_stage::FetchStage;
|
2018-10-23 14:42:48 -07:00
|
|
|
use ledger_write_stage::LedgerWriteStage;
|
2018-10-25 14:56:21 -07:00
|
|
|
use poh_service::Config;
|
2018-07-03 21:14:08 -07:00
|
|
|
use service::Service;
|
2018-05-25 15:52:17 -07:00
|
|
|
use sigverify_stage::SigVerifyStage;
|
2018-11-16 08:04:46 -08:00
|
|
|
use solana_sdk::hash::Hash;
|
2018-05-14 16:36:19 -07:00
|
|
|
use std::net::UdpSocket;
|
2018-09-14 14:34:32 -07:00
|
|
|
use std::sync::atomic::{AtomicBool, Ordering};
|
2018-09-18 13:49:10 -07:00
|
|
|
use std::sync::mpsc::Receiver;
|
2018-10-25 16:58:40 -07:00
|
|
|
use std::sync::Arc;
|
2018-09-13 14:00:17 -07:00
|
|
|
use std::thread;
|
2018-09-14 00:17:40 -07:00
|
|
|
|
|
|
|
pub enum TpuReturnType {
|
|
|
|
LeaderRotation,
|
|
|
|
}
|
2018-05-14 16:36:19 -07:00
|
|
|
|
|
|
|
pub struct Tpu {
|
2018-07-03 21:14:08 -07:00
|
|
|
fetch_stage: FetchStage,
|
|
|
|
sigverify_stage: SigVerifyStage,
|
|
|
|
banking_stage: BankingStage,
|
2018-10-23 14:42:48 -07:00
|
|
|
ledger_write_stage: LedgerWriteStage,
|
2018-09-14 14:34:32 -07:00
|
|
|
exit: Arc<AtomicBool>,
|
2018-05-14 16:36:19 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
impl Tpu {
|
2018-08-03 11:06:06 -07:00
|
|
|
pub fn new(
|
2018-07-11 13:40:46 -07:00
|
|
|
bank: &Arc<Bank>,
|
2018-09-26 05:52:13 -07:00
|
|
|
tick_duration: Config,
|
2018-09-06 14:13:40 -07:00
|
|
|
transactions_sockets: Vec<UdpSocket>,
|
2018-08-03 11:06:06 -07:00
|
|
|
ledger_path: &str,
|
2018-07-31 16:54:24 -07:00
|
|
|
sigverify_disabled: bool,
|
2018-10-18 22:57:48 -07:00
|
|
|
max_tick_height: Option<u64>,
|
2018-10-12 00:39:10 -07:00
|
|
|
last_entry_id: &Hash,
|
2018-09-14 14:34:32 -07:00
|
|
|
) -> (Self, Receiver<Vec<Entry>>, Arc<AtomicBool>) {
|
|
|
|
let exit = Arc::new(AtomicBool::new(false));
|
2018-05-29 10:18:12 -07:00
|
|
|
|
2018-09-18 08:02:57 -07:00
|
|
|
let (fetch_stage, packet_receiver) = FetchStage::new(transactions_sockets, exit.clone());
|
2018-05-14 16:36:19 -07:00
|
|
|
|
2018-07-31 16:54:24 -07:00
|
|
|
let (sigverify_stage, verified_receiver) =
|
|
|
|
SigVerifyStage::new(packet_receiver, sigverify_disabled);
|
2018-05-14 16:36:19 -07:00
|
|
|
|
2018-10-18 22:57:48 -07:00
|
|
|
let (banking_stage, entry_receiver) = BankingStage::new(
|
|
|
|
&bank,
|
|
|
|
verified_receiver,
|
|
|
|
tick_duration,
|
|
|
|
last_entry_id,
|
|
|
|
max_tick_height,
|
|
|
|
);
|
2018-05-14 16:36:19 -07:00
|
|
|
|
2018-11-07 18:25:36 -08:00
|
|
|
let (ledger_write_stage, entry_forwarder) =
|
|
|
|
LedgerWriteStage::new(Some(ledger_path), entry_receiver);
|
2018-07-03 21:14:08 -07:00
|
|
|
|
|
|
|
let tpu = Tpu {
|
|
|
|
fetch_stage,
|
|
|
|
sigverify_stage,
|
|
|
|
banking_stage,
|
2018-10-23 14:42:48 -07:00
|
|
|
ledger_write_stage,
|
2018-09-14 14:34:32 -07:00
|
|
|
exit: exit.clone(),
|
2018-07-03 21:14:08 -07:00
|
|
|
};
|
2018-11-02 15:49:14 -07:00
|
|
|
|
2018-09-14 14:34:32 -07:00
|
|
|
(tpu, entry_forwarder, exit)
|
|
|
|
}
|
|
|
|
|
2018-10-25 16:58:40 -07:00
|
|
|
pub fn exit(&self) {
|
2018-09-14 14:34:32 -07:00
|
|
|
self.exit.store(true, Ordering::Relaxed);
|
2018-07-03 21:14:08 -07:00
|
|
|
}
|
2018-07-09 13:53:18 -07:00
|
|
|
|
2018-10-10 16:49:41 -07:00
|
|
|
pub fn is_exited(&self) -> bool {
|
|
|
|
self.exit.load(Ordering::Relaxed)
|
|
|
|
}
|
|
|
|
|
2018-09-14 00:17:40 -07:00
|
|
|
pub fn close(self) -> thread::Result<Option<TpuReturnType>> {
|
2018-07-09 13:53:18 -07:00
|
|
|
self.fetch_stage.close();
|
|
|
|
self.join()
|
|
|
|
}
|
2018-07-03 21:14:08 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
impl Service for Tpu {
|
2018-09-14 00:17:40 -07:00
|
|
|
type JoinReturnType = Option<TpuReturnType>;
|
2018-07-03 21:14:08 -07:00
|
|
|
|
2018-09-14 00:17:40 -07:00
|
|
|
fn join(self) -> thread::Result<(Option<TpuReturnType>)> {
|
2018-09-13 14:00:17 -07:00
|
|
|
self.fetch_stage.join()?;
|
|
|
|
self.sigverify_stage.join()?;
|
2018-10-23 14:42:48 -07:00
|
|
|
self.ledger_write_stage.join()?;
|
2018-10-18 22:57:48 -07:00
|
|
|
match self.banking_stage.join()? {
|
|
|
|
Some(BankingStageReturnType::LeaderRotation) => Ok(Some(TpuReturnType::LeaderRotation)),
|
2018-09-14 00:17:40 -07:00
|
|
|
_ => Ok(None),
|
|
|
|
}
|
2018-05-14 16:36:19 -07:00
|
|
|
}
|
|
|
|
}
|