fix(db): update port and services for gossiped peers in `StatelessPee…

…rStore`
rustaceanrob · Jan 9, 2025 · abbd48e · abbd48e
1 parent 3cf1220
commit abbd48e
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/db/memory/peers.rs b/src/db/memory/peers.rs
@@ -28,10 +28,15 @@ impl StatelessPeerStore {
     }
 
     async fn update(&mut self, peer: PersistedPeer) -> Result<(), StatelessPeerStoreError> {
-        // Don't add back peers we already connected to this session.
         match peer.status {
             PeerStatus::New => {
-                self.list.entry(peer.clone().addr).or_insert(peer);
+                self.list
+                    .entry(peer.clone().addr)
+                    .and_modify(|stored| {
+                        stored.port = peer.port;
+                        stored.services = peer.services;
+                    })
+                    .or_insert(peer);
                 Ok(())
             }
             PeerStatus::Tried => Ok(()),