七、Lua中的协同程序

  • Post author:
  • Post category:其他




Lua中的协同程序|果冻想



本文是对于“果冻想”的《Lua入门系列》所作笔记,原文来自上面链接。

协同程序基础

Lua将所有关于协同程序的函数放置在一个名为“coroutine”的table中。函数create用于创建新的协同程序,它只有一个参数,就是一个函数。该函数的代码就是协同程序需要执行的内容。create会返回一个thread类型的值,用以表示 新的协同程序,一般create的参数是一个匿名函数,例如以下代码:
   
   
  1. local co = coroutine.create(function() print("hello world") end)

一个协同程序可以有四种不同的状态:挂起(suspended)、运行(running)、死亡(dead)和正常(normal)。当新创建一个协同程序时,它处于挂起状态。

   
   
  1. local co = coroutine.create(function() print("hello world") end)
  2. print(coroutine.status(co)) -- suspended

函数coroutine.resume用以启动或再次启动一个协同程序的执行,并将其状态有挂起改为运行:

   
   
  1. local co = coroutine.create(function() print("hello world") end)
  2. print(coroutine.status(co)) -- suspended
  3. coroutine.resume(co) -- hello world

调用了resume函数,将协同程序由suspended改为running状态,当打印了hello world之后,协同程序就处于死亡状态。

协同程序的真正强大之处在于函数yield的使用上,该函数可以让一个运行中的协同程序挂起,而之后可以再恢复它的运行。
   
   
  1. local co = coroutine.create(function ()
  2. for i = 1, 3 do
  3. print("co", i)
  4. coroutine.yield()
  5. end
  6. end)
  7. -- 打印初始状态
  8. print(coroutine.status(co)) -- suspended
  9. -- 唤醒协同程序co
  10. coroutine.resume(co) -- 打印co 1
  11. -- 打印协同程序的状态
  12. print(coroutine.status(co)) -- suspended
  13. -- 再次唤醒协同程序co
  14. coroutine.resume(co) -- 打印co 2
  15. -- 打印协同程序的状态
  16. print(coroutine.status(co)) -- suspended
  17. coroutine.resume(co) -- 打印co 3
  18. coroutine.resume(co) -- 什么都不打印
  19. print(coroutine.status(co)) -- dead
  20. coroutine.resume(co)

Lua的协同程序可以通过一对resume-yield来交换数据。在第一次调用resume时,并没有对应的yield在等待它,因此所有传递给resume的额外参数都视为协同程序主函数的参数:
   
   
  1. local co = coroutine.create(function(a, b, c)
  2. print("co", a, b, c)
  3. end)
  4. coroutine.resume(co, 1, 2, 3) -- co 1 2 3

当协同程序中存在yield时,一切就变的复杂了,先来分析一下这个流程:
1、调用resume,将协同程序唤醒;
2、协同程序运行;
3、运行到yield语句;
4、yield挂起协同程序,第一次resume返回;(注意:此处yield返回,参数是resume的参数)
5、第二次resume,再次唤醒协同程序;(注意:此处resume的参数中,除了第一个参数,剩下的参数将作为yield的参数)
6、yield返回;
7、协同程序继续运行;

   
   
  1. function foo (a)
  2. print("foo", a) -- foo 2
  3. return coroutine.yield(2 * a) -- return 2 * a
  4. end
  5. co = coroutine.create(function (a , b)
  6. print("co-body", a, b) -- co-body 1 10
  7. local r = foo(a + 1)
  8. print("co-body2", r)
  9. local r, s = coroutine.yield(a + b, a - b)
  10. print("co-body3", r, s)
  11. return b, "end"
  12. end)
  13. print("main", coroutine.resume(co, 1, 10)) -- true, 4
  14. print("------")
  15. print("main", coroutine.resume(co, "r")) -- true 11 -9
  16. print("------")
  17. print("main", coroutine.resume(co, "x", "y")) -- true 10 end
  18. print("------")
  19. print("main", coroutine.resume(co, "x", "y")) -- false cannot resume dead coroutine
  20. print("------")

生产者-消费者问题

   
   
  1. local newProductor
  2. function productor()
  3. local i = 0
  4. while true do
  5. i = i + 1
  6. send(i) -- 将生产的物品发送给消费者
  7. end
  8. end
  9. function consumer()
  10. for i = 1,10 do
  11. local status, value = coroutine.resume(newProductor) -- 从生产者那里得到物品
  12. print(value)
  13. end
  14. end
  15. function send(x)
  16. coroutine.yield(x) -- x表示需要发送的值,值返回以后,就挂起该协同程序
  17. end
  18. -- 启动程序
  19. newProductor = coroutine.create(productor)
  20. consumer()



版权声明:本文为chenbingzhao原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。