当前位置：首页 > news >正文

Lua Global环境

news 来源：原创 2024/5/9 15:24:41

Lua的全局变量实际上是用environment这个表来存储所有的全局变量，其优点就是简化了lua的内部实现，而且还能像其他类型的表一样去操作这个全局表。

_G

Lua将environment本身存储在全局变量_G中，其中_G._G = _G。如若感兴趣，可以直接用pairs函数打印当前的全局变量名：

A = {1,2,3}function testMySelf()
endfor n in pairs(_G) do print(n) end
--[[
assert
dofile
...
testMySelf
utf8
A
package
...
pairs
_G
...
]]--

_G的用法其实和直接赋值全局变量差不多，如下两句性质完全一样的代码所示：

_G["a"] = _G["var1"]
a = var1

getfield/setfield

手动写一个getfield函数，用以通过字符串查找相应的域，比如a.b.c.d的值是多少。实际上是从_G开始循环向域内部遍历：

function getfield (f) local v = _G -- start with the table of globals for w in string.gmatch(f, "([%w_]+)") dov = v[w]endreturn v
endA = {["a"] = 1}
print(getfield("A.a")) --1

setfield也是一样的道理，只不过需要判断是否是最后一个"."，需要独立处理最后一个域并进行赋值，比如A.a = 1：

function setfield (f, v) local t = _G -- start with the table of globals for w, d in string.gmatch(f, "([%w_]+)(.?)") doif 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 endend
endsetfield("A.a",2)
print(A.a) --2

声明全局变量

如若控制全局变量的声明和访问，则直接控制_G的newindex和index即可，但是需要注意一点的是如果在newindex中明确禁止了全局变量的声明操作，那么任何直接赋值都是有问题的，如以下代码：

setmetatable(_G, { __newindex = function (_, n)print("attempt to write to undeclared variable "..n)end,__index = function (_, n)print("attempt to read undeclared variable "..n)end,
})
A = {} --attempt to write to undeclared variable A

这时候只能通过rawget/rawset来直接进行访问操作。比如声明一个入口函数，所有全局变量的声明都必须经此函数初始化；判断这个域是否存在，也需要用rawget函数直接访问：

function declare (name, initval) rawset(_G, name, initval or false) 
end...
if rawget(_G, var) == nil then
-- 'var' is undeclared ... 
end

当然需要注意的是上述处理中初始化全局变量如果是nil赋值则默认赋值为false了。

如若不需要这个默认赋值，希望代码支持类似a = nil的声明，那么也可以维护一个注册表并放入newindex元方法中判断，当然这也是官方文档给的一个解决方案（不过对本人来说觉得有点鸡肋，毕竟类似a = nil的代码是可以避免的）：

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] thenerror("attempt to write to undeclared var. "..n, 2)elseprint("newindex")rawset(t, n, v) -- do the actual set endend,__index = function (_, n)if not declaredNames[n] thenerror("attempt to read undeclared var. "..n, 2)elsereturn nilendend,
})declare("A",nil)
A = {} --newindex