设置Lua函数环境,同时保留upvalues,使用pcall

这对我有用。对你有用吗?

卢阿福

local L = require('lib')
local print=print
_ENV=L

local b = 17
function t:foo()
    local a = 42
    print(a + b + c())
end

t()

图书馆

local lib = {}
function lib.c()
    return 41
end

local function go(t)
    for k,v in pairs(t) do
        if 'function'==type(v) then print(pcall(v)) end
    end
end

lib.t= setmetatable({}, {__call=go})

return lib

明白了,顺便说一句 this article 它为 getfenv 和 setfenv (依赖于 debug 图书馆)。

完整的代码如下,但解决方案大致如下:

setmetable(lib, {__index=getfenv(function_to_call)})
setfenv(function_to_call, lib)
pcall(function_to_call)

require 'debug'

local function setfenv(fn, env)
  local i = 1
  while true do
    local name = debug.getupvalue(fn, i)
    if name == "_ENV" then
      debug.upvaluejoin(fn, i, (function()
        return env
      end), 1)
      break
    elseif not name then
      break
    end

    i = i + 1
  end

  return fn
end

local function getfenv(fn)
  local i = 1
  while true do
    local name, val = debug.getupvalue(fn, i)
    if name == "_ENV" then
      return val
    elseif not name then
      break
    end
    i = i + 1
  end
end

local lib = {}
function lib:c()
    return 41
end
setmetatable(lib,{})

local function go(t)
    for k,v in pairs(t) do
        if 'function'==type(v) then
            getmetatable(lib).__index=getfenv(v)
            setfenv(v,lib)
            print(pcall(v))
        end
    end
end

return function()
    return setmetatable({}, {__call=go})
end