aiohttp_flood.py

#!/usr/bin/env python3
import asyncio
import aiohttp
import ujson


async def get(semaphore, session, url):
    async with semaphore as sem:
        return await session.get(url)

async def flood(total=10, concurrent=2):
    sem = asyncio.Semaphore(concurrent)
    async with aiohttp.client.ClientSession() as session:
        tasks = [get(sem, session, 'https://now.httpbin.org/') for _ in range(total)]
        results = await asyncio.gather(*tasks)

        epochs = [ujson.loads(await result.text())['now']['epoch'] for result in results]
        print('min:', min(epochs))
        print('max:', max(epochs))
        print('avg:', (max(epochs) - min(epochs)) / total)

if __name__ == '__main__':
    print('50 concurrent requests, 50 total:')
    asyncio.get_event_loop().run_until_complete(
        flood(total=50, concurrent=50)
    )
    print('25 concurrent requests, 50 total:')
    asyncio.get_event_loop().run_until_complete(
        flood(total=50, concurrent=25)
    )
    print('10 concurrent requests, 50 total:')
    asyncio.get_event_loop().run_until_complete(
        flood(total=50, concurrent=10)
    )
    print('5 concurrent requests, 50 total:')
    asyncio.get_event_loop().run_until_complete(
        flood(total=50, concurrent=5)
    )
    print('1 concurrent request, 50 total:')
    asyncio.get_event_loop().run_until_complete(
        flood(total=50, concurrent=1)
    )