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

protocols/gossipsub: Add public accessors (#1677)

Browse Source 
Co-authored-by: Toralf Wittner <tw@dtex.org>
Co-authored-by: Max Inden <mail@max-inden.de>
This commit is contained in:
Rüdiger Klaehn
2020-08-13 12:10:52 +02:00
committed by GitHub
parent e32ff74809
commit 0d26f50304
3 changed files with 48 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
protocols/gossipsub/src/behaviour.rs
View File

@ -221,6 +221,25 @@ impl Gossipsub {
}
}
/// Lists the hashes of the topics we are currently subscribed to.
pub fn topics(&self) -> impl Iterator<Item = &TopicHash> {
self.mesh.keys()
}
/// Lists peers for a certain topic hash.
pub fn peers(&self, topic_hash: &TopicHash) -> impl Iterator<Item = &PeerId> {
self.mesh.get(topic_hash).into_iter().map(|x| x.into_iter()).flatten()
}
/// Lists all peers for any topic.
pub fn all_peers(&self) -> impl Iterator<Item = &PeerId> {
let mut res = BTreeSet::new();
for peers in self.mesh.values() {
res.extend(peers);
}
res.into_iter()
}
/// Subscribe to a topic.
///
/// Returns true if the subscription worked. Returns false if we were already subscribed.
@ -1508,10 +1527,10 @@ impl fmt::Debug for Gossipsub {
impl fmt::Debug for PublishConfig {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
match self {
PublishConfig::Signing { author, .. } => f.write_fmt(format_args!("PublishConfig::Signing({})", author)),
PublishConfig::Author(author) => f.write_fmt(format_args!("PublishConfig::Author({})", author)),
PublishConfig::RandomAuthor => f.write_fmt(format_args!("PublishConfig::RandomAuthor")),
PublishConfig::Anonymous => f.write_fmt(format_args!("PublishConfig::Anonymous")),
PublishConfig::Signing { author, .. } => f.write_fmt(format_args!("PublishConfig::Signing({})", author)),
PublishConfig::Author(author) => f.write_fmt(format_args!("PublishConfig::Author({})", author)),
PublishConfig::RandomAuthor => f.write_fmt(format_args!("PublishConfig::RandomAuthor")),
PublishConfig::Anonymous => f.write_fmt(format_args!("PublishConfig::Anonymous")),
}
}
}