首页 > 编程语言 > 详细

Python高级编程之生成器(Generator)与coroutine(二):coroutine介绍

时间:2015-04-25 19:32:44      阅读:319      评论:0      收藏:0      [点我收藏+]
原创作品,转载请注明出处:点我

上一篇文章Python高级编程之生成器(Generator)与coroutine(一):Generator中,我们介绍了什么是Generator,以及写了几个使用Generator Function的示例,这一小节,我们会介绍Python的coroutine,以及会有一个小例子,再接下来的文章中会以代码的形式一步步介绍coroutine的高级用法。

coroutine(协程)

什么是coroutine?coroutine跟Generator有什么区别?下面先看一段代码: 

1 def grep_co(pattern):
2     print "Lookin for %s" % pattern
3     while True:
4         # 执行完下面这句,函数挂起,等待接收数据,通过send()函数
5         # yield line
6         line = (yield )7         if pattern in line:
8             print line

 

grep_co就是一个coroutine,从代码的角度来看,coroutine跟generator的唯一区别就是在第6行,coroutine是

line = (yield)

而Generator是:

yield line

那么这两者有什么区别呢?先别急,我们接着往下看:

在Python2.5及之后的版本中,yield可以用作为表达式,就是gerp_co()函数中的这种用法。那么问题来了,gerp_co()函数中的line的值应该是多少?这个值是谁给他的?

答案很简单:line的值是我们(grep_co())的调用者发送过去的。怎么发送的?Easy!使用send(value)函数即可。先来看下执行效果:

 1 >>> def grep_co(pattern):
 2     print "Looking for %s" %pattern
 3     while True:
 4         line = (yield)
 5         if pattern in line:
 6             print line
 7 
 8             
 9 >>> g = grep_co("python")
10 >>> g
11 <generator object grep_co at 0x01FA3B98>

跟Generator一样,当调用gerp_co("python")的时候,并不会立即执行grep_co函数,而是会返回一个generator类型的对象,同样是在这个对象调用了next()的时候才会开始执行这个函数:

>>> g.next()
Looking for python

调用了next()函数之后,函数开始执行,执行到第6行,也就是line = (yield)这一行代码的时候,碰到了yield关键字,跟Generator一样,此时,整个函数会挂起,由于yield后面没有跟随其他的变量(一般情况下,coroutine中的yield语句也不会跟随返回值,这个后面会讲到),所以此时不会返回任何数据,只是单纯的保存执行环境并挂起暂停执行。

既然coroutine跟Generator一样碰到yield关键字会挂起,那么是不是也跟Generator一样调用next()函数继续执行呢?其实你如果能够这样想我会很高兴,说明你有在认真的看,O(∩_∩)O~。不幸的是想法是好的,可惜是错的,O(∩_∩)O哈哈~,应该使用send(value)函数。接着上面往下走,上面调用了g.next(),函数开始执行,碰到了yield关键字,函数挂起暂停执行,现在调用g.send("I love python")函数,执行结果如下:

>>> g.send("Hello,I love python")
Hello,I love python

可以看到,send 函数有一个参数,这个参数就是传递个line这个变量的。调用了send("I love python")这个函数之后,grep_co()这个函数会接着上次挂起的地方往下执行,也就是在第六行line = (yield)这个地方,send("I love python")函数的参数会被传递给line这个变量,然后接着往下执行,直到执行完毕或者再次碰到yield关键字。在这个例子中,line的值是"I love pyhton",pattern的值是"python",if判断为真,打印输出line,接着往下执行,因为是在一个无限循环当中,再次碰到了yield这个关键字,挂起并暂停。所以我们会看到上面的执行结果。

>>> g.send("Life is short,Please use Python")
>>> g.send("Life is short,Please use python")
Life is short,Please use python

我们再继续调用send(value)函数,会重复上面的执行过程。

讲了这么多,那么什么才是coroutine呢?我相信聪明的你应该已经猜到了:

所谓的coroutine,也就是一个包含有yield关键字的函数,但是跟Generator不同的是,coroutine会以value = (yield)的方式使用yield关键字,并且接受调用者通过send(value)函数发送过来的数据,然后消费这个数据(consume the value)。

在使用coroutine,有一点很需要注意的就是:所有的coroutine必须要先调用.next()或者send(None)才行。在调用send传入非None值前,生成器必须处于挂起状态,否则将抛出异常。当然,也可以使用.next()恢复生成器,只不过此时coroutine接收到的value为None。

可以调用.close()关闭coroutine。关闭coroutine之后,再次调用.nect()或者.send(value)之后会抛出异常。

>>> g.close()
>>> g.send("corotuine has already closed")

Traceback (most recent call last):
  File "<pyshell#16>", line 1, in <module>
    g.send("corotuine has already closed")
StopIteration
>>> 

.close()会抛出GeneratorExit异常,我们可以在代码中捕获并处理这个异常,而且一般情况下也应该处理这个异常。

1 def grep(pattern):
2     print "Looking for %s" %pattern
3     try:
4         while True:
5             line = (yield)
6             if pattern in line:
7                 print line
8     except GeneratorExit:
9         print "Going away.Goodbye"

当然也可以通过throw()函数在生成器内部抛出一个指定的异常。

>>> g.send("Life is short,please use python")
Life is short,please use python
>>> g.throw(RuntimeError,"You‘ar hosed")

Traceback (most recent call last):
  File "<pyshell#14>", line 1, in <module>
    g.throw(RuntimeError,"You‘ar hosed")
  File "<pyshell#10>", line 5, in grep
    line = (yield)
RuntimeError: Youar hosed

好了,coroutine已经介绍的差不多了,我们可以看到coroutine可以很方便的挂起和执行,也有多个人口点和出口点,而普通的函数一般只有一个入口点和出口点。

Generator和coroutine用起来很像,但是仅此而已,Generator和coroutine是两个完全不相同的概念。Generator产生(返回)数据用来在迭代(iterator)中使用,而coroutine则是需要其他的地方发送数据过来,从而消费数据(consume value)。

接下来讲的是使用coroutine要注意的地方:

第一,就是千万别忘记了在使用coroutine前要先调用.next()函数。但是这一点经常容易忘记,所以可以使用一个function decorator.

1 # 作为装饰器用,因为经常容易会忘记调用.next()函数
2 def coroutine(func):
3     def start(*args,**kargs):
4         cr = func(*args,**kargs)
5         cr.next()
6         return cr
7     return start

第二:最好是不要把Generator和coroutine混合在一起用,也就是receive_value = (yield return_value)这种方式来用。因为这会很难以理解,而且也会出现某些很诡异的情况。先看代码:

1 def countdown_co(n):
2     print "Counting down from ",n
3     while n >= 0:
4         newvalue = (yield n)
5         # 如果接收到了newvalue,则重新设置n
6         if newvalue is not None:
7             n = newvalue
8         else:
9             n -= 1

代码很简单,同时使用了coroutine和Generator,诡异的情况发生了。先是写一个函数test_countdown_co():

1 def test_countdown_co():
2     c = countdown_co(5)
3     for n in c:
4         print n
5         if 5 == n:
6             c.send(3)

然后在IDLE终端调用这个函数,可以看到函数的执行结果为:

>>> test_countdown_co()
Counting down from  5
5
2
1
0

现在,我们在IDLE终端直接输入上面的test_countdown_co()代码来测试countdown_co()函数:

>>> c = countdown_co(5)
>>> for n in c:
    print n
    if 5 == n:
        c.send(3)

        
Counting down from  5
5
3
2
1
0
>>> 

可以看到一样的代码,执行结果却不一样,好诡异啊!到现在我都没有想明白这是为什么。如果有谁知道原因,请告诉我,O(∩_∩)O谢谢!

好了!这一篇介绍coroutine的Blog也写好了。接下来的文章会以完整的代码的形式来介绍coroutine的一些高级用法。敬请期待。O(∩_∩)O哈哈~

 

Python高级编程之生成器(Generator)与coroutine(二):coroutine介绍

原文:http://www.cnblogs.com/rio2607/p/4456332.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!