aiojobs 1.3.0

安装

$ pip3 install aiojobs

使用示例

import asyncio
import aiojobs

async def coro(timeout):
    await asyncio.sleep(timeout)

async def main():
    async with aiojobs.Scheduler() as scheduler:
        for i in range(100):
            # spawn jobs
            await scheduler.spawn(coro(i/10))

        await asyncio.sleep(5.0)
        # not all scheduled jobs are finished at the moment
    # Exit from context will gracefully wait on tasks before closing
    # any remaining spawned jobs

asyncio.run(main())

使用调度器保护任务

通常建议使用 asyncio.shield 来保护任务不被取消。然而，内部受保护的任务无法跟踪，因此在应用程序关闭期间存在被取消的风险。

为了解决这个问题，aiojobs 包含一个 aiojobs.Scheduler.shield 方法来屏蔽任务，同时在调度器中跟踪它们。结合 aiojobs.Scheduler.wait_and_close 方法，这允许在应用程序关闭期间，屏蔽的任务有足够的时间成功完成。

例如：

import asyncio
import aiojobs
from contextlib import suppress

async def important():
    print("START")
    await asyncio.sleep(5)
    print("DONE")

async def run_something(scheduler):
    # If we use asyncio.shield() here, then the task doesn't complete and DONE is never printed.
    await scheduler.shield(important())

async def main():
    async with aiojobs.Scheduler() as scheduler:
        t = asyncio.create_task(run_something(scheduler))
        await asyncio.sleep(0.1)
        t.cancel()
        with suppress(asyncio.CancelledError):
            await t

asyncio.run(main())

与 aiohttp.web 的集成

from aiohttp import web
from aiojobs.aiohttp import setup, spawn

async def handler(request):
    await spawn(request, coro())
    return web.Response()

app = web.Application()
app.router.add_get('/', handler)
setup(app)

或只需

from aiojobs.aiohttp import atomic

@atomic
async def handler(request):
    return web.Response()

有关更多信息，请参阅文档：https://aiojobs.readthedocs.io

交流渠道

aio-libs Google 群组：https://groups.google.com/forum/#!forum/aio-libs

请随时在此处发布您的问题和想法。

Gitter Chat https://gitter.im/aio-libs/Lobby

我们支持 Stack Overflow。请在那里的问题中添加 python-asyncio 或 aiohttp 标签。

作者和许可证

aiojobs 包由 Andrew Svetlov 编写。

它是 Apache 2 许可的，并且可以免费获取。