Wie Sie wissen, welche Coroutinen wurden mit asyncio.wait()

Question

Ich habe zwei StreamReader Objekte und möchte von ihnen in einer Schleife lesen. Ich verwende asyncio.wait wie folgt aus:

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

Jetzt done.pop() gibt mir die Zukunft, die zuerst fertig. Das Problem ist, ich weiß nicht, wie man die Operation read() abgeschlossen findet. Ich habe versucht [reader.read(1000), freader.read(1000)] in eine tasks Variable zu setzen und die fertige Zukunft mit diesen zu vergleichen. Dies scheint jedoch falsch zu sein, da die fertig gestellte Zukunft keiner der ursprünglichen Aufgaben entspricht. Wie soll ich herausfinden, welche Coroutine fertig war?

Answer 1

Sie müssen für jeden .read-Aufruf eine separate Aufgabe erstellen und diese Aufgaben an .wait übergeben. Sie können dann überprüfen, wo die Aufgaben in den Ergebnissen sind.

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: 
    ... 

... 

Siehe z.B. this example aus der websockets Dokumentation.