tsarevvv/rust-ipfs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

p2p: Enable floodsub.

Browse Source
This commit is contained in:
David Craven 2019-03-08 15:50:21 +01:00
parent 678f843f5e
commit 3cde9980ae
No known key found for this signature in database
GPG Key ID: DF438712EA50DBB1
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/p2p/behaviour.rs
View File

@ -9,6 +9,7 @@ use libp2p::identify::{Identify, IdentifyEvent};
use libp2p::kad::{Kademlia, KademliaOut as KademliaEvent};
use libp2p::mdns::{Mdns, MdnsEvent};
use libp2p::ping::{Ping, PingEvent};
use libp2p::floodsub::{Floodsub, FloodsubEvent};
//use parity_multihash::Multihash;
use std::sync::Arc;
use tokio::prelude::*;
@ -21,6 +22,7 @@ pub struct Behaviour<TSubstream: AsyncRead + AsyncWrite, TSwarmTypes: SwarmTypes
bitswap: Bitswap<TSubstream, TSwarmTypes>,
ping: Ping<TSubstream>,
identify: Identify<TSubstream>,
floodsub: Floodsub<TSubstream>,
}
impl<TSubstream: AsyncRead + AsyncWrite, TSwarmTypes: SwarmTypes>
@ -32,13 +34,15 @@ impl<TSubstream: AsyncRead + AsyncWrite, TSwarmTypes: SwarmTypes>
MdnsEvent::Discovered(list) => {
for (peer, _) in list {
debug!("mdns: Discovered peer {}", peer.to_base58());
self.bitswap.connect(peer);
self.bitswap.connect(peer.clone());
self.floodsub.add_node_to_partial_view(peer);
}
}
MdnsEvent::Expired(list) => {
for (peer, _) in list {
if !self.mdns.has_node(&peer) {
debug!("mdns: Expired peer {}", peer.to_base58());
self.floodsub.remove_node_from_partial_view(&peer);
}
}
}
@ -111,6 +115,15 @@ impl<TSubstream: AsyncRead + AsyncWrite, TSwarmTypes: SwarmTypes>
}
}
impl<TSubstream: AsyncRead + AsyncWrite, TSwarmTypes: SwarmTypes>
NetworkBehaviourEventProcess<FloodsubEvent> for
Behaviour<TSubstream, TSwarmTypes>
{
fn inject_event(&mut self, event: FloodsubEvent) {
debug!("floodsub: {:?}", event);
}
}
impl<TSubstream: AsyncRead + AsyncWrite, TSwarmTypes: SwarmTypes> Behaviour<TSubstream, TSwarmTypes>
{
/// Create a Kademlia behaviour with the IPFS bootstrap nodes.
@ -132,6 +145,7 @@ impl<TSubstream: AsyncRead + AsyncWrite, TSwarmTypes: SwarmTypes> Behaviour<TSub
"rust-ipfs".into(),
options.key_pair.to_public_key(),
);
let floodsub = Floodsub::new(options.peer_id.to_owned());
Behaviour {
mdns,
@ -139,6 +153,7 @@ impl<TSubstream: AsyncRead + AsyncWrite, TSwarmTypes: SwarmTypes> Behaviour<TSub
bitswap,
ping,
identify,
floodsub,
}
}