revert public api changes

anza-xyz · Dec 16, 2024 · aed704e · aed704e
1 parent 937458f
commit aed704e
Show file tree

Hide file tree

Showing 6 changed files with 13 additions and 11 deletions.
diff --git a/banks-server/src/banks_server.rs b/banks-server/src/banks_server.rs
@@ -458,7 +458,7 @@ pub async fn start_tcp_server(
                 &bank_forks,
                 None,
                 receiver,
-                connection_cache.clone(),
+                &connection_cache,
                 5_000,
                 0,
                 exit.clone(),

diff --git a/rpc/src/rpc.rs b/rpc/src/rpc.rs
@@ -383,7 +383,7 @@ impl JsonRpcRequestProcessor {
             &bank_forks,
             None,
             receiver,
-            connection_cache,
+            &connection_cache,
             1000,
             1,
             exit.clone(),
@@ -6496,7 +6496,7 @@ pub mod tests {
             &bank_forks,
             None,
             receiver,
-            connection_cache,
+            &connection_cache,
             1000,
             1,
             exit,
@@ -6770,7 +6770,7 @@ pub mod tests {
             &bank_forks,
             None,
             receiver,
-            connection_cache,
+            &connection_cache,
             1000,
             1,
             exit,

diff --git a/rpc/src/rpc_service.rs b/rpc/src/rpc_service.rs
@@ -479,7 +479,7 @@ impl JsonRpcService {
             &bank_forks,
             leader_info,
             receiver,
-            connection_cache,
+            &connection_cache,
             send_transaction_service_config,
             exit,
         ));

diff --git a/send-transaction-service/src/lib.rs b/send-transaction-service/src/lib.rs
@@ -3,6 +3,8 @@ pub mod send_transaction_service;
 pub mod send_transaction_service_stats;
 pub mod tpu_info;
 pub mod transaction_client;
+pub use send_transaction_service_stats::SendTransactionServiceStats;
+pub use transaction_client::{CurrentLeaderInfo, LEADER_INFO_REFRESH_RATE_MS};
 
 #[macro_use]
 extern crate solana_metrics;
diff --git a/send-transaction-service/src/send_transaction_service.rs b/send-transaction-service/src/send_transaction_service.rs
@@ -145,7 +145,7 @@ impl SendTransactionService {
         bank_forks: &Arc<RwLock<BankForks>>,
         leader_info: Option<T>,
         receiver: Receiver<TransactionInfo>,
-        connection_cache: Arc<ConnectionCache>,
+        connection_cache: &Arc<ConnectionCache>,
         retry_rate_ms: u64,
         leader_forward_count: u64,
         exit: Arc<AtomicBool>,
@@ -171,7 +171,7 @@ impl SendTransactionService {
         bank_forks: &Arc<RwLock<BankForks>>,
         leader_info: Option<T>,
         receiver: Receiver<TransactionInfo>,
-        connection_cache: Arc<ConnectionCache>,
+        connection_cache: &Arc<ConnectionCache>,
         config: Config,
         exit: Arc<AtomicBool>,
     ) -> Self {
@@ -180,7 +180,7 @@ impl SendTransactionService {
         let retry_transactions = Arc::new(Mutex::new(HashMap::new()));
 
         let client = ConnectionCacheClient::new(
-            connection_cache,
+            connection_cache.clone(),
             tpu_address,
             config.tpu_peers,
             leader_info,
@@ -524,7 +524,7 @@ mod test {
             &bank_forks,
             None,
             receiver,
-            connection_cache,
+            &connection_cache,
             1000,
             1,
             Arc::new(AtomicBool::new(false)),
@@ -558,7 +558,7 @@ mod test {
             &bank_forks,
             None,
             receiver,
-            connection_cache,
+            &connection_cache,
             1000,
             1,
             exit.clone(),

diff --git a/send-transaction-service/src/transaction_client.rs b/send-transaction-service/src/transaction_client.rs
@@ -129,7 +129,7 @@ pub const LEADER_INFO_REFRESH_RATE_MS: u64 = 1000;
 
 /// A struct responsible for holding up-to-date leader information
 /// used for sending transactions.
-pub(crate) struct CurrentLeaderInfo<T>
+pub struct CurrentLeaderInfo<T>
 where
     T: TpuInfo + std::marker::Send + 'static,
 {