【问题标题】:Python: what does await do in the following context?Python:await 在以下上下文中做了什么?
【发布时间】:2020-03-28 22:59:07
【问题描述】:

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

我在代码中多次看到await ff 是一个空的未来,为什么它需要是 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))

【问题讨论】:

  • @a-jesse-jiryu-davis 原作者有什么意见吗?
  • Stack Overflow 不允许您像这样 ping 任意用户。他们必须已经与您在one of a few specific ways 中评论的帖子进行过互动。

标签: python async-await coroutine


【解决方案1】:

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


Python 协同程序是在旧的迭代器和生成器机制之上实现的,并通过一些额外的强制措施来避免它们混淆。 get 像生成器一样工作,如果 f 是生成器,await fyield from f.__await__() 一样工作。由于f.__await__ 被实现为yield selfawait 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 分配给 ff.callback = self.step 设置future的回调,后面会用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.dataSelectorKey 中提取关联的Future,然后future.resolve() 调用Task.step,这会推进关联的协程,直到它再次产生或终止。

【讨论】:

    猜你喜欢
    • 2021-03-20
    • 2017-07-27
    • 2013-10-18
    • 1970-01-01
    • 2018-01-06
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2013-01-14
    相关资源
    最近更新 更多