PR feedback: async with host.run()

This commit is contained in:
mhchia
2020-01-26 16:44:42 +08:00
parent dcc4aa52fc
commit 6e01a7da31
8 changed files with 75 additions and 71 deletions

View File

@ -4,7 +4,7 @@ import sys
import multiaddr
import trio
from libp2p import new_host_trio
from libp2p import new_host
from libp2p.network.stream.net_stream_interface import INetStream
from libp2p.peer.peerinfo import info_from_p2p_addr
from libp2p.typing import TProtocol
@ -34,9 +34,8 @@ async def write_data(stream: INetStream) -> None:
async def run(port: int, destination: str) -> None:
localhost_ip = "127.0.0.1"
listen_addr = multiaddr.Multiaddr(f"/ip4/0.0.0.0/tcp/{port}")
async with new_host_trio(
listen_addrs=[listen_addr]
) as host, trio.open_nursery() as nursery:
host = new_host()
async with host.run(listen_addrs=[listen_addr]), trio.open_nursery() as nursery:
if not destination: # its the server
async def stream_handler(stream: INetStream) -> None:

View File

@ -3,7 +3,7 @@ import argparse
import multiaddr
import trio
from libp2p import new_host_trio
from libp2p import new_host
from libp2p.crypto.secp256k1 import create_new_key_pair
from libp2p.network.stream.net_stream_interface import INetStream
from libp2p.peer.peerinfo import info_from_p2p_addr
@ -34,9 +34,8 @@ async def run(port: int, destination: str, seed: int = None) -> None:
secret = secrets.token_bytes(32)
async with new_host_trio(
listen_addrs=[listen_addr], key_pair=create_new_key_pair(secret)
) as host:
host = new_host(key_pair=create_new_key_pair(secret))
async with host.run(listen_addrs=[listen_addr]):
print(f"I am {host.get_id().to_string()}")