Python: что ожидает в следующем контексте? - PullRequest
Новая
       27

Python: что ожидает в следующем контексте?

0 голосов
/ 29 марта 2020

Я пытаюсь выучить сопрограммы из примера, расположенного по адресу https://github.com/ajdavis/coroutines-demo/blob/master/50.py. Он был создан пользователем { ссылка } в 2015 году.

Я вижу 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))

1 Ответ

1 голос
/ 29 марта 2020

Этот код довольно странный способ использования await. Большая часть кода, который использует await, не взаимодействует с реализацией сопрограммы так же напрямую, как этот код.

Python сопрограммы реализованы поверх старого механизма итератора и генератора, с небольшим дополнительные меры предосторожности, чтобы не перепутать их. get работает как генератор, а await f работает как yield from f.__await__(), если бы f был генератором. Поскольку f.__await__ реализован как yield self, await f ведет себя как yield f. (Не пытайтесь заменить await f каким-либо видом yield - вручную yield ing не работает так же, как в сопрограмме.)

Код, который вы просматриваете, охватывает все 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, поэтому здесь Будущее будет прикреплено к SelectorKey.

В следующем l oop: 

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

events = selector.select() ждет, пока любой из зарегистрированных файлов будет доступен для чтения. future = key.data извлекает связанное Future из SelectorKey, а future.resolve() вызывает Task.step, который продвигает ассоциированную сопрограмму до тех пор, пока не вернется снова или не завершится.

...