use quic proxy in integration test

This commit is contained in:
GroovieGermanikus 2023-07-20 18:23:57 +02:00
parent c4093b58db
commit 4c603e4e85
1 changed files with 3 additions and 10 deletions

View File

@ -122,9 +122,7 @@ pub fn quic_proxy_and_solana_streamer() {
tokio::spawn(start_literpc_client( tokio::spawn(start_literpc_client(
listen_addr, listen_addr,
literpc_validator_identity, literpc_validator_identity,
TpuConnectionPath::QuicForwardProxyPath { proxy_listen_addr,
forward_proxy_address: proxy_listen_addr,
},
)); ));
}); });
@ -256,9 +254,9 @@ const QUIC_CONNECTION_PARAMS: QuicConnectionParameters = QuicConnectionParameter
async fn start_literpc_client( async fn start_literpc_client(
streamer_listen_addrs: SocketAddr, streamer_listen_addrs: SocketAddr,
literpc_validator_identity: Arc<Keypair>, literpc_validator_identity: Arc<Keypair>,
tpu_connection_path: TpuConnectionPath, forward_proxy_address: SocketAddr,
) -> anyhow::Result<()> { ) -> anyhow::Result<()> {
info!("Start lite-rpc test client using {} ...", tpu_connection_path); info!("Start lite-rpc test client using quic proxy at {} ...", forward_proxy_address);
let fanout_slots = 4; let fanout_slots = 4;
@ -274,11 +272,6 @@ async fn start_literpc_client(
// let tpu_connection_manager = // let tpu_connection_manager =
// TpuConnectionManager::new(certificate, key, fanout_slots as usize).await; // TpuConnectionManager::new(certificate, key, fanout_slots as usize).await;
let forward_proxy_address = match tpu_connection_path {
TpuConnectionPath::QuicForwardProxyPath { forward_proxy_address } => forward_proxy_address,
_ => panic!("Expected TpuConnectionPath::Quic"),
};
let quic_proxy_connection_manager = let quic_proxy_connection_manager =
QuicProxyConnectionManager::new(certificate, key, literpc_validator_identity.clone(), forward_proxy_address).await; QuicProxyConnectionManager::new(certificate, key, literpc_validator_identity.clone(), forward_proxy_address).await;