Блог

У меня есть два бесконечных цикла. Их обработка является легкой. Я не хочу, чтобы они блокировали друг друга. Является ли использование await asyncio.sleep(0) хорошей практикой?

Это мой код

 import asyncio

async def loop1():
    while True:
        print("loop1")
        # pull data from kafka
        await asyncio.sleep(0)


async def loop2():
    while True:
        print("loop2")
        # send data to all clients using asyncio stream api
        await asyncio.sleep(0)


async def main():
    await asyncio.gather(loop1(), loop2())


asyncio.run(main())
 

Две (и многие другие) asyncio задачи не будут блокировать друг друга до тех пор, пока одна из задач не выполнит длительную синхронизацию внутри.

В обеих ваших задачах есть только сетевые операции внутри (запросы Кафки и API), поэтому ни одна из них не заблокирует другую задачу.

Когда вы должны использовать asyncio.sleep(0) ?

Представьте, что у вас есть несколько длительных операций синхронизации — вычислений. Вычисления — это не операция ввода-вывода.

Этот пример больше похож на то, что полезно знать, если у вас есть такие операции в реальном приложении, вы должны перенести их loop.run_in_executor и использовать concurrent.futures.ProcessPoolExecutor в качестве исполнителя. Пример:

 import asyncio


async def long_calc():
    """
    Some Heavy CPU bound task.
    Better make it sync function and move to ProcessPoolExecutor
    """
    s = 0

    for _ in range(100):
        for i in range(1_000_000):
            s  = i**2

        # comment the line and watch result
        # you'll get no working messages
        # that's why I use sleep(0.0) here
        await asyncio.sleep(0.0)

    return s


async def pinger():
    """Task which shows that app is alive"""
    n = 0
    
    while True:
        await asyncio.sleep(1)
        print(f"Working {n}")
        n  = 1


async def amain():
    """Main async function  in this app"""
    # run in asyncio.create_task since we want the task
    # to run in parallel with long_calc  
    # we do not want to wait till it will be finished
    # If it were thread it would be called daemon thread
    asyncio.create_task(pinger())
    # await results of long task
    s = await long_calc()
    print(f"Done: {s}")


if __name__ == '__main__':
    asyncio.run(amain())

 

Если вам нужно, чтобы я привел вам run_in_executor пример — дайте мне знать.