确定变量是否是 Lua 中的多个值之一

Question

我有一个值armyName，需要一个函数来检查它是否是 8 个值之一。由于我对 Lua 很陌生，我正在寻找进行此类检查的最佳方法是什么。

这有效：

local function isPlayerArmyName(armyName)
    return armyName == "ARMY_1"
        or armyName == "ARMY_2"
        or armyName == "ARMY_3"
        or armyName == "ARMY_4"
        or armyName == "ARMY_5"
        or armyName == "ARMY_6"
        or armyName == "ARMY_7"
        or armyName == "ARMY_8"
end

这也有效：

local function isPlayerArmyName(armyName)
    local names = {
        ["ARMY_1"] = true,  ["ARMY_2"] = true, ["ARMY_3"] = true, ["ARMY_4"] = true,
        ["ARMY_5"] = true,  ["ARMY_6"] = true, ["ARMY_7"] = true, ["ARMY_8"] = true,
    }

    return names[armyName] or false
end

与此 PHP 代码相比，它们都相当愚蠢：

return in_array( $armyName, [ "ARMY_1", "ARMY_2", "ARMY_3", ..., "ARMY_8" ] )

有没有比提供的 Lua 代码片段更好的方法来进行这样的检查？如果没有，典型的 Lua 程序员更喜欢遇到什么？第一个还是第二个？

我正在使用 Lua 5.0。

Answer 1

你也可以这样做：

local names = {"ARMY_1", "ARMY_2", "ARMY_3", "ARMY_4", "ARMY_5", "ARMY_6", "ARMY_7", "ARMY_8"}

local function isPlayerArmyName(armyName)
    for index = 1, #names do
        if names[index] == armyName then
            return true
        end
    end
end

此选项是可扩展的，因为您可以添加新的军队，以防您的实施需要更改。您可以将该for部分提取到您自己的in_array函数中，您可以在所有代码中使用它：

local function in_array(value, array)
    for index = 1, #array do
        if array[index] == value then
            return true
        end
    end

    return false -- We could ommit this part, as nil is like false
end

local armyNames = {"ARMY_1", "ARMY_2", "ARMY_3", "ARMY_4", "ARMY_5", "ARMY_6", "ARMY_7", "ARMY_8"}

local function isPlayerArmyName(armyName)
    return in_array(armyName, armyNames)
end

此实现可能比您的第二个选项慢一点，因为您的第二个选项使用密钥。您的第二个选择还不错，它甚至可能是最快的。它看起来很傻，但它很容易配置。

您也可以使用模式匹配，如果该表不是必需的并且将坚持该军队命名约定：

local function isPlayerArmyName(armyName)
    return (string.match(armyName, "^ARMY_[1-8]$")) ~= nil
end

您甚至可以删除~= nil，因为它将返回一个字符串，这在 Lua 中也是如此：

local function isPlayerArmyName(armyName)
    return string.match(armyName, "^ARMY_[1-8]$")
end

if isPlayerArmyName("ARMY_1") then
    -- This will be executed, since it returned "ARMY_1", which evaluates to true
end

if isPlayerArmyName("ARMY_9") then
    -- This will not be executed, since the function returned nil.
end

至于您提供的第一个示例，我不建议这样做，因为它不是模块化的或可扩展的，不是在 Lua 中，也不是在任何其他语言中。