Lua 协同程序(coroutine)

用心去追梦

于 2024-08-10 10:53:55 发布

阅读量168

点赞数

文章标签： lua junit 开发语言

本文链接：https://blog.csdn.net/qq_33240556/article/details/141087593

版权

在 Lua 中，协同程序（coroutine）是一种轻量级的线程机制，允许程序在多个不同的控制点之间进行切换执行。协同程序在 Lua 中主要用于实现并发编程和协作式多任务处理。下面是如何在 Lua 中使用协同程序的示例和说明：

创建协同程序

在 Lua 中，你可以使用 coroutine.create 函数来创建一个新的协同程序。

示例

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

启动协同程序

你可以使用 coroutine.resume 函数来启动一个协同程序。

示例

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

coroutine.resume(co)  -- 输出 "Hello from coroutine!"

传递值

你可以通过 coroutine.resume 函数向协同程序传递值，并且可以从协同程序返回值。

示例

local co = coroutine.create(function(x, y)
    local sum = x + y
    print("Sum:", sum)
    return sum
end)

local result = coroutine.resume(co, 10, 5)
print("Result:", result)  -- 输出 "Result: 15"

使用 `yield` 和 `resume`

协同程序可以通过 coroutine.yield 函数暂停执行，并且可以通过 coroutine.resume 继续执行。

示例

local co = coroutine.create(function()
    print("Coroutine started.")
    local x = coroutine.yield(10)
    print("Received:", x)
    coroutine.yield(20)
    print("Coroutine finished.")
end)

local status, x = coroutine.resume(co)
print("Status:", status, "X:", x)  -- 输出 "Status: true X: 10"
status, x = coroutine.resume(co, "Data")
print("Status:", status, "X:", x)  -- 输出 "Status: true X: Data"
status, x = coroutine.resume(co)
print("Status:", status, "X:", x)  -- 输出 "Coroutine finished." "Status: true X: 20"

示例：使用协同程序实现生产者-消费者模型

下面是一个简单的示例，展示了如何使用协同程序来实现一个生产者-消费者模型。

local producer = coroutine.create(function(queue)
    print("Producer: Starting...")
    queue:put(1)
    queue:put(2)
    queue:put(3)
    print("Producer: Finished.")
end)

local consumer = coroutine.create(function(queue)
    print("Consumer: Starting...")
    while true do
        local item = queue:get()
        if item == nil then
            break
        end
        print("Consumer: Got item:", item)
    end
    print("Consumer: Finished.")
end)

local Queue = {}
Queue.__index = Queue

function Queue:new()
    local q = setmetatable({items = {}}, Queue)
    return q
end

function Queue:put(item)
    table.insert(self.items, item)
end

function Queue:get()
    if #self.items > 0 then
        return table.remove(self.items, 1)
    else
        return nil
    end
end

local queue = Queue:new()
coroutine.resume(producer, queue)
coroutine.resume(consumer, queue)

在这个示例中，我们创建了一个简单的队列来存储项，并使用两个协同程序来模拟生产者和消费者的行为。

示例：使用协同程序实现简单的任务调度

下面是一个简单的示例，展示了如何使用协同程序来实现一个简单的任务调度器。

local TaskScheduler = {}
TaskScheduler.__index = TaskScheduler

function TaskScheduler:new()
    local scheduler = setmetatable({
        tasks = {},
        current = nil
    }, TaskScheduler)
    return scheduler
end

function TaskScheduler:add(task)
    table.insert(self.tasks, task)
end

function TaskScheduler:run()
    while #self.tasks > 0 do
        local task = self.tasks[1]
        table.remove(self.tasks, 1)
        self.current = task
        coroutine.resume(task)
        self.current = nil
    end
end

function TaskScheduler:yield()
    if self.current then
        coroutine.yield()
    end
end

local scheduler = TaskScheduler:new()

local task1 = coroutine.create(function()
    print("Task 1: Started.")
    scheduler:yield()
    print("Task 1: Finished.")
end)

local task2 = coroutine.create(function()
    print("Task 2: Started.")
    scheduler:yield()
    print("Task 2: Finished.")
end)

scheduler:add(task1)
scheduler:add(task2)
scheduler:run()