Question

我尝试从https://github.com/ajdavis/coroutines-demo/blob/master/50.py的示例中学习协程。它是由用户https://stackoverflow.com/users/618967/a-jesse-jiryu-davis在2015年创建的。

我在代码中多次看到await f。 f是一个空虚的Future，为什么它必须是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。（不要尝试用任何类型的await f替换yield-手动yield的协同工作方式并不相同。）

您正在查看的代码将所有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：在以下情况下等待做什么？

1 个答案: