Python:await在以下上下文中做什么?

2024-03-02 19:52:11 发布

您现在位置:Python中文网/ 问答频道 /正文

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

我在代码中多次看到await ff是一个空的未来,为什么它需要是{}。有人能更清楚地解释这个概念吗

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))

1条回答
网友
1楼 · 发布于 2024-03-02 19:52:11

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


Python协同程序是在旧的迭代器和生成器机器上实现的,为了避免混淆它们,需要进行一些额外的强制get的工作原理类似于生成器,如果{}是生成器,{}的工作原理类似于{}。由于f.__await__被实现为yield self,因此await f的行为类似于yield f。(不要尝试用任何类型的yield替换await f-手动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)推进协程,直到它产生未来,并将未来分配给ff.callback = self.step设置未来的回调,稍后将使用future.resolve()调用该回调

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

在以下循环中:

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

events = selector.select()等待任何已注册的文件可供读取future = key.dataSelectorKey中提取相关联的未来,并且future.resolve()调用Task.step,这将推进相关联的协程,直到它再次产生或终止

相关问题 更多 >