Function main

benchmark/bench.py:410–449

()

def main():
    response_sizes = ["20k", "50k", "200k"]
    requests_number = 200
    thread_counts = [1, 8, 32]

    sync_packages = [
        ("tls_client", tls_client.Session),
        ("httpx", httpx.Client),
        ("requests", requests.Session),
        ("niquests", niquests.Session),
        ("rnet", rnet.BlockingClient),
        ("curl_cffi", curl_cffi.requests.Session),
        ("pycurl", PycurlSession),
    ]
    async_packages = [
        ("httpx", httpx.AsyncClient),
        ("aiohttp", aiohttp.ClientSession),
        ("rnet", rnet.Client),
        ("curl_cffi", curl_cffi.requests.AsyncSession),
    ]

    sync_packages = add_package_version(sync_packages)
    async_packages = add_package_version(async_packages)

    all_results = []

    for size in response_sizes:
        url = f"http://127.0.0.1:8000/{size}"

        all_results += run_sync_tests(sync_packages, url, requests_number)
        all_results += run_async_tests(async_packages, url, requests_number)

        for threads in thread_counts:
            all_results += run_threaded_tests(
                sync_packages, url, requests_number, threads
            )

    df = pd.DataFrame(all_results)
    df.to_csv("benchmark_results.csv", index=False)
    plot_benchmark_multi(df, "benchmark_multi.jpg")


if __name__ == "__main__":