2.1 yield 生成器

Python 中通过 yield 来实现类似协程的功能,下面来看一个简单的 demo:

  1. def task1():
  2. while True:
  3. print("----task1---")
  4. yield
  5. time.sleep(2)
  6. def task2():
  7. while True:
  8. print("----task2---")
  9. yield
  10. time.sleep(0.5)
  11. def main():
  12. w1 = task1()
  13. w2 = task2()
  14. while True:
  15. next(w1)
  16. next(w2)
  17. if __name__ == "__main__":
  18. main()

输出结果:

  1. ----task1---
  2. ----task2---
  3. ----task1---
  4. ----task2---
  5. ...

2.2 区别和相似点

协程和生成器主要有以下差异:

  • 生成器总是生成值,一般用于迭代的序列
  • 协程关注的是消耗值,是数据的消费者
  • 协程不会迭代操作关联,但是生成器会
  • 协程强调协同控制程序流,生成器强调保存状态和产生数据

相似点:

  • 都不要 return 来实现重复调用的函数或对象
  • 都通过 yield 方式来实现