关于学习Python的协程以及生成器的一点理解

从廖大的教程说起

  看廖大的教程学习Python的时候，学到协程这里[https://www.liaoxuefeng.com/wiki/0014316089557264a6b348958f449949df42a6d3a2e542c000/001432090171191d05dae6e129940518d1d6cf6eeaaa969000]，被yield卡了很久，想明白之后，写下这篇博客，记录一下理解思路。

  廖大的源码是下面这样的：

def consumer():
    r = ''
    while True:
        n = yield r
        if not n:
            return
        print('[CONSUMER] Consuming %s...' % n)
        r = '200 OK'

def produce(c):
    c.send(None)
    n = 0
    while n < 5:
        n = n + 1
        print('[PRODUCER] Producing %s...' % n)
        r = c.send(n)
        print('[PRODUCER] Consumer return: %s' % r)
    c.close()

c = consumer()
produce(c)

  程序执行后输出是这样的：

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

分析与理解

  其实主要想不明白的就是

n=yield r

  应该是之前的生成器那边学的不透彻，generator函数遇到yield语句就返回，所以当produce函数那里第一次执行c.send(None)之后在consumer函数那里执行了yield r之后就返回了，注意这时n并没有被定义或赋值，暂时理解是因为send的是None，函数直接返回了（这可能是比较难理解的地方），当下一次执行c.send(1)时，consumer函数这里就执行n的赋值了，此时n 等于send传过来的值，并走完一个循环，直到r=’200OK’之后进入下一个循环，此时consumer执行yield r后返回。

  下面自己写了一段程序便于自己理解：

def consumer():
    r = '这是第一次yield r返回的内容'
    while True:
        #第一次执行yield r之后函数就返回了
        #第二次执行到这里是执行的yield r之后的语句
        #也就是对n的赋值，n拿到的是csm.send(m)中的m
        #所以这里的yield作为一个双面侠，既能返回r，又能传值
        n = yield r
        if not n:
            return
        print('[CONSUMER] Consuming %s...' % n)
        r = n

def produce(csm):
    #send(None)作用是启动生成器，触发consumer函数里yield r语句
    #我是这样理解的，由于send的是None，所以相当于对n什么也没干
    #到下次send时才有值传给n
    test = csm.send(None)
    #这是consumer函数第一次执行yield r之后返回的内容
    print(test)
    #为了与上面的n区分开，我用了不同的命名
    m = 0
    while m < 3:
        m = m + 1
        print('[PRODUCER] Producing %s...' % m)
        result = csm.send(m)
        print('[PRODUCER] Consumer return: %s' % result)
    csm.close()

c = consumer()
produce(c)

  输出的内容如下：

这是第一次yield r返回的内容
[PRODUCER] Producing 1...
[CONSUMER] Consuming 1...
[PRODUCER] Consumer return: 1
[PRODUCER] Producing 2...
[CONSUMER] Consuming 2...
[PRODUCER] Consumer return: 2
[PRODUCER] Producing 3...
[CONSUMER] Consuming 3...
[PRODUCER] Consumer return: 3

  好了，基本就是这样了