如何在Lua中为选定的函数执行额外的代码

2016-1-20 9:22:46
收藏：0
阅读：56
评论：1

有没有一种干净优雅的方式，在Lua中“注入”代码到现有函数中？

问题是我们有一个包含许多函数的表格，并且我们必须对它们的参数和结果执行简单的日志记录。显而易见的解决方案是将执行记录的行复制粘贴到每个函数中，但这似乎不正确，并且耗时很长。

我试图使用元表机制，即创建一个充当包装器的表格，但我找不到从其级别执行代码的方法：

log = function(msg) trace(msg) end

mapl = {} --包装表格
mt = {
    __index = map; -- map是包含许多方法的原始表
                   --这里有一种召唤日志函数的方法吗？
}

setmetatable(mapl, mt)

mapl:getMap(params) --通过mapl包装器调用map中的getMap函数

或者这种方法是完全错误的？也许已经有一种机制来跟踪/记录函数调用，我不知道？

点赞
用户5675002
用户5675002

考虑你最初的方法，可以这样做：

local log = function(func_name, ...)
    print("调用 "..func_name.."，参数为:")
    for i,v in ipairs {...} do
        print(i,v)
    end
end

-- 原始的带函数的映射
local map = {
    getMap = function() print "原始 getMap" end,
    setNode = function() print "原始 setNode" end
}

-- 记录的接口
local mapl = {}

-- 辅助函数返回新的记录封装
local function grab_args(_, func_name)
    local func = map[func_name]

    -- 检查函数是否存在
    if type(func) == "function" then
        local wrapper = function(_, ...)
            log(func_name, map, ...)
            return func(map, ...)
        end
        -- 缓存记录封装以备后用
        mapl[func_name] = wrapper
        return wrapper
    end

    -- 返回映射中的非函数的值
    return func
end

-- 设置接口
setmetatable(mapl, {__index = grab_args})

-- 测试记录
mapl:getMap(1,2,3)
mapl:setNode(42)

样例中有几点需要注意。你使用冒号作为调用函数的方式，如 mapl:getMap()。因此这个样例将实际调用原始函数的地方将 mapl 替换为了 map。如果你要使用点号调用函数，如 mapl.getMap()，可以微调封装的创建，如 local wrapper = function()。

2016-01-20 10:54:37

作者:

用户5814720

用户5814720
推荐文章