Python stompest 和 yield

Question

我在使用 Python Stompest 库在帧到达时生成帧时遇到问题。当我在函数中包含 yield 关键字时，代码不会像它应该那样阻塞并等待帧。

def receive(stomp):
    received_messages = 0
    while True:
        frame = stomp.receiveFrame()
        stomp.ack(frame)
        yield frame.body

但是，如果我所做的只是打印框架主体，那么它会像它应该的那样等待

def receive(stomp):
    while True:
        frame = stomp.receiveFrame()
        stomp.ack(frame)
        print frame.body

在这种情况下使用yield 有什么问题吗？

Answer 1

当您将yield 放入函数中时，它会变成一个返回生成器的函数，该生成器仅在迭代或某种理解中需要它时才会被延迟评估。在你给出的第一种情况下，

>> x = receive(stomp)
>> type(x)
<type 'generator'>

要让生成器x 真正做点什么，你可以这样做：

for frame in x: print(frame)

当您迭代生成器时，它实际上会执行您定义的函数中的语句（包括阻塞直到帧到达）并返回每次迭代的帧。使用yield 创建生成器对象可以省去您手动实现状态以记住您正在进行的迭代的麻烦。