Lua教程（十）： 全局变量和非全局的环境

--该table用于存储所有已经声明过的全局变量名
local declaredNames = {}
local mt = {
    __newindex = function(table,name,value)
        --先检查新的名字是否已经声明过，如果存在，这直接通过rawset函数设置即可。
        if not declaredNames[name] then
            --再检查本次操作是否是在主程序或者C代码中完成的，如果是，就继续设置，否则报错。
            local w = debug.getinfo(2,"S").what
            if w ~= "main" and w ~= "C" then
                error("attempt to write to undeclared variable " .. name)
            end
            --在实际设置之前，更新一下declaredNames表，下次再设置时就无需检查了。
            declaredNames[name] = true
        end
        print("Setting " .. name .. " to " .. value)
        rawset(table,name,value)
    end,
   
    __index = function(_,name)
        if not declaredNames[name] then
            error("attempt to read undeclared variable " .. name)
        else
            return rawget(_,name)
        end
    end
}   
setmetatable(_G,mt)

a = 11
local kk = aa

--输出结果为：
--[[
Setting a to 11
lua: d:/test.lua:21: attempt to read undeclared variable aa
stack traceback:
        [C]: in function 'error'
        d:/test.lua:21: in function <d:/test.lua:19>
        d:/test.lua:30: in main chunk
        [C]: ?
--]]

a = 1
setfenv(1,{})
print(a)

--输出结果为：
--[[
lua: d:/test.lua:3: attempt to call global 'print' (a nil value)
stack traceback:
        d:/test.lua:3: in main chunk
        [C]: ?
--]]

a = 1
local newgt = {}  --新环境表
setmetatable(newgt,{__index = _G})
setfenv(1,newgt)
print(a)  --输出1

a = 10
print(a)  --输出10
print(_G.a) --输出1
_G.a = 20
print(a)  --输出10

function factory()
    return function() return a end
end
a = 3
f1 = factory()
f2 = factory()
print(f1())  --输出3
print(f2())  --输出3

setfenv(f1,{a = 10})
print(f1())  --输出10
print(f2())  --输出3