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

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 并希望始终继承环境。