web-dev-qa-db-ja.com

asyncioでタスクをスケジュールおよびキャンセルする方法

クライアントサーバーアプリケーションを書いています。接続されている間、クライアントはサーバーに「毎秒」などの「ハートビート」信号を送信します。サーバー側では、非同期で実行されるタスク（またはコルーチンなど）を追加できるメカニズムが必要です。さらに、クライアントがその「ハートビート」信号の送信を停止したときに、クライアントからのタスクをキャンセルしたいと思います。

言い換えると、サーバーがタスクを開始すると、一種のタイムアウトまたはttl、たとえば3秒になります。サーバーは「ハートビート」信号を受信すると、タスクが完了するか、クライアントが切断される（信号の送信を停止する）まで、タイマーをさらに3秒間リセットします。

これはpymotw.comのasyncioチュートリアルからタスクをキャンセルする 例 です。しかし、ここではevent_loopが開始する前にタスクがキャンセルされているため、私には適していません。

import asyncio

async def task_func():
    print('in task_func')
    return 'the result'


event_loop = asyncio.get_event_loop()
try:
    print('creating task')
    task = event_loop.create_task(task_func())

    print('canceling task')
    task.cancel()

    print('entering event loop')
    event_loop.run_until_complete(task)
    print('task: {!r}'.format(task))
except asyncio.CancelledError:
    print('caught error from cancelled task')
else:
    print('task result: {!r}'.format(task.result()))
finally:
    event_loop.close()
pythonpython-asyncio
7
Sergey Belash

asyncioTaskラッパーを使用して、 ensure_future() メソッドでタスクを実行できます。

_ensure_future_は、コルーチンをTaskラッパーで自動的にラップし、イベントループにアタッチします。 Taskラッパーは、コルーチンがawaitからawaitステートメントに（またはコルーチンが終了するまで）「クランクオーバー」することも保証します。

つまり、通常のコルーチンを_ensure_future_に渡して、結果のTaskオブジェクトを変数に割り当てるだけです。その後、停止する必要があるときに Task.cancel() を呼び出すことができます。

_import asyncio

async def task_func():
    print('in task_func')
    # if the task needs to run for a while you'll need an await statement
    # to provide a pause point so that other coroutines can run in the mean time
    await some_db_or_long_running_background_coroutine()
    # or if this is a once-off thing, then return the result,
    # but then you don't really need a Task wrapper...
    # return 'the result'

async def my_app():
    my_task = None
    while True:
        await asyncio.sleep(0)

        # listen for trigger / heartbeat
        if heartbeat and my_task is None:
            my_task = asyncio.ensure_future(task_func())

        # also listen for termination of hearbeat / connection
        Elif not heartbeat and my_task:
            if not my_task.cancelled():
                my_task.cancel()
            else:
                my_task = None

run_app = asyncio.ensure_future(my_app())
event_loop = asyncio.get_event_loop()
event_loop.run_forever()
_

タスクは、メインフローを中断することなくバックグラウンドで作業を継続する必要がある、長時間実行されるタスクを対象としています。必要なのが1回限りの迅速なメソッドである場合は、代わりに関数を直接呼び出すだけです。

14
songololo