Python并发编程之协程（greenlet、gevent）

偏执的太偏执、 2022-11-27 10:15 372阅读 0赞

友情链接：

*  [Python并发编程之协程【必知必会】][Python]

### 文章目录 ###

*   *  一、greenlet
     *   *  1.1 安装方式
         *  1.2 greenlet的使用
     *  二、gevent
     *   *  2.1 安装方式
         *  2.2 gevent的使用
         *  2.3 gevent切换执行
         *  2.4 给程序打补丁

## 一、greenlet ##

为了更好使用协程来完成多任务，Python中的`greenlet`模块对其封装，从而使得切换任务变的更加简单。

### 1.1 安装方式 ###

使用如下命令安装`greenlet`模块：

sudo pip3 install greenlet

### 1.2 greenlet的使用 ###

from greenlet import greenlet
    import time
    
    
    def test1():
        while True:
            print("---A--")
            gr2.switch()
            time.sleep(0.5)
    
    
    def test2():
        while True:
            print("---B--")
            gr1.switch()
            time.sleep(0.5)
    
    
    gr1 = greenlet(test1)
    gr2 = greenlet(test2)
    
    # 切换到gr1中运行
    gr1.switch()

![在这里插入图片描述][20200814213914200.png_pic_left]

## 二、gevent ##

`greenlet`已经实现了协程，但是这个还需要人工切换，比较麻烦。Python为了更好的实现协程，还有一个比`greenlet`更强大的并且能够自动切换任务的模块`gevent`。

其原理是当一个`greenlet`遇到IO（指的是input output 输入输出，比如网络、文件操作等）操作时，比如访问网络，就自动切换到其他的`greenlet`，等到IO操作完成，再在适当的时候切换回来继续执行。

由于IO操作非常耗时，经常使程序处于等待状态，有了`gevent`为我们自动切换协程，就保证总有`greenlet`在运行，而不是等待IO。

### 2.1 安装方式 ###

pip3 install gevent

### 2.2 gevent的使用 ###

`gevent.getcurrent()：保存该协程的相关信息。`

import gevent
    
    
    def f(n):
        for i in range(n):
            print(gevent.getcurrent(), i)
    
    
    g1 = gevent.spawn(f, 5)
    g2 = gevent.spawn(f, 5)
    g3 = gevent.spawn(f, 5)
    g1.join()
    g2.join()
    g3.join()

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0MDM0Mzg0_size_16_color_FFFFFF_t_70_pic_left]  
可以看到，3个`greenlet`是依次运行而不是交替运行。

### 2.3 gevent切换执行 ###

import gevent
    
    
    def f(n):
        for i in range(n):
            print(gevent.getcurrent(), i)
            # 用来模拟一个耗时操作，注意不是time模块中的sleep
            gevent.sleep(1)
    
    
    g1 = gevent.spawn(f, 5)
    g2 = gevent.spawn(f, 5)
    g3 = gevent.spawn(f, 5)
    g1.join()
    g2.join()
    g3.join()

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0MDM0Mzg0_size_16_color_FFFFFF_t_70_pic_left 1]

### 2.4 给程序打补丁 ###

from gevent import monkey
    import gevent
    import random
    import time
    
    
    def coroutine_work(coroutine_name):
        for i in range(10):
            print(coroutine_name, i)
            time.sleep(random.random())
    
    
    gevent.joinall([
        gevent.spawn(coroutine_work, "work1"),
        gevent.spawn(coroutine_work, "work2")
    ])

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0MDM0Mzg0_size_16_color_FFFFFF_t_70_pic_left 2]

from gevent import monkey
    import gevent
    import random
    import time
    
    # 有耗时操作时需要
    monkey.patch_all()  # 将程序中用到的耗时操作的代码，换为gevent中自己实现的模块
    
    
    def coroutine_work(coroutine_name):
        for i in range(10):
            print(coroutine_name, i)
            time.sleep(random.random())
    
    
    gevent.joinall([
        gevent.spawn(coroutine_work, "work1"),
        gevent.spawn(coroutine_work, "work2")
    ])

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0MDM0Mzg0_size_16_color_FFFFFF_t_70_pic_left 3]

[Python]: https://luckylifes.blog.csdn.net/article/details/108014107
[20200814213914200.png_pic_left]: /images/20221124/5670601ad28441168ef602d3e11debc7.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0MDM0Mzg0_size_16_color_FFFFFF_t_70_pic_left]: /images/20221124/31207819951d4010a13a8138f0c50c38.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0MDM0Mzg0_size_16_color_FFFFFF_t_70_pic_left 1]: /images/20221124/c27f0c783770400699694133907bfe55.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0MDM0Mzg0_size_16_color_FFFFFF_t_70_pic_left 2]: /images/20221124/91ce3e7416f54f51a17e4fb7c0ae63a5.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0MDM0Mzg0_size_16_color_FFFFFF_t_70_pic_left 3]: /images/20221124/568ee8fc004d409fa685643bca890575.png