asyncio.wait（）で行ったコルーチンを知る方法

Question

私は2つのオブジェクトを持っていて、それらをループで読みたいと思っています。私はこのようなasyncio.waitを使用しています：

done, pending = await asyncio.wait(
    [reader.read(1000), freader.read(1000)], 
    return_when=asyncio.FIRST_COMPLETED) 

今done.pop()は私に最初に終了した未来を与えます。問題は私がどのようにread()操作が完了したかを知ることができないことです。私はtasks変数に[reader.read(1000), freader.read(1000)]を入れて、それらと未来を比較してみました。しかし、これは、未来が元のタスクのどれとも等しくないため、間違っているようです。だからどのコルーチンが完成したのか、私はどのようにして見つけ出すのですか？

Answer 1

.readコールごとに別々のタスクを作成し、それらのタスクを.waitに渡す必要があります。その後、タスクが結果に含まれているかどうかを確認できます。

reader_task = asyncio.ensure_future(reader.read(1000)) 
... 

done, pending = await asyncio.wait(
    [reader_task, ...], 
    return_when=asyncio.FIRST_COMPLETED, 
) 

if reader_task in done: 
    ... 

... 

websocketsのドキュメントのthis example。