fix update_identity

Author: lvboudre

crates/tpu-client/src/core.rs

@@ -1326,6 +1326,11 @@ where
         }
     }
 
+    pub fn clear(&mut self) {
+        self.peer_stake_map.clear();
+        self.sorted_map.clear();
+    }
+
     ///
     /// Inserts or updates a peer with the given stake.
     ///
@@ -1656,7 +1661,7 @@ where
 
         if self.connection_map.contains_key(&remote_peer_addr) {
             // We already have an active connection for the remote peer address
-            tracing::info!(
+            tracing::debug!(
                 "Re-using existing active connection to remote peer address: {} for remote peer identity: {}",
                 remote_peer_addr,
                 remote_peer_identity
@@ -2435,7 +2440,7 @@ where
                 self.last_peer_activity.remove(&remote_peer_identity);
                 drop(worker_tx);
 
-                tracing::warn!(
+                tracing::trace!(
                     "Tx worker for remote peer: {:?} completed, err: {:?}, canceled: {}, evicted: {}",
                     remote_peer_identity,
                     worker_completed.err,
@@ -2533,6 +2538,7 @@ where
         let tx_worker_sender_map = std::mem::take(&mut self.tx_worker_handle_map);
         let mut tx_worker_set = std::mem::take(&mut self.tx_worker_set);
         let mut tx_queues = std::mem::take(&mut self.tx_queues);
+        self.active_staked_sorted_remote_peer.clear();
         let response_outlet = self.response_outlet.clone();
         let fut = async move {
             drop(tx_worker_sender_map);
@@ -2613,6 +2619,8 @@ where
         self.connecting_remote_peers.clear();
         self.connecting_remote_peers_addr.clear();
         self.connecting_blocked_by_eviction_list.clear();
+        // active staked sroted remote peer map is clear in `schedule_graceful_drop_all_worker`
+        self.active_staked_sorted_remote_peer_addr.clear();
 
         let connecting_meta = std::mem::take(&mut self.connecting_meta);