【python】协程（greenlet、gevent）的简单使用

以你之姓@ 2022-11-25 10:20 215阅读 0赞

#### 1.协程 ####

协程，又称微线程，纤程（Coroutine），相比进程和线程，协程的开销最小

协程是在一个线程中执行多个函数，当一个函数遇到耗时操作（比如说网络IO、文件读写），它会自动切换到其他函数继续执行，之后再切换回来继续执行任务，因为没有切换线程的开销，所以执行效率更高

#### 2.greenlet ####

greenlet是一个封装的协程库，使用方法很简单，自由控制切换  
**安装**

pip install greenlet

**使用**

from greenlet import greenlet
    import time
    
    def run():
        while True:
            print("running...")
            g2.switch()  # 切换到g2
            time.sleep(0.5)
    
    def eat():
        while True:
            print("eating...")
            g1.switch()  # 切换到g1
            time.sleep(0.5)
    
    if __name__ == "__main__":
        g1 = greenlet(run)
        g2 = greenlet(eat)
        g1.switch()  # 切换到g1

#### 3.gevent ####

greenlet用起来很简单，但是需要手动控制协程切换，gevent也是一个第三方的库，但它可以自动切换任务，可以轻松实现协程

**安装**

pip install gevent

**使用**  
gevent默认是不能识别耗时操作的，比如说遇到time.sleep(0.5)不会自动切换任务，需要用gevent.sleep(0.5)才能让它认为是耗时操作

为了让协程更好地工作，我们可以给它打上monkey补丁让它识别耗时任务

import gevent
    from gevent import monkey
    import time
    
    # 打补丁，可以让gevent识别所有耗时操作，以便自动切换
    monkey.patch_all()
    
    
    def task(name):
        for i in range(5):
            print(name, '在执行任务')
            time.sleep(1)  # 如果不打monkey补丁，需要用gevent.sleep(0.5)
    
    
    if __name__ == '__main__':
        # 创建协程
        g1 = gevent.spawn(task, 'xiaoming')
        g2 = gevent.spawn(task, 'hong')
    
        # join，让主线程等到（相当于执行耗时操作）
        g1.join()
        g2.join()