基础入门_Python-模块和包.Gevent事件/队列/组/池/信号量/子进程?

常用结构:

1.Event类,事件主要用于Greenlet之间的异步通信

e = gevent.event.Event() -> Event

说明: 创建一个信号对象

e.set() -> None

说明: 设置标志位

e.clear() -> None

说明: 清除标志位

e.wait() -> None

说明: 阻塞直至标志位被设置

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
#
# Authors: limanman
# OsChina: http://xmdevops.blog.51cto.com/
# Purpose:
#
"""
# 说明: 导入公共模块
import gevent
from gevent.event import Event
# 说明: 导入其它模块
def setter(e):
    print ‘I‘
    gevent.sleep(3)
    print ‘LOVE‘
    e.set()
def waiter(e):
    e.wait()
    print ‘You‘
if __name__ == ‘__main__‘:
    e = Event()
    gevent.joinall([
        gevent.spawn(setter, e),
        gevent.spawn(waiter, e),
        gevent.spawn(waiter, e),
        gevent.spawn(waiter, e),
        gevent.spawn(waiter, e)
    ])

说明: 创建一个扩展可携带数据的信号对象

a.set(value=None) -> None

说明: 设置带数据的标志位

a.e.clear() -> None

说明: 清除带数据的标志位

a.get(block=True, timeout=None) -> obj

说明: 阻塞直至标志位被设置并返回value值,可设置timeout到点抛出Timeout异常

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
#
# Authors: limanman
# OsChina: http://xmdevops.blog.51cto.com/
# Purpose:
#
"""
# 说明: 导入公共模块
import gevent
from gevent.event import AsyncResult
# 说明: 导入其它模块
def setter(a):
    print ‘I‘
    gevent.sleep(3)
    print ‘LOVE‘
    a.set(‘I LOVE‘)
def waiter(a):
    value = a.get()
    if e.successful:
        print ‘found notice: recv data from setter %s.‘ % (value,)
    print ‘You‘
if __name__ == ‘__main__‘:
    a = AsyncResult()
    gevent.joinall([
        gevent.spawn(setter, a),
        gevent.spawn(waiter, a),
        gevent.spawn(waiter, a),
        gevent.spawn(waiter, a),
        gevent.spawn(waiter, a)
    ])

q = gevent.queue.Queue(maxsize=None, items=None) -> Queue

说明: 创建一个指定大小包含指定items的队列对象

q.empty() -> Boolean

说明: 队列是否为空

q.full() -> Boolean

说明: 队列是否已满,如果初始化时maxsize为None时队列永远不会满,除非内存耗尽

q.get(block=True, timeout=None) -> obj

说明: 队列尾部弹出并返回末尾元素,block为True如果队列为空会一直等待,否则会抛出gevent.queue.Empty异常

q.get_nowait() -> obj

说明: 同q.get(block=False),如果队列为空直接返回gevent.queue.Empty

q.put(item, block=True, timeout=None) -> None

说明: 队列头部插入item,如果block为True则等待队列有空间时再插入,如果超出timeout则抛出Timeout异常,否则直接抛出gevent.queue.Full异常

q.put_nowait(self, item) -> None

说明: 同q.put(block=False),如果队列满直接返回gevent.queue.Full

q.qsize() -> int

说明: 返回队列的元素数

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
#
# Authors: limanman
# OsChina: http://xmdevops.blog.51cto.com/
# Purpose:
#
"""
# 说明: 导入公共模块
import gevent
from gevent.queue import Queue
# 说明: 导入其它模块
def boss(q):
    for _ in xrange(25):
        q.put_nowait(_)
def work(name, q):
    while not q.empty():
        task = q.get()
        print ‘found notice: worker %s got task-%s‘ % (name, task)
        gevent.sleep(0)
if __name__ == ‘__main__‘:
    q = Queue()
    g = gevent.spawn(boss, q)
    g.join()
    workers = []
    for _ in (‘limanman‘, ‘liuzhen‘, ‘zhenbao‘):
        workers.append(gevent.spawn(work, _, q))
    gevent.joinall(workers)

3. Group类,常用于不限制数量的管理异步任务的分组且可搜集运行结果

g = gevent.pool.Group(*args) -> Group

说明: 创建一个组对象,其实就是一个不限greenlets数量的pool,可以随时添加/删除/关闭greenlet对象

g.add(greenlet) -> None

说明: 向组中添加一个greenlet对象

g.discard(greenlet) -> None

说明: 从组中删除一个greenlet对象

g.join(timeout=None, raise_error=False) -> None

说明: 等待组中所有的greenlets都执行完毕再进行下一步

g.kill(exception=<class ‘greenlet.GreenletExit‘>, block=True, timeout=None) -> None

说明: 关闭组内所有运行中的greenlet对象

g.killone(self, greenlet, exception=<class ‘greenlet.GreenletExit‘>, block=True, timeout=None) -> None

说明: 关闭组内指定运行的greenlet对象

g.apply(self, func, args=None, kwds=None) -> obj

说明: 同apply单次调用func并返回运行结果

g.apply_async(self, func, args=None, kwds=None, callback=None) -> greenlet

说明: 同上但是返回的不是直接的运行结果而是异步对象greenlet,需要再次调用其.get()方法才能获取最终结果

g.imap(self, func, *iterables, **kwargs) -> IMap

g.imap_unordered(self, func, *iterables, **kwargs) -> IMapUnordered

g.map(self, func, iterable) -> list

g.map_async(self, func, iterable, callback=None) -> GreenletGroup

说明: 为了简化协程的使用,调用map/imap实现任务快速分组,基本上返回的对象都需要再次调用get()方法获取最终运行结果,经测试发现调用函数传递多个参数时通过map/imap类方法传参时有bug,已反馈GitHub.

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
#
# Authors: limanman
# OsChina: http://xmdevops.blog.51cto.com/
# Purpose:
#
"""
# 说明: 导入公共模块
import gevent
import itertools
from gevent.pool import Group
from gevent.queue import Queue
# 说明: 导入其它模块
def add_worker(name, q):
    while not q.empty():
        task = q.get()
        print ‘found notice: worker(%s:%s) got task(%s)‘ % (name, gevent.getcurrent(), task)
        gevent.sleep(0)
if __name__ == ‘__main__‘:
    q = Queue()
    g = Group()
    # 添加任务
    for _ in xrange(100):
        q.put_nowait(_)
    # 通用方法
    for worker in (‘langlang‘, ‘fengfeng‘, ‘shuishui‘):
        g.add(gevent.spawn(add_worker, worker, q))
    g.join()

p =  gevent.pool.Pool(size=None, greenlet_class=None) -> Pool

说明: 创建一个协程池对象,可以指定其实协程数

p.add(greenlet) -> None

说明: 尝试向协程池中添加greenlet对象,阻塞直到有池中有协程完毕有剩余空间

p.free_count() -> int

说明: 返回池中剩余的空间,也就是可以add的greenlet数

p.full() -> boolean

说明: 返回池是否已满

p.start(greenlet) -> None

说明: 启动未启动的greenlet且将其加入pool监控

p.imap(self, func, *iterables, **kwargs) -> IMap

p.imap_unordered(self, func, *iterables, **kwargs) -> IMapUnordered

p.map(self, func, iterable) -> list

p.map_async(self, func, iterable, callback=None) -> GreenletGroup

说明: 为了简化协程的使用,调用map/imap实现任务快速分组,基本上返回的对象都需要再次调用get()方法获取最终运行结果,经测试发现调用函数传递多个参数时通过map/imap类方法传参时有bug,已反馈GitHub.

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
#
# Authors: limanman
# OsChina: http://xmdevops.blog.51cto.com/
# Purpose:
#
"""
# 说明: 导入公共模块
import gevent
from gevent.pool import Pool
# 说明: 导入其它模块
def pool_size(p):
    print ‘found notice: current pool size: %d free %d‘ % (len(p), p.free_count())
if __name__ == ‘__main__‘:
    p = Pool(2)
    p.map(pool_size, [p]*3)

b = gevent.lock.BoundedSemaphore(n) -> BoundedSemaphore

说明: 创建一个信号量对象,限制资源同时只能被n个协程访问,其实就是每次分发n把锁,只有有人释放了锁才会重新追加分发锁

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
#
# Authors: limanman
# OsChina: http://xmdevops.blog.51cto.com/
# Purpose:
#
"""
# 说明: 导入公共模块
import gevent
from gevent.pool import Pool
from gevent.lock import BoundedSemaphore
# 说明: 导入其它模块
b = BoundedSemaphore(2)
def work(name):
    with b:
        print ‘found notice: worker %s acquire sem lock.‘ % (name,)
        gevent.sleep(0)
    print ‘found notice: worker %s release sem lock.‘ % (name,)
if __name__ == ‘__main__‘:
    p = Pool()
    p.map(work, [‘李满满‘, ‘刘珍珍‘, ‘罗诗瑶‘])

说明:  Gevent内部实现以greenlet的getcurrent()为键,在一个私有命名空间寻址的全局查找,使得local对象可以在不同的greenlet对象中存在且相互隔离,很多继承了gevent的web框架将HTTP会话对象以局部变量的形式存储在gevent内

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
#
# Authors: limanman
# OsChina: http://xmdevops.blog.51cto.com/
# Purpose:
#
"""
# 说明: 导入公共模块
import time
import gevent
from gevent.local import local
# 说明: 导入其它模块
def session_manager(l, user):
    l.name = user
    l.timestamp = time.time()
    print ‘found notice: user %s login in and online.‘ % (user,)
    gevent.sleep(5)
    l.name = user
    l.timestamp = time.time()
    print ‘found notice: user %s session expire outs.‘ % (user,)
    print ‘‘‘
    name: %s
    status: offline
    timestamp: %s
    ‘‘‘ % (l.name, l.timestamp)
if __name__ == ‘__main__‘:
    l = local()
    gevent.joinall([
        gevent.spawn(session_manager, l, ‘李满满‘),
        gevent.spawn(session_manager, l, ‘刘珍珍‘),
    ])

1. gevent.subprocess类,常用于作为subprocess模块增强版使用,支持异步协作式等待子进程

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
#
# Authors: limanman
# OsChina: http://xmdevops.blog.51cto.com/
# Purpose:
#
"""
# 说明: 导入公共模块
import time
import gevent
from gevent.subprocess import PIPE, Popen
# 说明: 导入其它模块
def task001():
    while True:
        print time.time()
        gevent.sleep(1)
if __name__ == ‘__main__‘:
    task = gevent.spawn(task001)
    task.start()
    # 中间插入子进程
    p = Popen([‘ls;sleep 5‘], shell=True, stdout=PIPE, stderr=PIPE)
    out, err = p.communicate()
    print out
    # 等待协程的结束
    task.join()