dht_crawler: cleanup, try not to reset key

This commit is contained in:
Victor Shyba 2022-06-11 20:53:25 -03:00 committed by Victor Shyba
parent baf422fc03
commit 174439f517

View file

@ -179,10 +179,10 @@ class Crawler:
latency = None latency = None
for _ in range(3): for _ in range(3):
try: try:
async with self.semaphore:
ping_start = time.perf_counter_ns() ping_start = time.perf_counter_ns()
async with self.semaphore:
await self.node.protocol.get_rpc_peer(make_kademlia_peer(None, address, port)).ping() await self.node.protocol.get_rpc_peer(make_kademlia_peer(None, address, port)).ping()
key = self.node.protocol.peer_manager.get_node_id_for_endpoint(address, port) key = key or self.node.protocol.peer_manager.get_node_id_for_endpoint(address, port)
latency = time.perf_counter_ns() - ping_start latency = time.perf_counter_ns() - ping_start
except asyncio.TimeoutError: except asyncio.TimeoutError:
pass pass
@ -200,7 +200,6 @@ class Crawler:
peers = set() peers = set()
factor = 2048 factor = 2048
for i in range(200): for i in range(200):
#print(i, len(peers), key.hex(), host)
new_peers = await self.request_peers(address, port, key) new_peers = await self.request_peers(address, port, key)
if not new_peers: if not new_peers:
break break