有没有办法确定 Lua 函数的签名？

Posted 2023-03-28

【中文标题】有没有办法确定 Lua 函数的签名？

【英文标题】：Is there a way to determine the signature of a Lua function?

【发布时间】：2010-09-13 15:12:15

【问题描述】：

最近，Lee Baldwin 展示了如何编写generic, variable argument memoize function。我认为最好返回一个只需要一个参数的更简单的函数。这是我的全部虚假尝试：

local function memoize(f)
   local cache = 

   if select('#', ...) == 1 then
      return function (x)
                if cache[x] then
                   return cache[x]
                else
                   local y = f(x)
                   cache[x] = y
                   return y
                end
              end
   else
      return function (...)
                local al = varg_tostring(...)
                if cache[al] then
                   return cache[al]
                else
                   local y = f(...)
                   cache[al] = y
                   return y
                end
             end
   end
end

显然，select('#', ...) 在这种情况下会失败，并且不会真正做我想做的事。有没有办法告诉 memoize 里面有多少参数 f 期望？

---

如果您确定，“否”是一个很好的答案。使用两个单独的 memoize 函数没什么大不了的。

【参考方案1】：

我想你可以进入调试信息并从源代码中确定这一点，但基本上它是“不”，抱歉。

【讨论】：

没问题。那我就不会再浪费时间搜索了。 ;-)

仅供参考，您也许可以在 C 中实现这一点，但它只能检查 Lua 函数而不是 C 函数。

【参考方案2】：

是的，对于 Lua 函数而不是 C 函数。这有点折磨人，有点粗略。

debug.getlocal 适用于被调用的函数，因此您必须调用有问题的函数。除非调用传递了足够的参数，否则它不会显示任何... 的提示。下面的代码尝试了 20 个参数。

debug.sethook 带有“call”事件提供了在函数运行任何代码之前拦截函数的机会。

此算法适用于 Lua 5.2。旧版本会相似但不一样：

assert(_VERSION=="Lua 5.2", "Must be compatible with Lua 5.2")

一个小辅助迭代器（可以内联以提高效率）：

local function getlocals(l)
  local i = 0
  local direction = 1
  return function ()
    i = i + direction
    local k,v = debug.getlocal(l,i)
    if (direction == 1 and (k == nil or k.sub(k,1,1) == '(')) then 
      i = -1 
      direction = -1 
      k,v = debug.getlocal(l,i) 
    end
    return k,v
  end
end

返回签名（但可以返回参数计数并使用可变参数）：

local function dumpsig(f)
  assert(type(f) == 'function', 
    "bad argument #1 to 'dumpsig' (function expected)")
  local p = 
  pcall (function() 
    local oldhook
    local hook = function(event, line)
      for k,v in getlocals(3) do 
        if k == "(*vararg)" then 
          table.insert(p,"...") 
          break
        end 
        table.insert(p,k) end
      debug.sethook(oldhook)
      error('aborting the call')
    end
    oldhook = debug.sethook(hook, "c")
    -- To test for vararg must pass a least one vararg parameter
    f(1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20)
  end)
  return "function("..table.concat(p,",")..")"  
end

【参考方案3】：

我很确定你不能在 Lua 中这样做。