Lua sethook设置'r'模式时函数异常退出无法触发钩子问题求助

Lua标准库的debug.sethook注册的返回钩子仅会捕获函数正常执行完成的返回流程，函数因错误（主动抛出、原生运行时错误）异常退出时，会跳过当前调用栈上所有未捕获错误的函数的返回钩子触发，最终导致耗时统计出现遗漏，可通过以下几种方案解决：

方案1：返回钩子+错误钩子联动统计 #

这是标准Lua环境下兼容性最好的方案，适配所有Lua版本：

• 注册钩子时同时开启调用、返回、错误三类事件监听，对应mask参数为"cre"
• 自行维护一个调用栈结构，调用事件触发时将函数标识、进入时间压入栈；返回事件触发时弹出栈顶元素统计正常返回耗时
• 错误事件触发时，将当前自定义调用栈中剩余的所有元素逐个弹出，统一标记为异常退出，补全耗时统计
• 核心代码示例：

local stat_stack = {}
debug.sethook(function(event)
    local curr_time = os.clock()
    if event == "call" or event == "tail call" then
        -- 尾调用特殊处理，避免栈层级错位
        if event == "tail call" then
            table.remove(stat_stack)
        end
        table.insert(stat_stack, {
            func = debug.getinfo(2, "f").func,
            start = curr_time
        })
    elseif event == "return" then
        local item = table.remove(stat_stack)
        local cost = curr_time - item.start
        -- 正常返回统计逻辑
    elseif event == "err" then
        -- 错误触发时栈内剩余函数均为异常退出
        while #stat_stack > 0 do
            local item = table.remove(stat_stack)
            local cost = curr_time - item.start
            -- 异常返回统计逻辑
        end
    end
end, "cre")

方案2：封装错误抛出接口拦截 #

如果项目中主动抛出的错误占绝大多数，且可统一收口错误触发逻辑，可以用该方案降低性能损耗：

• 全局覆盖error、assert等错误抛出接口，在封装逻辑中触发错误前先遍历当前调用栈，补全所有未统计的函数耗时
• 劣势是无法覆盖Lua虚拟机原生抛出的错误（比如nil索引、类型不匹配等），仅适合错误逻辑全可控的场景

方案3：LuaJIT环境专属优化 #

如果项目使用LuaJIT运行，可以直接利用LuaJIT对sethook的扩展能力：

• LuaJIT的返回钩子默认会同时捕获正常返回和异常退出的事件，不需要额外搭配错误钩子，只需要注册常规的调用、返回钩子即可完成全场景统计
• 性能比方案1更高，但仅支持LuaJIT环境，标准Lua不兼容

注意：所有方案都需要额外处理尾调用场景，尾调用不会触发对应函数的返回钩子，需要在调用事件中判断事件类型为尾调用时，先弹出上一层栈元素再压入当前尾调用函数信息，避免统计栈层级错位。

内容的提问来源于stack exchange，提问作者tyChen