Question

所以我有一个Lua脚本，我在其中设置函数的环境并定义我在环境中可用的其他函数。问题在于，当我调用函数setfenv时，它在环境中调用的任何函数都使用全局_G环境来查找值，而不是调用初始函数时设置的环境。我已经复制了以下问题：

> function f() print(t) end
> t = 5
> f()
5
> env = {}
> env['print'] = print
> env['t'] = 7
> env['f'] = f
> setfenv(f, env)
> f()
7
> setfenv(f, _G)
> f()
5
> function g() f() end
> g()
5
> setfenv(g, env)
> g()
5

有没有办法让环境传播到所有调用，而不必对我要添加到setfenv的每个函数都执行env（基本上，以便示例中的最后一个调用g()会返回7而不是5）？

使用Lua 5.1。

Answer 1

您可以编写一个函数，以便在每次调用时自动更改自己的环境

-- This code works on Lua 5.1, 5.2, 5.3
local getfenv = getfenv or function() end
local setfenv = setfenv or getfenv

a = 1

function f(env)
   local _ENV = env or getfenv(2) or _ENV; setfenv(1, _ENV)
   print(a)
end

env = { a = 2, print = print }

-- inherit environment for this call
f()    --> 1
-- set special environment for this call
f(env) --> 2
f()    --> 1

或者只是setfenv(1, getfenv(2))如果您只使用Lua 5.1并希望始终继承环境。

在setfenv函数中调用函数时的Lua环境

1 个答案: