我尝试从位于https://github.com/ajdavis/coroutines-demo/blob/master/50.py 的示例中学习协程。它是由用户https://stackoverflow.com/users/618967/a-jesse-jiryu-davis在 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))
这段代码是一种奇怪的使用方式await。大多数使用的await代码不像这段代码那样直接与协程实现交互。
Python 协程是在旧的迭代器和生成器机制之上实现的，并进行了一些额外的实施以避免将它们混淆。get就像一台发电机，并await f喜欢的作品yield from f.__await__()会如果f是一台发电机。由于f.__await__实现为yield self，因此await f行为类似于yield f。（不要尝试用await f任何类型的替换yield- 手动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从 中提取关联的 FutureSelectorKey并future.resolve()调用Task.step，这会推进关联的协程，直到它再次产生或终止。
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句