Python: await 在下面的上下文中做了什么？

Question

我尝试从 2015 年位于 https://github.com/ajdavis/coroutines-demo/blob/master/50.py. It was created by user https://whosebug.com/users/618967/a-jesse-jiryu-davis 的示例中学习协程。

我在代码中多次看到 await f。 f 是一个空的未来，为什么它需要 await。有人可以更清楚地解释这个概念吗？

from selectors import DefaultSelector, EVENT_WRITE, EVENT_READ
import socket
import time

selector = DefaultSelector()
n_jobs = 0

class Future:
    def __init__(self):
        self.callback = None

    def resolve(self):
        self.callback()

    def __await__(self):
        yield self

class Task:
    def __init__(self, coro):
        self.coro = coro
        self.step()

    def step(self):
        try:
            f = self.coro.send(None)
        except StopIteration:
            return

        f.callback = self.step

async def get(path):
    global n_jobs
    n_jobs += 1
    s = socket.socket()
    s.setblocking(False)
    try:
        s.connect(('localhost', 5000))
    except BlockingIOError:
        pass

    f = Future()
    selector.register(s.fileno(), EVENT_WRITE, f)
    await f
    selector.unregister(s.fileno())

    s.send(('GET %s HTTP/1.0\r\n\r\n' % path).encode())
    buf = []

    while True:
        f = Future()
        selector.register(s.fileno(), EVENT_READ, f)
        await f
        selector.unregister(s.fileno())
        chunk = s.recv(1000)
        if chunk:
            buf.append(chunk)
        else:
            break

    # Finished.
    print((b''.join(buf)).decode().split('\n')[0])
    n_jobs -= 1

start = time.time()
Task(get('/foo'))
Task(get('/bar'))

while n_jobs:
    events = selector.select()
    for key, mask in events:
        future = key.data
        future.resolve()

print('took %.2f seconds' % (time.time() - start))

Answer 1

这段代码是一种奇怪的使用方式await。大多数使用 await 的代码并不像这段代码那样直接与协程实现交互。

Python 协程是在旧的迭代器和生成器机制之上实现的，有一点额外的强制执行以避免混淆它们。 get 像发电机一样工作，await f 像 yield from f.__await__() 一样工作，如果 f 是发电机。由于 f.__await__ 实现为 yield self，await f 的行为类似于 yield f。（不要尝试用任何类型的 yield 替换 await f - 手动 yielding 在协程中的工作方式不同。）

您正在查看的代码将所有 get 协程包装在一个 Task 对象中，并且 Task.step 看起来像这样：

def step(self):
    try:
        f = self.coro.send(None)
    except StopIteration:
        return

    f.callback = self.step

f = self.coro.send(None) 推进协程直到产生 Future，并将 Future 分配给 f。 f.callback = self.step 设置未来的回调，稍后将用 future.resolve() 调用。

get 呼叫 selector.register(s.fileno(), EVENT_READ, f)。这会向选择器注册指定的文件，因此当文件准备好读取时，selector.select() 的输出将包含一个 SelectorKey 来指示这一事实。作为第三个 register 参数传递的任何对象都将附加到 SelectorKey，因此在这里，Future 将附加到 SelectorKey.

在以下循环中：

while n_jobs:
    events = selector.select()
    for key, mask in events:
        future = key.data
        future.resolve()

events = selector.select() 等待任何已注册的文件可供阅读。 future = key.data 从 SelectorKey 中提取关联的 Future，然后 future.resolve() 调用 Task.step，这会推进关联的协程，直到它再次产生或终止。

Python: await 在下面的上下文中做了什么？

Python: what does await do in the following context?

python

coroutine

async-await