Use cluster info functions for tpu (#345) (#347)

jito-foundation · Sep 22, 2023 · 3230a7d · 3230a7d
1 parent d117fb4
commit 3230a7d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 7 deletions.
diff --git a/core/Cargo.toml b/core/Cargo.toml
@@ -71,9 +71,9 @@ strum_macros = { workspace = true }
 sys-info = { workspace = true }
 tempfile = { workspace = true }
 thiserror = { workspace = true }
-tokio = { version = "~1.14.1", features = ["full"] }
 tip-distribution = { workspace = true }
 tip-payment = { workspace = true }
+tokio = { version = "~1.14.1", features = ["full"] }
 tonic = { workspace = true }
 trees = { workspace = true }
 

diff --git a/core/src/proxy/fetch_stage_manager.rs b/core/src/proxy/fetch_stage_manager.rs
@@ -159,11 +159,8 @@ impl FetchStageManager {
         tpu_address: SocketAddr,
         tpu_forward_address: SocketAddr,
     ) -> Result<(), contact_info::Error> {
-        let mut new_contact_info = cluster_info.my_contact_info();
-        // TODO (LB): double check protocol!!!!!!
-        new_contact_info.set_tpu(tpu_address)?;
-        new_contact_info.set_tpu_forwards(tpu_forward_address)?;
-        cluster_info.set_my_contact_info(new_contact_info);
+        cluster_info.set_tpu(tpu_address)?;
+        cluster_info.set_tpu_forwards(tpu_forward_address)?;
         Ok(())
     }
 

diff --git a/validator/src/admin_rpc_service.rs b/validator/src/admin_rpc_service.rs
@@ -942,7 +942,6 @@ mod tests {
         },
         std::{
             collections::HashSet,
-            str::FromStr,
             sync::{atomic::AtomicBool, Mutex},
         },
     };
-Original file line number
+Diff line change
@@ Expand Up / @@ -942,7 +942,6 @@ mod tests { @@
             },
             std::{
                 collections::HashSet,
-                str::FromStr,
                 sync::{atomic::AtomicBool, Mutex},
             },
         };
@@ Expand Down @@