Lua 用一个名为 environment 普通的表来保存所有的全局变量。(更精确的说,Lua在一系列的 environment 中保存他的“global”变量,但是我们有时候可以忽略这种多样性)这种结果的优点之一是他简化了 Lua 的内部实现,因为对于所有的全局变量没有必要非要有不同的数据结构。另一个(主要的)优点是我们可以像其他表一样操作这个保存全局变量的表。为了简化操作,Lua 将环境本身存储在一个全局变量_G 中,(_G._G 等 于_G)。例如,下面代码打印在当前环境中所有的全局变量的名字:
for n in pairs(_G) do print(n) end
这一章我们将讨论一些如何操纵环境的有用的技术。
通常,赋值操作对于访问和修改全局变量已经足够。然而,我们经常需要一些原编程(meta-programming)的方式,比如当我们需要操纵一个名字被存储在另一个变量中的全局变量,或者需要在运行时才能知道的全局变量。为了获取这种全局变量的值,有的程序员可能写出下面类似的代码:
loadstring("value = " .. varname)()
or
value = loadstring("return " .. varname)()
如果 varname 是 x,上面连接操作的结果为:“return x”(第一种形式为 “value = x”),当运行时才会产生最终的结果。
然而这段代码涉及到一个新的 chunk 的创建和编译以及其他很多额外的问题。你可以换种方式更高效更简洁的完成同样的功能,代码如下:
value = _G[varname]
因为环境是一个普通的表,所以你可以使用你需要获取的变量(变量名)索引表即可。
也可以用相似的方式对一个全局变量赋值:_G[varname] = value。小心:一些程序员对这些函数很兴奋,并且可能写出这样的代码:
_G["a"] = _G["var1"]
,这只是 a = var1的复杂的写法而已。
对前面的问题概括一下,表域可以是型如"io.read" or "a.b.c.d"的动态名字。我们用循环解决这个问题,从_G 开始,一个域一个域的遍历:
function getfield (f)
local v = _G -- start with the table of globals
for w in string.gfind(f, "[%w_]+") do
v = v[w]
end
return v
end
我们使用 string 库的 gfind 函数来迭代 f 中的所有单词(单词指一个或多个子母下划线的序列)。相对应的,设置一个域的函数稍微复杂些。赋值如:
a.b.c.d.e = v
实际等价于:
local temp = a.b.c.d
temp.e = v
也就是说,我们必须记住最后一个名字,必须独立的处理最后一个域。新的 setfield函数当其中的域(译者注:中间的域肯定是表)不存在的时候还需要创建中间表。
function setfield (f, v)
local t = _G -- start with the table of globals
for w, d in string.gfind(f, "([%w_]+)(.?)") do
if d == "." then -- not last field?
t[w] = t[w] or {} -- create table if absent
t = t[w] -- get the table
else -- last field
t[w] = v -- do the assignment
end
end
end
这个新的模式匹配以变量 w 加上一个可选的点(保存在变量 d 中)的域。如果一个域名后面不允许跟上点,表明它是最后一个名字。(我们将在第 20 章讨论模式匹配问题)。使用上面的函数
setfield("t.x.y", 10)
创建一个全局变量表 t,另一个表 t.x,并且对 t.x.y 赋值为 10:
print(t.x.y) --> 10
print(getfield("t.x.y")) --> 10
全局变量不需要声明,虽然这对一些小程序来说很方便,但程序很大时,一个简单的拼写错误可能引起 bug 并且很难发现。然而,如果我们喜欢,我们可以改变这种行为。因为 Lua 所有的全局变量都保存在一个普通的表中,我们可以使用 metatables 来改变访问全局变量的行为。第一个方法如下:
setmetatable(_G, {
__newindex = function (_, n)
error("attempt to write to undeclared variable "..n, 2)
end,
__index = function (_, n)
error("attempt to read undeclared variable "..n, 2)
end,
})
这样一来,任何企图访问一个不存在的全局变量的操作都会引起错误:
> a = 1
stdin:1: attempt to write to undeclared variable a
但是我们如何声明一个新的变量呢?使用 rawset,可以绕过 metamethod:
function declare (name, initval)
rawset(_G, name, initval or false)
end
or 带有 false 是为了保证新的全局变量不会为 nil。注意:你应该在安装访问控制以前(before installing the access control)定义这个函数,否则将得到错误信息:毕竟你是在企图创建一个新的全局声明。只要刚才那个函数在正确的地方,你就可以控制你的全局变量了:
> a = 1
stdin:1: attempt to write to undeclared variable a
> declare "a"
> a = 1 -- OK
但是现在,为了测试一个变量是否存在,我们不能简单的比较他是否为 nil。如果他是 nil 访问将抛出错误。所以,我们使用 rawget 绕过
metamethod:
if rawget(_G, var) == nil then
-- 'var' is undeclared
...
end
改变控制允许全局变量可以为 nil 也不难,所有我们需要的是创建一个辅助表用来保存所有已经声明的变量的名字。不管什么时候 metamethod 被调用的时候,他会检查这张辅助表看变量是否已经存在。代码如下:
local declaredNames = {}
function declare (name, initval)
rawset(_G, name, initval)
declaredNames[name] = true
end
setmetatable(_G, {
__newindex = function (t, n, v)
if not declaredNames[n] then
error("attempt to write to undeclared var. "..n, 2)
else
rawset(t, n, v) -- do the actual set
end
end,
__index = function (_, n)
if not declaredNames[n] then
error("attempt to read undeclared var. "..n, 2)
else
return nil
end
end,
})
两种实现方式,代价都很小可以忽略不计的。第一种解决方法:metamethods 在平常操作中不会被调用。第二种解决方法:他们可能被调用,不过当且仅当访问一个值为nil 的变量时。
全局环境的一个问题是,任何修改都会影响你的程序的所有部分。例如,当你安装一个 metatable 去控制全局访问时,你的整个程序都必须遵循同一个指导方针。如果你想使用标准库,标准库中可能使用到没有声明的全局变量,你将碰到坏运。
Lua 5.0 允许每个函数可以有自己的环境来改善这个问题,听起来这很奇怪;毕竟,全局变量表的目的就是为了全局性使用。然而在 Section 15.4 我们将看到这个机制带来很多有趣的结构,全局的值依然是随处可以获取的。
可以使用 setfenv 函数来改变一个函数的环境。Setfenv 接受函数和新的环境作为参数。除了使用函数本身,还可以指定一个数字表示栈顶的活动函数。数字 1 代表当前函数,数字 2 代表调用当前函数的函数(这对写一个辅助函数来改变他们调用者的环境是很方便的)依此类推。下面这段代码是企图应用 setfenv 失败的例子:
a = 1 -- create a global variable
-- change current environment to a new empty table
setfenv(1, {})
print(a)
导致:
stdin:5: attempt to call global `print' (a nil value)
(你必须在单独的 chunk 内运行这段代码,如果你在交互模式逐行运行他,每一行都是一个不同的函数,调用 setfenv 只会影响他自己的那一行。)一旦你改变了你的环境,所有全局访问都使用这个新的表,如果她为空,你就丢失所有你的全局变量,甚至_G,所以,你应该首先使用一些有用的值封装(populate)她,比如老的环境:
a = 1 -- create a global variable
-- change current environment
setfenv(1, {_G = _G})
_G.print(a) --> nil
_G.print(_G.a) --> 1
现在,当你访问"global" _G,他的值为旧的环境,其中你可以使用 print 函数。你也可以使用继承封装(populate)你的新的环境:
a = 1
local newgt = {} -- create new environment
setmetatable(newgt, {__index = _G})
setfenv(1, newgt) -- set it
print(a) --> 1
在这段代码新的环境从旧的环境中继承了 print 和 a;然而,任何赋值操作都对新表进行,不用担心误操作修改了全局变量表。另外,你仍然可以通过_G 修改全局变量:
-- continuing previous code
a = 10
print(a) --> 10
print(_G.a) --> 1
_G.a = 20
print(_G.a) --> 20
当你创建一个新的函数时,他从创建他的函数继承了环境变量。所以,如果一个chunk 改变了他自己的环境,这个 chunk 所有在改变之后定义的函数都共享相同的环境,都会受到影响。这对创建命名空间是非常有用的机制