快捷導(dǎo)航

Lua中的協(xié)同程序詳解

更新時(shí)間：2014年09月28日 13:03:09 投稿：junjie

這篇文章主要介紹了Lua中的協(xié)同程序詳解,本文非常詳細(xì)的講解了Lua中的協(xié)同程序,同時(shí)講解了生產(chǎn)者-消費(fèi)者問題,需要的朋友可以參考下

前言

協(xié)同程序與線程差不多，也就是一條執(zhí)行序列，擁有自己獨(dú)立的棧、局部變量和指令指針，同時(shí)又與其它協(xié)同程序共享全局變量和其它大部分東西。從概念上講，線程與協(xié)同程序的主要區(qū)別在于，一個(gè)具有多個(gè)線程的程序可以同時(shí)運(yùn)行幾個(gè)線程，而協(xié)同程序卻需要彼此協(xié)作的運(yùn)行。就是說，一個(gè)具有多個(gè)協(xié)同程序的程序在任意時(shí)刻只能運(yùn)行一個(gè)協(xié)同程序，并且正在運(yùn)行的協(xié)同程序只會(huì)在其顯式地要求掛起時(shí)，它的執(zhí)行才會(huì)暫停。

協(xié)同程序基礎(chǔ)

Lua將所有關(guān)于協(xié)同程序的函數(shù)放置在一個(gè)名為“coroutine”的table中。函數(shù)create用于創(chuàng)建新的協(xié)同程序，它只有一個(gè)參數(shù)，就是一個(gè)函數(shù)。該函數(shù)的代碼就是協(xié)同程序需要執(zhí)行的內(nèi)容。create會(huì)返回一個(gè)thread類型的值，用以表示新的協(xié)同程序，一般create的參數(shù)是一個(gè)匿名函數(shù)，例如以下代碼：

復(fù)制代碼代碼如下:

local co = coroutine.create(function () print("Hello WOrld") end)

一個(gè)協(xié)同程序可以有四種不同的狀態(tài)：掛起（suspended）、運(yùn)行（running）、死亡（dead）和正常（normal）。當(dāng)新創(chuàng)建一個(gè)協(xié)同程序時(shí)，它處于掛起狀態(tài)，言外之意就是，協(xié)同程序不會(huì)在創(chuàng)建它時(shí)自動(dòng)執(zhí)行其內(nèi)容，我們可以通過函數(shù)status來檢查協(xié)同程序的狀態(tài)。

復(fù)制代碼代碼如下:

local co = coroutine.create(function () print("Hello WOrld") end)

print(coroutine.status(co))     -- suspended

函數(shù)coroutine.resume用于啟動(dòng)或再次啟動(dòng)一個(gè)協(xié)同程序的執(zhí)行，并將其狀態(tài)由掛起改為運(yùn)行：

復(fù)制代碼代碼如下:

local co = coroutine.create(function () print("Hello WOrld") end)

print(coroutine.status(co))     -- suspended

coroutine.resume(co)          -- Hello World

上面的代碼中，我調(diào)用了resume函數(shù)，將協(xié)同程序co由suspended改為running狀態(tài)，當(dāng)打印了Hello World之后，協(xié)同程序co就處于死亡狀態(tài)。

到目前為止，協(xié)同程序就是一種函數(shù)調(diào)用。其實(shí)，協(xié)同程序的真正強(qiáng)大之處在于函數(shù)yield的使用上，該函數(shù)可以讓一個(gè)運(yùn)行中的協(xié)同程序掛起，而之后可以再恢復(fù)它的運(yùn)行，例如以下代碼：

復(fù)制代碼代碼如下:

local co = coroutine.create(function ()

     for i = 1, 10 do

          print("co", i)

          coroutine.yield()

     end

end)

-- 打印初始狀態(tài)

print(coroutine.status(co))     -- suspended

-- 喚醒協(xié)同程序co

coroutine.resume(co)     -- 打印co 1

-- 打印協(xié)同程序的狀態(tài)

print(coroutine.status(co))     -- suspended

-- 再次喚醒協(xié)同程序co

coroutine.resume(co)     -- 打印co 2

-- 打印協(xié)同程序的狀態(tài)

print(coroutine.status(co))     -- suspended

coroutine.resume(co)     -- 打印co 3

coroutine.resume(co)     -- 打印co 4

coroutine.resume(co)     -- 打印co 5

coroutine.resume(co)     -- 打印co 6

coroutine.resume(co)     -- 打印co 7

coroutine.resume(co)     -- 打印co 8

coroutine.resume(co)     -- 打印co 9

coroutine.resume(co)     -- 打印co 10

coroutine.resume(co)     -- 什么都不打印

print(coroutine.status(co))     -- dead

coroutine.resume(co)

當(dāng)在協(xié)同程序的執(zhí)行中發(fā)生任何錯(cuò)誤，Lua是不會(huì)顯示錯(cuò)誤消息的，而是將執(zhí)行權(quán)返回給resume調(diào)用。當(dāng)coroutine.resume的第一個(gè)返回值為false時(shí)，就表明協(xié)同程序在運(yùn)行過程中發(fā)生了錯(cuò)誤；當(dāng)值為true時(shí)，則表明協(xié)同程序運(yùn)行正常。

當(dāng)一個(gè)協(xié)同程序A喚醒另一個(gè)協(xié)同程序B時(shí)，協(xié)同程序A就處于一個(gè)特殊狀態(tài)，既不是掛起狀態(tài)（無法繼續(xù)A的執(zhí)行），也不是運(yùn)行狀態(tài)（是B在運(yùn)行）。所以將這時(shí)的狀態(tài)稱為“正常”狀態(tài)。

Lua的協(xié)同程序還具有一項(xiàng)有用的機(jī)制，就是可以通過一對(duì)resume-yield來交換數(shù)據(jù)。在第一次調(diào)用resume時(shí)，并沒有對(duì)應(yīng)的yield在等待它，因此所有傳遞給resume的額外參數(shù)都視為協(xié)同程序主函數(shù)的參數(shù)。如下述代碼：

當(dāng)協(xié)同程序中沒有yield時(shí)，第一次調(diào)用resume，所有傳遞給resume的額外參數(shù)都將視為協(xié)同程序主函數(shù)的參數(shù)，如以下代碼：

復(fù)制代碼代碼如下:

local co = coroutine.create(function (a, b, c)

     print("co", a, b, c)

end)

coroutine.resume(co, 1, 2, 3)     -- co 1 2 3

當(dāng)協(xié)同程序中存在yield時(shí)，一切就變的復(fù)雜了，先來分析一下這個(gè)流程：

1.調(diào)用resume，將協(xié)同程序喚醒；
2.協(xié)同程序運(yùn)行；
3.運(yùn)行到y(tǒng)ield語句；
4.yield掛起協(xié)同程序，第一次resume返回；（注意：此處yield返回，參數(shù)是resume的參數(shù)）
5.第二次resume，再次喚醒協(xié)同程序；（注意：此處resume的參數(shù)中，除了第一個(gè)參數(shù)，剩下的參數(shù)將作為yield的參數(shù)）
6.yield返回；
7.協(xié)同程序繼續(xù)運(yùn)行；

此處從其它博客中借鑒的一部分代碼，可以說明上面的調(diào)用流程：

復(fù)制代碼代碼如下:

function foo (a)

    print("foo", a)  -- foo 2

    return coroutine.yield(2 * a) -- return 2 * a

end

co = coroutine.create(function (a , b)

    print("co-body", a, b) -- co-body 1 10

    local r = foo(a + 1)

    print("co-body2", r)

    local r, s = coroutine.yield(a + b, a - b)

    print("co-body3", r, s)

    return b, "end"

end)

print("main", coroutine.resume(co, 1, 10)) -- true, 4

print("------")

print("main", coroutine.resume(co, "r")) -- true 11 -9

print("------")

print("main", coroutine.resume(co, "x", "y")) -- true 10 end

print("------")

print("main", coroutine.resume(co, "x", "y")) -- false cannot resume dead coroutine

print("------")

輸出結(jié)果如下：

復(fù)制代碼代碼如下:

>lua -e "io.stdout:setvbuf 'no'" "test.lua"

co-body     1     10

foo     2

main     true     4

------

co-body2     r

main     true     11     -9

------

co-body3     x     y

main     true     10     end

------

main     false     cannot resume dead coroutine

------

>Exit code: 0

resume和yield的配合強(qiáng)大之處在于，resume處于主程中，它將外部狀態(tài)（數(shù)據(jù)）傳入到協(xié)同程序內(nèi)部；而yield則將內(nèi)部的狀態(tài)（數(shù)據(jù)）返回到主程中。

生產(chǎn)者-消費(fèi)者問題

現(xiàn)在我就使用Lua的協(xié)同程序來完成生產(chǎn)者-消費(fèi)者這一經(jīng)典問題。生產(chǎn)者生產(chǎn)東西，消費(fèi)者消費(fèi)生產(chǎn)者生產(chǎn)的東西。

復(fù)制代碼代碼如下:

local newProductor

function productor()

     local i = 0

     while true do

          i = i + 1

          send(i)     -- 將生產(chǎn)的物品發(fā)送給消費(fèi)者

     end

end

function consumer()

     while true do

          local i = receive()     -- 從生產(chǎn)者那里得到物品

          print(i)

     end

end

function receive()

     local status, value = coroutine.resume(newProductor)

     return value

end

function send(x)

     coroutine.yield(x)     -- x表示需要發(fā)送的值，值返回以后，就掛起該協(xié)同程序

end

-- 啟動(dòng)程序

newProductor = coroutine.create(productor)

consumer()