对于普通的生成器,第一个next调用,相当于启动生成器,会从生成器函数的第一行代码开始执行,直到第一次执行完yield语句(第4行)后,跳出生成器函数。
然后第二个next调用,进入生成器函数后,从yield语句的下一句语句(第5行)开始执行,然后重新运行到yield语句,执行后,跳出生成器函数,
后面再次调用next,依次类推。下面是一个列子:
1 def consumer(): 2 r = ‘here‘ 3 for i in xrange(3): 4 yield r 5 r = ‘200 OK‘+ str(i)
6
7 c = consumer() 8 n1 = c.next() 9 n2 = c.next() 10 n3 = c.next()
了解了next()如何让包含yield的函数执行后,我们再来看另外一个非常重要的函数send(msg)。其实next()和send()在一定意义上作用是相似的,区别是send()可以传递yield表达式的值进去,而next()不能传递特定的值,只能传递None进去。因此,我们可以看做c.next() 和 c.send(None) 作用是一样的。需要提醒的是,第一次调用时,请使用next()语句或是send(None),不能使用send发送一个非None的值,否则会出错的,因为没有Python yield语句来接收这个值。
下面来着重说明下send执行的顺序。当第一次send(None)(对应11行)时,启动生成器,从生成器函数的第一行代码开始执行,直到第一次执行完yield(对应第4行)后,跳出生成器函数。这个过程中,n1一直没有定义。
下面运行到send(1)时,进入生成器函数,注意这里与调用next的不同。这里是从第4行开始执行,把1赋值给n1,但是并不执行yield部分。下面继续从yield的下一语句继续执行,然后重新运行到yield语句,执行后,跳出生成器函数。
即send和next相比,只是开始多了一次赋值的动作,其他运行流程是相同的。
1 def consumer(): 2 r = ‘here‘ 3 while True: 4 n1 = yield r 5 if not n1: 6 return 7 print(‘[CONSUMER] Consuming %s...‘ % n1) 8 r = ‘200 OK‘+str(n1) 9 10 def produce(c): 11 aa = c.send(None) 12 n = 0 13 while n < 5: 14 n = n + 1 15 print(‘[PRODUCER] Producing %s...‘ % n) 16 r1 = c.send(n) 17 print(‘[PRODUCER] Consumer return: %s‘ % r1) 18 c.close() 19 20 c = consumer() 21 produce(c)
时间: 2024-11-06 04:42:00