是否有一个表可用于访问Lua中的局部变量?

时间:2016-11-28 06:09:55

标签: lua

就像我们如何做到这一点:

a = 3
print(_G['a']) -- 3

我希望能够做到这样的事情:

local a = 3
print(_L['a']) -- 3

我基本上希望能够使用名称作为字符串来访问局部变量。是否有一个表可以执行此操作,也许可以作为函数参数传递?它就像ActionScript中的this关键字。

1 个答案:

答案 0 :(得分:3)

这可以通过debug库 - 即getlocalsetlocal函数实现。如果您无法使用此库(或访问C API),那么您运气不佳。

您可以使用特制的_L表来扩展您的全局环境,当访问该表时,会执行当前本地组的线性查找。

读取局部变量只是找到匹配的变量名,并返回其值。写入局部变量需要您在堆栈帧中发现其索引,然后相应地更新该值。请注意,您无法创建本地人。

这是一个与Lua 5.1(但不是Lua 5.2 +)一起使用的简单示例。

local function find_local (key)
    local n = 1
    local name, sname, sn, value

    repeat
        name, value = debug.getlocal(3, n)

        if name == key then
            sname = name
            sn = n
        end

        n = n + 1
    until not name

    return sname, sn
end

_G._L = setmetatable({}, {
    metatable = false,
    __newindex = function (self, key, value)
        local _, index = find_local(key)

        if not index then
            error(('local %q does not exist.'):format(key))
        end

        debug.setlocal(2, index, value)
    end,
    __index = function (_, key)
        return find_local(key)
    end
})

使用中:

local foo = 'bar'

print(_L['foo']) --> 'bar'
_L['foo'] = 'qux'
print(_L['foo']) --> 'qux'

local function alter_inside (key)
    local a, b, c = 5, 6, 7
    _L[key] = 11
    print(a, b, c)
end

alter_inside('a') --> 11     6     7            
alter_inside('b') -->  5    11     7            
alter_inside('c') -->  5     6    11

您可以使用普通函数而不是表结合读/写操作(__index__newindex)以不同的方式编写它。

如果以上使用metatables是一个全新的主题,请参阅§2.4 – Metatables and Metamethods

在Lua 5.2+中,您可以使用特殊的_ENV表来调整当前的chunk环境,但请注意,这与使用local变量不同。

local function clone (t)
    local o = {}

    for k, v in pairs(t) do o[k] = v end

    return o
end

local function alter_inside (key)
    local _ENV = clone(_ENV)

    a = 5
    b = 6
    c = 7

    _ENV[key] = 11

    print(a, b, c)
end

alter_inside('a') --> 11     6     7            
alter_inside('b') -->  5    11     7            
alter_inside('c') -->  5     6    11

作为最后一点,还要考虑这种(ab)使用本地人可能不是最好的方法。

您可以在适当的时候将变量存储在表中,以便以更少的开销实现相同的结果。强烈建议采用这种方法。

local function alter_inside (key)
    -- `ls` is an arbitrary name, always use smart variable names.
    local ls = { a = 5, b = 6, c = 7 }

    ls[key] = 11

    print(ls.a, ls.b, ls.c)    
end

alter_inside('a') --> 11     6     7            
alter_inside('b') -->  5    11     7            
alter_inside('c') -->  5     6    11

不要陷入试图解决不必要问题的洞。