fluencelabs/rust-libp2p
1
0
Fork 0
mirror of https://github.com/fluencelabs/rust-libp2p synced 2025-06-01 04:01:20 +00:00
Code Issues Projects Releases Wiki Activity

protocols/kad/behaviour: Use HashSet to deduplicate GetProviders (#1528)

Browse Source 
Given that the order of `PeerId`s within the `GetProvidersOk.providers`
set is irrelevant but duplication is at best confusing this commit makes
use of a `HashSet` instead of a `Vec` to return unique `PeerId`s only.
This commit is contained in:
Max Inden 2020-03-31 12:00:17 +02:00 committed by GitHub
parent 0d3e4f2051
commit 2296a87b73
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
protocols/kad/src/behaviour.rs
View File

@ -42,7 +42,7 @@ use libp2p_swarm::{
use log::{info, debug, warn};
use smallvec::SmallVec;
use std::{borrow::{Borrow, Cow}, error, iter, time::Duration};
use std::collections::VecDeque;
use std::collections::{HashSet, VecDeque};
use std::num::NonZeroUsize;
use std::task::{Context, Poll};
use wasm_timer::Instant;
@ -534,7 +534,7 @@ where
pub fn get_providers(&mut self, key: record::Key) {
let info = QueryInfo::GetProviders {
key: key.clone(),
providers: Vec::new(),
providers: HashSet::new(),
};
let target = kbucket::Key::new(key);
let peers = self.kbuckets.closest_keys(&target);
@ -1233,7 +1233,7 @@ where
providers, ..
} = &mut query.inner.info {
for peer in provider_peers {
providers.push(peer.node_id);
providers.insert(peer.node_id);
}
}
}
@ -1701,7 +1701,7 @@ pub type GetProvidersResult = Result<GetProvidersOk, GetProvidersError>;
#[derive(Debug, Clone)]
pub struct GetProvidersOk {
pub key: record::Key,
pub providers: Vec<PeerId>,
pub providers: HashSet<PeerId>,
pub closest_peers: Vec<PeerId>
}
@ -1710,7 +1710,7 @@ pub struct GetProvidersOk {
pub enum GetProvidersError {
Timeout {
key: record::Key,
providers: Vec<PeerId>,
providers: HashSet<PeerId>,
closest_peers: Vec<PeerId>
}
}
@ -1843,7 +1843,7 @@ enum QueryInfo {
/// The key for which to search for providers.
key: record::Key,
/// The found providers.
providers: Vec<PeerId>,
providers: HashSet<PeerId>,
},
/// A query that searches for the closest closest nodes to a key to be