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

Lua将其所有的全局变量保存在一个常规的table中，这个table被称为“环境”。它被保存在全局变量_G中。

1. 全局变量声明：

Lua中的全局变量不需要声明就可以使用。尽管很方便，但是一旦出现笔误就会造成难以发现的错误。我们可以通过给_G表加元表的方式来保护全局变量的读取和设置，这样就能降低这种笔误问题的发生几率了。见如下示例代码：

代码如下：

--该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]: ?

--]]

 2. 非全局的环境：

全局环境存在一个刚性的问题，即它的修改将影响到程序的所有部分。Lua 5为此做了一些改进，新的特征可以支持每个函数拥有自己独立的全局环境，而由该函数创建的closure函数将继承该函数的全局变量表。这里我们可以通过setfenv函数来改变一个函数的环境，该函数接受两个参数，一个是函数名，另一个是新的环境table。第一个参数除了函数名本身，还可以指定为一个数字，以表示当前函数调用栈中的层数。数字1表示当前函数，2表示它的调用函数，以此类推。见如下代码：

代码如下：

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]: ?

--]]

为什么得到这样的结果呢？因为print和变量a一样，都是全局表中的字段，而新的全局表是空的，所以print调用将会报错。

为了应对这一副作用，我们可以让原有的全局表_G作为新全局表的内部表，在访问已有全局变量时，可以直接转到_G中的字段，而对于新的全局字段，则保留在新的全局表中。这样即便是函数中的误修改，也不会影响到其他用到全局变量(_G)的地方。见如下代码：

代码如下：

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