varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2025-12-31 20:36:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add opened_stream call for non-initiator

Browse Source
This commit is contained in:
Stuckinaboot
2019-03-14 12:47:40 -04:00
parent a0ef64fff7
commit 4b9327c5a3
2 changed files with 107 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
libp2p/network/swarm.py
View File

@ -5,6 +5,7 @@ from libp2p.protocol_muxer.multiselect import Multiselect
from libp2p.peer.id import id_b58_decode
from .network_interface import INetwork
from .notifee_interface import INotifee
from .stream.net_stream import NetStream
from .connection.raw_connection import RawConnection
@ -175,8 +176,8 @@ class Swarm(INetwork):
"""
:param notifee: object implementing Notifee interface
"""
# TODO: Add check to ensure notifee conforms to Notifee interface
self.notifees.append(notifee)
if isinstance(notifee, INotifee):
self.notifees.append(notifee)
def add_transport(self, transport):
# TODO: Support more than one transport
@ -195,6 +196,10 @@ def create_generic_protocol_handler(swarm):
# Perform protocol muxing to determine protocol to use
_, handler = await multiselect.negotiate(muxed_stream)
# Call notifiers since event occurred
for notifee in swarm.notifees:
await notifee.opened_stream(swarm, muxed_stream)
# Give to stream handler
asyncio.ensure_future(handler(muxed_stream))