Lua 协程模拟 Golang 的 go defer 编程模式

封装go函数用于创建并启动一个协程:

lua 复制代码
---go函数创建并启动一个协程
---@param _co_task function @函数原型 fun(_co:thread)
function go(_co_task)
    local co = coroutine.create(_co_task) -- 创建一个暂停的协程
    coroutine.resume(co, co) -- 调用coroutine.resume激活协程执行
end

封装项目中的异步函数用于成为协程函数:

lua 复制代码
---封装c_model.c_foo异步函数,成为协程函数
---@param _co thread @协程对象
---@return boolean,string
function co_foo(_co)
    c_model.c_foo(function(_ok, _result)
        coroutine.resume(_co, _ok, _result) -- 2. 回调函数被调用后,激活本协程继续执行,并把_ok和_result传递给yield
    end)
    return coroutine.yield() -- 1. 主动放弃运行,本协程被切换出去
end

使用例子:

lua 复制代码
---通过顺序编写代码解决回调函数造成同块逻辑被撕裂的例子
---@param _co thread @协程对象
function test(_co)
    for i = 1, 10, 1 do
        local ok, result = co_foo(_co) -- co_foo函数会先yield切出,等内部回调被执行时再通过resume重新切回来继续执行
        print(ok, result)
    end
end

-- 启动test协程
go(test)

封装defer用于处理协程的正常和异常退出:

首先定义defer函数,让它具备能够多次被调用的能力:

lua 复制代码
function defer(_co_wrap, h)
    table.insert(_co_wrap.defer_handlers, h)
end

同时定义co_error对象用于判断是否有错误:

lua 复制代码
---@class co_error
---@field ok boolean

定义invoke_defer_handlers函数来执行defer操作:

lua 复制代码
function invoke_defer_handlers(_co_wrap, _co_error)
    for i=#_co_wrap.defer_handlers, 1, -1 do
        local h = _co_wrap.defer_handlers[i]
        xpcall(h, function(err) print(err) end, _co_error)
    end
end

定义coroutine_resume函数来处理协程的异常退出:

lua 复制代码
function coroutine_resume(_co_wrap, ...)
    local ok, errmsg = coroutine.resume(_co_wrap.co, ...)
    if not ok then
        invoke_defer_handlers(_co_wrap, {ok=false}) -- 异常退出
    end
end

定义go函数来处理协程的正常退出:

lua 复制代码
function go(_co_task)
    local co = coroutine.create(function(_co_wrap)
        _co_task(_co_wrap)
        invoke_defer_handlers(_co_wrap, {ok=true}) -- 正常退出
    end)
    local cowrap = { co = co, defer_handlers = {} } ---@type co_wrap
    coroutine_resume(cowrap, cowrap) -- 初创建的协程是暂停的,手动触发执行
end

通过以上封装,可以在Lua中以类似Golang的方式编写协程代码。

协程间的通信暂时不需要在多线程中实现,因为项目中目前只有一个lua_state对象管理。

相关推荐
玩转C语言和数据结构12 小时前
Lua下载和安装教程(附安装包)
lua·lua下载·lua安装教程·lua下载和安装教程·lua安装包
Arva .21 小时前
HTTP Client
网络协议·http·lua
爱吃小胖橘2 天前
Lua语法(2)
开发语言·unity·lua
ellis19703 天前
LuaC API知识点汇总
unity·lua
爱吃小胖橘5 天前
Lua语法
开发语言·unity·lua
东方芷兰5 天前
JavaWeb 课堂笔记 —— 20 SpringBootWeb案例 配置文件
java·开发语言·笔记·算法·log4j·intellij-idea·lua
1nullptr6 天前
Lua上值与闭包
开发语言·lua
半夏知半秋7 天前
skynet-socket.lua源码分析
服务器·开发语言·学习·架构·lua
夜猫逐梦11 天前
【lua】luajit 命令行使用指南
开发语言·lua
利来利往11 天前
【奇怪的bug】lua的nil不报错
开发语言·bug·lua