asyncio: корутины и цикл событий

import asyncio, time

async def make_coffee(name, seconds):
    print(f"  {name}: начал")
    await asyncio.sleep(seconds)          # «ждём» и отдаём управление другим
    print(f"  {name}: готово")
    return name

async def main():
    start = time.perf_counter()
    results = await asyncio.gather(
        make_coffee("эспрессо", 0.3),
        make_coffee("капучино", 0.5),
        make_coffee("латте", 0.4),
    )
    print(results, f"за {time.perf_counter() - start:.1f} с")   # ≈ 0.5, а не 1.2

asyncio.run(main())

import asyncio, inspect

async def add(a, b):
    await asyncio.sleep(0)
    return a + b

coro = add(2, 3)
print(coro, inspect.iscoroutine(coro))
print(asyncio.run(coro))                   # выполнить

async def main():
    x = await add(1, 2)                    # внутри корутины — через await
    y = await add(x, 10)
    return y

print(asyncio.run(main()))

import asyncio, time

async def job(name, delay):
    await asyncio.sleep(delay)
    return f"{name} ({delay} с)"

async def main():
    start = time.perf_counter()
    print(await job("A", 0.3), await job("B", 0.3))          # последовательно
    print(f"  {time.perf_counter() - start:.1f} с")

    start = time.perf_counter()
    task_a = asyncio.create_task(job("A", 0.3))               # уже запущены
    task_b = asyncio.create_task(job("B", 0.3))
    print(await task_a, await task_b)
    print(f"  {time.perf_counter() - start:.1f} с")

    start = time.perf_counter()
    print(await asyncio.gather(job("A", 0.3), job("B", 0.3), job("C", 0.3)))
    print(f"  {time.perf_counter() - start:.1f} с")

asyncio.run(main())

import asyncio, time

async def bad_worker(name):
    time.sleep(0.3)                      # ✗ блокирует весь поток — и цикл событий
    return name

async def good_worker(name):
    await asyncio.sleep(0.3)             # ✓ отдаёт управление
    return name

async def main():
    start = time.perf_counter()
    await asyncio.gather(*(bad_worker(i) for i in range(3)))
    print(f"time.sleep:          {time.perf_counter() - start:.1f} с")
    start = time.perf_counter()
    await asyncio.gather(*(good_worker(i) for i in range(3)))
    print(f"await asyncio.sleep: {time.perf_counter() - start:.1f} с")

asyncio.run(main())

import asyncio

class Connection:
    async def __aenter__(self):              # асинхронный контекстный менеджер
        await asyncio.sleep(0.01)
        print("  подключились")
        return self
    async def __aexit__(self, *exc):
        await asyncio.sleep(0.01)
        print("  отключились")

    async def rows(self):                    # асинхронный генератор
        for i in range(3):
            await asyncio.sleep(0.01)        # «следующая порция из БД»
            yield {"id": i}

async def main():
    async with Connection() as conn:
        async for row in conn.rows():
            print("  ", row)
        print([r["id"] async for r in conn.rows()])   # асинхронное включение

asyncio.run(main())

import asyncio

async def child():
    await asyncio.sleep(0)               # здесь корутина уступает управление
    return 42

coro = child()
print(coro.send(None))                   # 1-й шаг: корутина «отдала» управление (yield внутри sleep(0))
try:
    coro.send(None)                      # 2-й шаг: продолжили — корутина завершилась
except StopIteration as stop:
    print("результат:", stop.value)      # return корутины = StopIteration.value