refactor host setup helper

This commit is contained in:
zixuanzh 2019-03-17 19:33:10 -04:00 committed by Stuckinaboot
parent 474602c4f7
commit 9aa41b106a

View File

@ -1,6 +1,8 @@
import asyncio
from contextlib import suppress
import asyncio
import multiaddr
from libp2p import new_node
async def cleanup():
@ -12,3 +14,18 @@ async def cleanup():
# Cancelled task raises asyncio.CancelledError that we can suppress:
with suppress(asyncio.CancelledError):
await task
async def set_up_nodes_by_transport_opt(transport_opt_list):
nodes_list = []
for transport_opt in transport_opt_list:
node = await new_node(transport_opt=transport_opt)
await node.get_network().listen(multiaddr.Multiaddr(transport_opt[0]))
nodes_list.append(node)
return tuple(nodes_list)
async def echo_stream_handler(stream):
while True:
read_string = (await stream.read()).decode()
resp = "ack:" + read_string
await stream.write(resp.encode())