lua setmetatable

setmetatable 是 Lua 语言中的一个核心函数，它用于设置表的元表（metatable）。元表是 Lua 中非常强大的一个特性，可以用来改变或扩展表的行为，比如实现运算符重载、方法重载、访问控制等。

基本用法

在 Lua 中，表（table）是一种既简单又强大的数据结构。你可以将表当作数组、字典或其它数据结构来使用。但是，当你希望改变表的默认行为时，元表就派上用场了。

local myTable = {}
local myMetatable = {}

setmetatable(myTable, myMetatable)

上述代码中，我们创建了两个空表 myTable 和 myMetatable，然后使用 setmetatable 将 myMetatable 设为 myTable 的元表。通过这种方式，myTable 的行为可以通过 myMetatable 进行调整。

元方法与元表

为了理解 setmetatable 的强大之处，我们需要了解元方法（metamethods）。元方法是一些预定义的键，它们对应的值是函数，用于定义表在某些情况下的特定行为。以下是一些常用的元方法：

• __index：用于访问表中不存在的键时的行为。
• __newindex：用于给表中不存在的键赋值时的行为。
• __add：定义了两个表相加的行为。
• __tostring：定义了当使用 tostring 函数时的行为。

实例：使用 __index 和 __newindex

local defaultTable = {a = 1, b = 2}

local mt = {
    __index = function(table, key)
        return defaultTable[key] or "Key not found"
    end,
    __newindex = function(table, key, value)
        print("Assigning a new value!")
        rawset(table, key, value)
    end
}

local customTable = {}
setmetatable(customTable, mt)

print(customTable.a)  -- 输出 1，来自 defaultTable
print(customTable.b)  -- 输出 2，来自 defaultTable
print(customTable.c)  -- 输出 "Key not found"

customTable.c = 3  -- 输出 "Assigning a new value!"
print(customTable.c)  -- 输出 3

运算符重载

Lua 支持通过元方法对运算符进行重载。例如，使用 __add 可以定义表的加法行为：

local mt = {
    __add = function(table1, table2)
        local sumTable = {}
        for k, v in pairs(table1) do
            sumTable[k] = v + (table2[k] or 0)
        end
        return sumTable
    end
}

local t1 = {a = 1, b = 2}
local t2 = {a = 2, b = 3}

setmetatable(t1, mt)
local result = t1 + t2

for k, v in pairs(result) do
    print(k, v)  -- 输出 a: 3 和 b: 5
end

类型保护与封装

Lua 中的表并不具备传统语言中的封装保护，但是通过元表，你可以模拟简单的封装。例如，通过 __index 和 __newindex 来控制对表的访问：

local privateTable = {}

local mt = {
    __index = privateTable,
    __newindex = function(table, key, value)
        if key:match("^_") then
            rawset(privateTable, key, value) -- 允许以 _ 开头的字段写入
        else
            error(string.format("Attempt to modify readonly key: %s", key))
        end
    end
}

local myTable = {}
setmetatable(myTable, mt)

myTable._privateField = 10  -- 允许设置
print(myTable._privateField)  -- 输出 10

myTable.publicField = 20  -- 抛出错误

在上面的代码中，只允许以 _ 开头的键来进行设置，而普通键则是只读的。这种机制可以防止错误地修改表中的数据。

结论

在 Lua 中，setmetatable 与元表及元方法的结合使其成为一个功能强大的工具，能够扩展和自定义表的行为。通过元表，Lua 程序员可以实现运算符重载、访问控制和其他高级特性，使得程序更加灵活和易于维护。掌握这些特性将在 Lua 开发中大有帮助。