2022-03-09 18:33:05 -08:00
|
|
|
use {
|
2022-05-19 12:29:27 -07:00
|
|
|
crate::connection_cache::ConnectionCacheStats,
|
2022-03-24 08:40:26 -07:00
|
|
|
rayon::iter::{IntoParallelIterator, ParallelIterator},
|
2022-04-12 20:04:40 -07:00
|
|
|
solana_metrics::MovingStat,
|
2022-03-21 18:45:22 -07:00
|
|
|
solana_sdk::{transaction::VersionedTransaction, transport::Result as TransportResult},
|
2022-04-12 20:04:40 -07:00
|
|
|
std::{
|
2022-05-16 11:13:18 -07:00
|
|
|
net::SocketAddr,
|
2022-04-12 20:04:40 -07:00
|
|
|
sync::{atomic::AtomicU64, Arc},
|
|
|
|
},
|
2022-03-09 18:33:05 -08:00
|
|
|
};
|
|
|
|
|
2022-04-12 20:04:40 -07:00
|
|
|
#[derive(Default)]
|
|
|
|
pub struct ClientStats {
|
|
|
|
pub total_connections: AtomicU64,
|
|
|
|
pub connection_reuse: AtomicU64,
|
|
|
|
pub connection_errors: AtomicU64,
|
2022-05-10 09:44:07 -07:00
|
|
|
pub zero_rtt_accepts: AtomicU64,
|
|
|
|
pub zero_rtt_rejects: AtomicU64,
|
2022-04-12 20:04:40 -07:00
|
|
|
|
|
|
|
// these will be the last values of these stats
|
|
|
|
pub congestion_events: MovingStat,
|
|
|
|
pub tx_streams_blocked_uni: MovingStat,
|
|
|
|
pub tx_data_blocked: MovingStat,
|
|
|
|
pub tx_acks: MovingStat,
|
2022-05-18 13:23:29 -07:00
|
|
|
pub make_connection_ms: AtomicU64,
|
2022-04-12 20:04:40 -07:00
|
|
|
}
|
|
|
|
|
2022-03-09 18:33:05 -08:00
|
|
|
pub trait TpuConnection {
|
2022-05-19 12:29:27 -07:00
|
|
|
fn new(tpu_addr: SocketAddr, connection_stats: Arc<ConnectionCacheStats>) -> Self;
|
2022-03-09 18:33:05 -08:00
|
|
|
|
|
|
|
fn tpu_addr(&self) -> &SocketAddr;
|
|
|
|
|
2022-03-21 18:45:22 -07:00
|
|
|
fn serialize_and_send_transaction(
|
|
|
|
&self,
|
|
|
|
transaction: &VersionedTransaction,
|
2022-04-12 20:04:40 -07:00
|
|
|
stats: &ClientStats,
|
2022-03-21 18:45:22 -07:00
|
|
|
) -> TransportResult<()> {
|
|
|
|
let wire_transaction =
|
|
|
|
bincode::serialize(transaction).expect("serialize Transaction in send_batch");
|
2022-04-12 20:04:40 -07:00
|
|
|
self.send_wire_transaction(&wire_transaction, stats)
|
2022-03-09 18:33:05 -08:00
|
|
|
}
|
|
|
|
|
2022-04-12 20:04:40 -07:00
|
|
|
fn send_wire_transaction<T>(
|
|
|
|
&self,
|
|
|
|
wire_transaction: T,
|
|
|
|
stats: &ClientStats,
|
|
|
|
) -> TransportResult<()>
|
2022-03-24 08:40:26 -07:00
|
|
|
where
|
|
|
|
T: AsRef<[u8]>;
|
2022-03-09 18:33:05 -08:00
|
|
|
|
2022-04-12 20:04:40 -07:00
|
|
|
fn send_wire_transaction_async(
|
|
|
|
&self,
|
|
|
|
wire_transaction: Vec<u8>,
|
|
|
|
stats: Arc<ClientStats>,
|
|
|
|
) -> TransportResult<()>;
|
2022-04-12 07:15:59 -07:00
|
|
|
|
2022-03-21 18:45:22 -07:00
|
|
|
fn par_serialize_and_send_transaction_batch(
|
|
|
|
&self,
|
2022-03-24 08:40:26 -07:00
|
|
|
transactions: &[VersionedTransaction],
|
2022-04-12 20:04:40 -07:00
|
|
|
stats: &ClientStats,
|
2022-03-21 18:45:22 -07:00
|
|
|
) -> TransportResult<()> {
|
2022-03-24 08:40:26 -07:00
|
|
|
let buffers = transactions
|
|
|
|
.into_par_iter()
|
2022-03-21 18:45:22 -07:00
|
|
|
.map(|tx| bincode::serialize(&tx).expect("serialize Transaction in send_batch"))
|
2022-03-24 08:40:26 -07:00
|
|
|
.collect::<Vec<_>>();
|
2022-03-21 18:45:22 -07:00
|
|
|
|
2022-04-12 20:04:40 -07:00
|
|
|
self.send_wire_transaction_batch(&buffers, stats)
|
2022-03-21 18:45:22 -07:00
|
|
|
}
|
2022-03-24 08:40:26 -07:00
|
|
|
|
2022-04-12 20:04:40 -07:00
|
|
|
fn send_wire_transaction_batch<T>(
|
|
|
|
&self,
|
|
|
|
buffers: &[T],
|
|
|
|
stats: &ClientStats,
|
|
|
|
) -> TransportResult<()>
|
2022-03-24 08:40:26 -07:00
|
|
|
where
|
|
|
|
T: AsRef<[u8]>;
|
2022-04-14 19:20:34 -07:00
|
|
|
|
|
|
|
fn send_wire_transaction_batch_async(
|
|
|
|
&self,
|
|
|
|
buffers: Vec<Vec<u8>>,
|
|
|
|
stats: Arc<ClientStats>,
|
|
|
|
) -> TransportResult<()>;
|
2022-03-09 18:33:05 -08:00
|
|
|
}
|