2019-03-17 12:14:28 +00:00
|
|
|
use crate::rpc::{RPCEvent, RPCMessage, Rpc};
|
2019-03-07 05:17:06 +00:00
|
|
|
use futures::prelude::*;
|
|
|
|
use libp2p::{
|
|
|
|
core::swarm::{NetworkBehaviourAction, NetworkBehaviourEventProcess},
|
2019-03-12 06:28:11 +00:00
|
|
|
gossipsub::{Gossipsub, GossipsubConfig, GossipsubEvent},
|
2019-03-07 05:17:06 +00:00
|
|
|
tokio_io::{AsyncRead, AsyncWrite},
|
|
|
|
NetworkBehaviour, PeerId,
|
|
|
|
};
|
2019-03-13 04:37:44 +00:00
|
|
|
use types::Topic;
|
2019-03-07 05:17:06 +00:00
|
|
|
|
|
|
|
/// Builds the network behaviour for the libp2p Swarm.
|
|
|
|
/// Implements gossipsub message routing.
|
|
|
|
#[derive(NetworkBehaviour)]
|
2019-03-12 06:28:11 +00:00
|
|
|
#[behaviour(out_event = "BehaviourEvent", poll_method = "poll")]
|
2019-03-07 05:17:06 +00:00
|
|
|
pub struct Behaviour<TSubstream: AsyncRead + AsyncWrite> {
|
|
|
|
gossipsub: Gossipsub<TSubstream>,
|
|
|
|
// TODO: Add Kademlia for peer discovery
|
|
|
|
/// The events generated by this behaviour to be consumed in the swarm poll.
|
2019-03-14 15:48:09 +00:00
|
|
|
serenity_rpc: Rpc<TSubstream>,
|
2019-03-07 05:17:06 +00:00
|
|
|
#[behaviour(ignore)]
|
2019-03-12 06:28:11 +00:00
|
|
|
events: Vec<BehaviourEvent>,
|
2019-03-07 05:17:06 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Implement the NetworkBehaviourEventProcess trait so that we can derive NetworkBehaviour for Behaviour
|
|
|
|
impl<TSubstream: AsyncRead + AsyncWrite> NetworkBehaviourEventProcess<GossipsubEvent>
|
|
|
|
for Behaviour<TSubstream>
|
|
|
|
{
|
|
|
|
fn inject_event(&mut self, event: GossipsubEvent) {
|
2019-03-12 06:28:11 +00:00
|
|
|
match event {
|
|
|
|
GossipsubEvent::Message(message) => {
|
|
|
|
let gs_message = String::from_utf8_lossy(&message.data);
|
|
|
|
// TODO: Remove this type - debug only
|
|
|
|
self.events
|
|
|
|
.push(BehaviourEvent::Message(gs_message.to_string()))
|
|
|
|
}
|
|
|
|
_ => {}
|
|
|
|
}
|
2019-03-07 05:17:06 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-03-17 12:14:28 +00:00
|
|
|
impl<TSubstream: AsyncRead + AsyncWrite> NetworkBehaviourEventProcess<RPCMessage>
|
2019-03-14 15:48:09 +00:00
|
|
|
for Behaviour<TSubstream>
|
|
|
|
{
|
2019-03-17 12:14:28 +00:00
|
|
|
fn inject_event(&mut self, event: RPCMessage) {
|
|
|
|
match event {
|
|
|
|
RPCMessage::PeerDialed(peer_id) => {
|
|
|
|
self.events.push(BehaviourEvent::PeerDialed(peer_id))
|
|
|
|
}
|
|
|
|
RPCMessage::RPC(rpc_event) => self.events.push(BehaviourEvent::RPC(rpc_event)),
|
|
|
|
}
|
2019-03-14 15:48:09 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-03-07 05:17:06 +00:00
|
|
|
impl<TSubstream: AsyncRead + AsyncWrite> Behaviour<TSubstream> {
|
2019-03-17 12:14:28 +00:00
|
|
|
pub fn new(local_peer_id: PeerId, gs_config: GossipsubConfig, log: &slog::Logger) -> Self {
|
2019-03-07 05:17:06 +00:00
|
|
|
Behaviour {
|
|
|
|
gossipsub: Gossipsub::new(local_peer_id, gs_config),
|
2019-03-17 12:14:28 +00:00
|
|
|
serenity_rpc: Rpc::new(log),
|
2019-03-07 05:17:06 +00:00
|
|
|
events: Vec::new(),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-03-12 06:28:11 +00:00
|
|
|
/// Consumes the events list when polled.
|
|
|
|
fn poll<TBehaviourIn>(
|
|
|
|
&mut self,
|
|
|
|
) -> Async<NetworkBehaviourAction<TBehaviourIn, BehaviourEvent>> {
|
2019-03-07 05:17:06 +00:00
|
|
|
if !self.events.is_empty() {
|
|
|
|
return Async::Ready(NetworkBehaviourAction::GenerateEvent(self.events.remove(0)));
|
|
|
|
}
|
|
|
|
|
|
|
|
Async::NotReady
|
|
|
|
}
|
|
|
|
}
|
2019-03-12 06:28:11 +00:00
|
|
|
|
2019-03-18 12:34:44 +00:00
|
|
|
/// Implements the combined behaviour for the libp2p service.
|
2019-03-12 06:28:11 +00:00
|
|
|
impl<TSubstream: AsyncRead + AsyncWrite> Behaviour<TSubstream> {
|
2019-03-18 12:34:44 +00:00
|
|
|
/// Subscribes to a gossipsub topic.
|
2019-03-13 04:37:44 +00:00
|
|
|
pub fn subscribe(&mut self, topic: Topic) -> bool {
|
|
|
|
self.gossipsub.subscribe(topic)
|
|
|
|
}
|
|
|
|
|
2019-03-18 12:34:44 +00:00
|
|
|
/// Sends an RPC Request/Response via the RPC protocol.
|
|
|
|
pub fn send_rpc(&mut self, peer_id: PeerId, rpc_event: RPCEvent) {
|
|
|
|
self.serenity_rpc.send_rpc(peer_id, rpc_event);
|
2019-03-12 06:28:11 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// The types of events than can be obtained from polling the behaviour.
|
|
|
|
pub enum BehaviourEvent {
|
2019-03-17 12:14:28 +00:00
|
|
|
RPC(RPCEvent),
|
|
|
|
PeerDialed(PeerId),
|
2019-03-12 06:28:11 +00:00
|
|
|
// TODO: This is a stub at the moment
|
|
|
|
Message(String),
|
|
|
|
}
|