Refactor collect_path_diversity to run probes concurrently using asyncio.gather. (#25)

Co-authored-by: google-labs-jules[bot] <161369871+google-labs-jules[bot]@users.noreply.github.com>
This commit is contained in:
Horizontal42 2026-08-19 16:56:22 +03:00 • committed by GitHub
parent 67f677a536
commit 03d54b900c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -1,5 +1,6 @@
from __future__ import annotations
import asyncio
import re
import httpx
@ -158,8 +159,9 @@ async def collect_path_diversity(
timeout: float,
source_ip: str | None = None,
) -> PathDiversity:
hops = [
await probe_edge(client, host, timeout=timeout, source_ip=source_ip)
coros = [
probe_edge(client, host, timeout=timeout, source_ip=source_ip)
for _label, host in targets[:max_targets]
]
return build_path_diversity(client_country, hops)
hops = await asyncio.gather(*coros)
return build_path_diversity(client_country, list(hops))