对lua协程的一点理解

OStack程序员社区-中国程序员成长平台 › 门户 › 编程› LUA›LUA教程

原作者: [db:作者] 来自: [db:来源] 收藏邀请

读《Programming In Lua》协程那一章，比较困惑的还是procuer-consumer那个例子：

function consumer(prod)
    while true do
        local x = receive(prod)
        print(x)
    end
end

function receive(prod)
    local status, value = coroutine.resume(prod)
    return value
end

function send(x)
    coroutine.yield(x) -- go back to where resumed
end

function producer()
    return coroutine.create(function()
        while true do
            local x = io.read()
            send(x)
        end
    end)
end

-- consumer-driven design
consumer(producer())

producer产生数据，consumer消费数据，producer与consumer都在各自的协程中完成，代码很短，但是很难读 - 至少不是那么一目了然，尤其比起这个直接的循环：

function produce()
    return io.read()
end

function consume(x)
    print(x)
end

while true do
    local x = produce()
    consume(x)
end

好在哪里？

书中说可以添加缓存控制速度，或者进行数据过滤 - 但是这在循环版本的producer-consumer中也都能做到，无非在在实现produce是多些逻辑，或者再加个filter的函数处理produce的返回值，协程版本毫无优势可言。

实在要说其优点，恐怕只是：producer和consumer的代码在各自的协程中实现，并通过resume-yield交换数据 - 实现了松耦合。这是个优点，可以还是不太满意，再往下看时，看到了全排列那个例子，稍作修改，让我比较直观的感觉到了协程这种控制结构的灵活性：

function send(a)
    coroutine.yield(a)
end

function receive(prod)
    local status, value = coroutine.resume(prod)
    return value
end

function consumer(prod)
    local function print_result(a)
        for _, v in ipairs(a) do
            io.write(v, " ")
        end
        io.write('\n')
    end
    
    while true do
        local a = receive(prod)
        if not a then break end
        print_result(a)
    end
end


function producer(a)
    function permgen(a, n)
        if n == 0 then
            send(a) -- send something for consuming from an recursive call
        else
            for i=1, n do
                a[n], a[i] = a[i], a[n]
                permgen(a, n-1)
                a[n], a[i] = a[i], a[n]
            end
        end
    end

    local n = table.getn(a)
    return coroutine.create(function() permgen(a, n) end)
end

consumer(producer({1,2,3,4}))