Question

我目前有这个代码，但我觉得这可以更好地进行优化：

local config = {
    val = 1,
    background = {
        foo = 5,
        textures = {
            bar = 'okay',
            layers = {
                alpha = 1,
                color = {
                    red = 255,
                    blue = 0,
                    green = 100
                }
            }
        }
    },
    sub = {
        val = 5,
        foo = {
            val = true,
            bar = {
            }
        }
    }
}

local function set_config(...)
    local arg = {...}
    if type(arg[1]) == 'table' then
        table.extend(config, arg[1])
    elseif #arg == 2 then
        if type(arg[2]) == 'table' then
            table.extend(config[arg[1]], arg[2])
        else
            config[arg[1]] = arg[2]
        end
    elseif #arg == 3 then
        if type(arg[3]) == 'table' then
            table.extend(config[arg[1]][arg[2]], arg[3])
        else
            config[arg[1]][arg[2]] = arg[3]
        end
    -- And now for the 4th time. This is starting to get way to repetitive
    elseif #arg == 4 then
        if type(arg[4]) == 'table' then
            table.extend(config[arg[1]][arg[2]][arg[3]], arg[4])
        else
            config[arg[1]][arg[2]][arg[3]] = arg[4]
        end
    -- And for what I try to achieve, I should continue this elseif's until indefinite
    end
end
-- And this should go as deep into the table as I want to

现在我可以做到

set_config('val', 3)
set_config('sub', {
    foo = false
})

但是我想在没有set_config

的限制的情况下深入到表中

所以我可以做到

set_config('background', 'textures', 'layers', 'medium', 'color', 'red', 255)

澄清一下，table.extend

function table.extend(t1, t2)
    for k, v in pairs(t2) do
        if (type(v) == "table") and (type(t1[k] or false) == "table") then
            table.extend(t1[k], t2[k])
        else
            t1[k] = v
        end
    end
    return t1
end

我尝试了一些事情，但我的Lua知识相当有限。

lhf的回答几乎是正确的，这是我想要的结果：

local function set_config(...)
    local a={...}
    local n=#a
    local t=config
    if #a == 1 and type(a[1]) == 'table' then
        table.extend(config, a[1])
        return
    end
    for i=1,n-2 do
        local k=a[i]
        t[k]=t[k] or {}
        t=t[k]
    end
    if type(t[a[n-2]]) == 'table' then
        table.extend(t[a[n-1]], a[n])
    else
        t[a[n-1]] = a[n]
    end
end

Answer 1

试试这个：

local config={}

function set_config(...)
    local a={...}
    local n=#a
    local t=config
    for i=1,n-2 do
        local k=a[i]
        t[k]=t[k] or {}
        t=t[k]
    end
    t[a[n-1]]=a[n]
end

set_config('background', 'textures', 'layers', 'medium', 'color', 'red', 255)

print(config.background.textures.layers.medium.color.red)

Answer 2

通过一些可元化的魔法，你可以完全摆脱set_config：

local config, config_mt
config_mt = {
    __index = function(t, k)
        t[k] = setmetatable({}, config_mt)
        return t[k]
    end,
}
config = setmetatable({}, config_mt)

config.background.textures.layers.medium.color.red = 255
print(config.background.textures.layers.medium.color.red)

虽然有一个缺点：从存储中恢复配置后，您必须遍历结构一次才能再次应用元数据。

Lua多个参数进入表深度

2 个答案: