如何实现支持多表参数的Lua无状态迭代器？

嘿，我懂你这种感觉——闭包那套已经摸得差不多了，但要搞一个能同时处理多个表、挨个遍历元素的迭代器，确实得费点脑子。让我给你拆解一下怎么实现你要的功能，分两种方式讲：一种是用闭包的（直观好写，适合这个场景），另一种是你提到的无状态迭代器（满足你对无状态实现的需求）。

一、闭包实现（最直观的方式） #

这个思路很简单：用闭包记住当前遍历到哪个表，以及该表内的哪个元素位置，每次调用迭代器时就按顺序取元素，当前表遍历完就切换到下一个表。

function iter(...)
    -- 先收集所有非空表，跳过空表（避免遍历空表浪费资源）
    local valid_tables = {}
    for _, tbl in ipairs({...}) do
        -- 只保留表类型且非空的输入
        if type(tbl) == "table" and next(tbl) ~= nil then
            table.insert(valid_tables, tbl)
        end
    end

    -- 闭包内部维护遍历状态：当前表的索引、当前表内的元素索引
    local current_table_idx = 1
    local current_elem_idx = 0

    -- 返回迭代器函数
    return function()
        -- 循环检查当前及后续的表，直到找到有元素的位置
        while valid_tables[current_table_idx] do
            current_elem_idx = current_elem_idx + 1
            local val = valid_tables[current_table_idx][current_elem_idx]
            if val ~= nil then
                -- 找到元素，返回它
                return val
            else
                -- 当前表遍历完了，切换到下一个表，重置元素索引
                current_table_idx = current_table_idx + 1
                current_elem_idx = 0
            end
        end
        -- 所有表都遍历完了，返回nil终止for循环
        return nil
    end
end

测试你的例子 #

把你的测试代码放进去跑：

for i in iter({1,2,3}, {'a', 'b', 'c'}, {100, 101}, {}) do
    print(i)
end

输出完全符合你的预期：

1
2
3
a
b
c
100
101

二、无状态迭代器实现 #

如果你一定要用无状态迭代器（迭代器函数本身不保存状态，所有状态由外部传递），那我们可以把遍历状态封装在一个表里，由for循环来管理这个状态。

-- 无状态迭代器函数：接收状态和上一次的返回值（这里我们用不上第二个参数，用下划线忽略）
function multi_table_iterator(state, _)
    state.elem_idx = state.elem_idx + 1
    local val = state.tables[state.table_idx][state.elem_idx]
    
    if val ~= nil then
        return _, val
    else
        -- 当前表遍历完，切换到下一个表
        state.table_idx = state.table_idx + 1
        state.elem_idx = 0
        
        -- 继续检查后续的表，直到找到有元素的位置
        while state.tables[state.table_idx] do
            state.elem_idx = state.elem_idx + 1
            val = state.tables[state.table_idx][state.elem_idx]
            if val ~= nil then
                return _, val
            else
                state.table_idx = state.table_idx + 1
                state.elem_idx = 0
            end
        end
        -- 所有表遍历完毕，返回nil
        return nil
    end
end

-- 包装函数：处理输入参数，初始化状态
function multi_table_iter(...)
    local valid_tables = {}
    for _, tbl in ipairs({...}) do
        if type(tbl) == "table" and next(tbl) ~= nil then
            table.insert(valid_tables, tbl)
        end
    end

    -- 初始化遍历状态
    local state = {
        tables = valid_tables,
        table_idx = 1,
        elem_idx = 0
    }

    -- 返回迭代器函数、状态、初始值（第三个参数会作为迭代器的第二个参数传入第一次调用）
    return multi_table_iterator, state, nil
end

使用无状态版本 #

调用方式和之前类似：

for i in multi_table_iter({1,2,3}, {'a', 'b', 'c'}, {100, 101}, {}) do
    print(i)
end

输出结果和闭包版本完全一致。

额外注意点 #

• 上面的实现都是数组式遍历（和ipairs逻辑一致），只遍历表中索引为连续整数的元素；如果需要遍历表的所有键值对（类似pairs），可以把内部的索引遍历改成用next函数来遍历键值。
• 代码里已经处理了空表和非表的输入，确保只会遍历有效的表元素。
• 如果你需要保留空表的“占位”（虽然你的例子里空表没有输出），可以去掉过滤空表的逻辑，不过空表本身没有元素可遍历，所以不会影响最终输出。

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