From 29312d005d234ab06bea8939e34ccbfd7b9c5dc4 Mon Sep 17 00:00:00 2001 From: ValdikSS Date: Tue, 29 Mar 2022 16:21:36 +0300 Subject: [PATCH] Propagate timeout value into TCP checker --- src/tor_relay_scanner/scanner.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/tor_relay_scanner/scanner.py b/src/tor_relay_scanner/scanner.py index 6f27f42..35f881e 100644 --- a/src/tor_relay_scanner/scanner.py +++ b/src/tor_relay_scanner/scanner.py @@ -96,9 +96,9 @@ class TorRelay: ret.append((parsed.hostname, parsed.port)) return ret - async def check(self): + async def check(self, timeout=10.0): for i in self.iptuples: - s = TCPSocketConnectChecker(i[0], i[1]) + s = TCPSocketConnectChecker(i[0], i[1], timeout=timeout) sc = await s.connect() if sc[0]: self.reachable.append(i) @@ -150,7 +150,7 @@ async def main_async(args): tasks = list() for relay in test_relays: - tasks.append(asyncio.create_task(relay.check())) + tasks.append(asyncio.create_task(relay.check(TIMEOUT))) fin = await asyncio.gather(*tasks) print("", file=sys.stderr)