β

python协程使用

茶瓯葱丝 259 阅读

协程是一种比线程更轻量级的执行体。它不是有系统内核控制调度,而是自己协同调度。Python通过yield提供了对协程的基本支持,但是不完全。而第三方的gevent为Python提供了比较完善的协程支持。

from gevent import monkey
# 由于切换是在IO操作时自动完成,所以gevent需要修改Python自带的一些标准库,这一过程在启动时通过monkey
monkey.patch_socket()
import gevent
def f(n):
    for i in range(n):
        print gevent.getcurrent(), i
        gevent.sleep(0) # 要想协程起作用,可以通过gevent.sleep()交出控制权
g1 = gevent.spawn(f, 5)
g2 = gevent.spawn(f, 5)
g3 = gevent.spawn(f, 5)
g1.join()
g2.join()
g3.join()
output:
<Greenlet at 0x10bc44c30: f(5)> 0
<Greenlet at 0x10bd600f0: f(5)> 0
<Greenlet at 0x10bd60190: f(5)> 0
<Greenlet at 0x10bc44c30: f(5)> 1
<Greenlet at 0x10bd600f0: f(5)> 1
<Greenlet at 0x10bd60190: f(5)> 1
<Greenlet at 0x10bc44c30: f(5)> 2
<Greenlet at 0x10bd600f0: f(5)> 2
<Greenlet at 0x10bd60190: f(5)> 2
<Greenlet at 0x10bc44c30: f(5)> 3
<Greenlet at 0x10bd600f0: f(5)> 3
<Greenlet at 0x10bd60190: f(5)> 3
<Greenlet at 0x10bc44c30: f(5)> 4
<Greenlet at 0x10bd600f0: f(5)> 4
<Greenlet at 0x10bd60190: f(5)> 4

可以看到协程的交替运行了。

在实际的使用中,我们不需要使用gevent.sleep去控制,而是在遇到io时候,gevent会自动控制。

from gevent import monkey
# 由于切换是在IO操作时自动完成,所以gevent需要修改Python自带的一些标准库,这一过程在启动时通过monkey
monkey.patch_socket()
import gevent
import urllib2
def f(url):
    print 'GET: %s' % url
    resp = urllib2.urlopen(url)
    data = resp.read()
    print '%d bytes received from %s.' % (len(data), url)
gevent.joinall([
    gevent.spawn(f, 'http://www.baidu.com/'),
    gevent.spawn(f, 'http://www.taobao.com/'),
    gevent.spawn(f, 'http://www.163.com/'),
])
output:
GET: http://www.baidu.com/
GET: http://www.taobao.com/
GET: http://www.163.com/
 bytes received from http://www.taobao.com/.
 bytes received from http://www.baidu.com/.
 bytes received from http://www.163.com/.

可以看出在同一个进程中执行,但是顺序却不同了,可见网络io的时候是并发执行的。

作者:茶瓯葱丝
本站内容为个人技术博客及其它软件编程信息。本站采用python开发,用markdown来进行编辑,生成静态文本放在github上。
原文地址:python协程使用, 感谢原作者分享。

发表评论