Merge pull request #3155 from autonomys/skip-peers-with-successful-se…

…gment-headers-response Skip peers that already replied to segment headers response successfully
autonomys · Oct 21, 2024 · 4555261 · 4555261
2 parents b779d93 + 2476aca
commit 4555261
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/crates/subspace-service/src/sync_from_dsn/segment_header_downloader.rs b/crates/subspace-service/src/sync_from_dsn/segment_header_downloader.rs
@@ -143,7 +143,16 @@ impl<'a> SegmentHeaderDownloader<'a> {
 
             // Acquire segment headers from peers.
             let peers = match get_peers_result {
-                Ok(get_peers_stream) => get_peers_stream.collect::<Vec<_>>().await,
+                Ok(get_peers_stream) => {
+                    get_peers_stream
+                        .filter(|peer_id| {
+                            let known_peer = peer_segment_headers.contains_key(peer_id);
+
+                            async move { !known_peer }
+                        })
+                        .collect::<Vec<_>>()
+                        .await
+                }
                 Err(err) => {
                     warn!(?err, "get_closest_peers returned an error");