Function run_slowread

tools/loadtest.py:586–673

(args: argparse.Namespace)

async def run_slowread(args: argparse.Namespace) -> dict:
    import urllib.parse as _up

    u = _up.urlparse(args.url)
    host = u.hostname or "127.0.0.1"
    port = u.port or (443 if u.scheme == "https" else 80)
    path = u.path or "/explain"
    if not path.endswith("/explain"):
        path = path.rstrip("/") + "/explain"

    rng = random.Random(args.seed)
    test_start = time.perf_counter()
    deadline = test_start + args.duration

    stats = {
        "opened": 0,
        "completed": 0,
        "errors": 0,
        "read_timeouts": 0,
        "total_bytes": 0,
        "durations_s": [],
        "err_hist": {},
    }
    witness_results: list[Result] = []

    tasks = [
        asyncio.create_task(
            _slow_reader(
                host,
                port,
                path,
                random.Random(args.seed + i),
                args.read_bps,
                args.chunk,
                deadline,
                test_start,
                stats,
            )
        )
        for i in range(args.concurrency)
    ]
    if args.witness_interval > 0:
        tasks.append(
            asyncio.create_task(
                _witness(
                    args.url,
                    rng,
                    deadline,
                    args.witness_interval,
                    witness_results,
                    test_start,
                )
            )
        )
    await asyncio.gather(*tasks, return_exceptions=True)

    elapsed = time.perf_counter() - test_start
    durs = stats.pop("durations_s")