python协程

最新推荐文章于 2024-04-30 11:41:18 发布

蓝鲸123

最新推荐文章于 2024-04-30 11:41:18 发布

阅读量104

点赞数

分类专栏： python

本文链接：https://blog.csdn.net/TH_NUM/article/details/115741495

版权

python 专栏收录该内容

94 篇文章 1 订阅

订阅专栏

这篇博客介绍了Python中的协程（Coroutine）概念，通过一个简单的例子展示了如何使用生成器实现协程。协程允许在子程序内部中断并切换到其他子程序执行，之后再返回继续执行，实现了无锁的并发，提高了效率。示例中，`consumer`和`produce`两个协程通过`yield`和`send`协同工作，模拟了一个生产者-消费者模型。

摘要由CSDN通过智能技术生成

协程，又称微线程，纤程。英文名Coroutine。

协程的概念很早就提出来了，但直到最近几年才在某些语言（如Lua）中得到广泛应用。

子程序，或者称为函数，在所有语言中都是层级调用，比如A调用B，B在执行过程中又调用了C，C执行完毕返回，B执行完毕返回，最后是A执行完毕。

所以子程序调用是通过栈实现的，一个线程就是执行一个子程序。

子程序调用总是一个入口，一次返回，调用顺序是明确的。而协程的调用和子程序不同。

协程看上去也是子程序，但执行过程中，在子程序内部可中断，然后转而执行别的子程序，在适当的时候再返回来接着执行。

#coding=UTF-8
import logging
from tensorflow.python.framework import device
import functools
import six
import abc
from  contextlib import  contextmanager

# 通过yield变成generator
def consumer():
    r = ''
    while True:
        n = yield r
        if not n:
            return
        print('[CONSUMER] Consuming %s...' % n)
        r = '200 OK'
k = consumer()

def produce(c):
  c.send(None)
  n = 0
  for i in range(5):
    n = n + 1
    print('[PRODUCER] Producing %s...' % i)
    # send 执行的次数
    r = c.send(i+1)
    print('[PRODUCER] Consumer return: %s' % r)
  c.close()

produce(k)

输出：

[PRODUCER] Producing 0...
[CONSUMER] Consuming 1...
[PRODUCER] Consumer return: 200 OK
[PRODUCER] Producing 1...
[CONSUMER] Consuming 2...
[PRODUCER] Consumer return: 200 OK
[PRODUCER] Producing 2...
[CONSUMER] Consuming 3...
[PRODUCER] Consumer return: 200 OK
[PRODUCER] Producing 3...
[CONSUMER] Consuming 4...
[PRODUCER] Consumer return: 200 OK
[PRODUCER] Producing 4...
[CONSUMER] Consuming 5...
[PRODUCER] Consumer return: 200 OK

注意到consumer函数是一个generator，把一个consumer传入produce后：

首先调用c.send(None)启动生成器；

然后，一旦生产了东西，通过c.send(n)切换到consumer执行；

consumer通过yield拿到消息，处理，又通过yield把结果传回；

produce拿到consumer处理的结果，继续生产下一条消息；

produce决定不生产了，通过c.close()关闭consumer，整个过程结束。

整个流程无锁，由一个线程执行，produce和consumer协作完成任务，所以称为“协程”，而非线程的抢占式多任务。