Core/grin
0
0
Fork 0
mirror of https://github.com/mimblewimble/grin.git synced 2025-02-01 08:51:08 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

use choose_multiple to clean get_peers() up (#3457)

Browse source
This commit is contained in:
Antioch Peverell 2020-10-01 19:42:57 +01:00 committed by GitHub
parent cc4e608d1b
commit f25b7ae032
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
p2p/src/store.rs
View file

@ -16,8 +16,7 @@
use chrono::Utc;
use num::FromPrimitive;
use rand::seq::SliceRandom;
use rand::thread_rng;
use rand::prelude::*;
use crate::core::ser::{self, Readable, Reader, Writeable, Writer};
use crate::types::{Capabilities, PeerAddr, ReasonForBan};
@ -146,31 +145,28 @@ impl PeerStore {
batch.commit()
}
/// Find some peers in our local db.
pub fn find_peers(
&self,
state: State,
cap: Capabilities,
count: usize,
) -> Result<Vec<PeerData>, Error> {
let mut peers = self
let key = to_key(PEER_PREFIX, "");
let peers = self
.db
.iter::<PeerData>(&to_key(PEER_PREFIX, ""))?
.iter::<PeerData>(&key)?
.map(|(_, v)| v)
.filter(|p| p.flags == state && p.capabilities.contains(cap))
.collect::<Vec<_>>();
peers[..].shuffle(&mut thread_rng());
Ok(peers.iter().take(count).cloned().collect())
.choose_multiple(&mut thread_rng(), count);
Ok(peers)
}
/// List all known peers
/// Used for /v1/peers/all api endpoint
pub fn all_peers(&self) -> Result<Vec<PeerData>, Error> {
let key = to_key(PEER_PREFIX, "");
Ok(self
.db
.iter::<PeerData>(&key)?
.map(|(_, v)| v)
.collect::<Vec<_>>())
Ok(self.db.iter::<PeerData>(&key)?.map(|(_, v)| v).collect())
}
/// Convenience method to load a peer data, update its status and save it