如何在Lua中实现运行时单赋值变量？

Lua本身没有原生支持Rust那种编译期检查的单赋值变量，但可以通过元表或闭包在运行时实现类似效果——变量仅允许赋值一次，第二次赋值会触发错误。

方法一：基于元表的批量单赋值变量 #

这种方式适合管理多个单赋值变量，通过元表的__newindex元方法拦截赋值操作，检测变量是否已被赋值：

-- 创建单赋值变量容器
local singleAssign = setmetatable({}, {
    __newindex = function(t, key, val)
        -- 检查变量是否已存在值
        if rawget(t, key) ~= nil then
            error("无法重复赋值单赋值变量 '" .. key .. "'", 2)
        end
        -- 首次赋值，直接设置值
        rawset(t, key, val)
    end
})

-- 使用示例
singleAssign.x = 10  -- 首次赋值成功
-- singleAssign.x = 20  -- 取消注释会触发运行时错误

local y = 20
print(string.format("%d + %d = %d", singleAssign.x, y, singleAssign.x + y))

方法二：基于闭包的独立单赋值变量 #

如果需要更贴近局部变量的使用体验，可以用闭包封装单个变量的赋值逻辑：

-- 生成单赋值变量的工厂函数
local function createSingleVar()
    local hasAssigned = false
    local varValue
    -- 返回的函数既可以赋值也可以取值
    return function(newVal)
        if newVal ~= nil then
            if hasAssigned then
                error("无法重复赋值该单赋值变量", 2)
            end
            hasAssigned = true
            varValue = newVal
        end
        return varValue
    end
end

-- 使用示例
local x = createSingleVar()
x(10)  -- 首次赋值
-- x(20)  -- 取消注释会触发运行时错误

local y = 20
print(string.format("%d + %d = %d", x(), y, x() + y))

注意事项 #

Lua是动态语言，无法像Rust那样在编译期检测重复赋值，上述实现都是在运行时抛出错误，需要在代码中提前测试验证。

内容的提问来源于stack exchange，提问作者earl_igginotnum