如何在Lua中实现语法更友好的自定义foreach迭代器？

问题背景 #

需求：实现一个类似Lua中table.foreach的自定义迭代器，语法要更简洁友好。参考示例代码如下（不确定是纯Lua实现还是C扩展技巧）：

-- creature是Creature对象的集合
-- "ps"用作过滤变量

foreach creature cre "ps" do
    if cre.name ~= $name then
    -- 代码...
    end
end

尝试过table.foreach、ipairs等方法，但无法实现上述示例的效果，求实现思路。

实现思路 #

1. 纯Lua方案：利用语法糖模拟接近的效果 #

Lua允许函数调用时省略括号（当参数是字符串或表时），可以把foreach封装成一个高阶函数，接收集合、迭代变量名、过滤标识，再传入要执行的代码块（用字符串包裹）。

示例代码

-- 先定义过滤规则映射，比如"ps"对应特定的筛选逻辑
local filter_map = {
    ps = function(creature)
        -- 这里替换成"ps"实际要实现的过滤条件，比如筛选存活的对象
        return creature.is_alive == true
    end
}

function foreach(col, var_name, filter_key, code_block)
    -- 第一步：过滤集合
    local target_col = col
    if filter_key and filter_map[filter_key] then
        target_col = {}
        for _, item in ipairs(col) do
            if filter_map[filter_key](item) then
                table.insert(target_col, item)
            end
        end
    end

    -- 第二步：生成迭代代码并执行
    local exec_code = string.format([[
        for _, %s in ipairs(...) do
            %s
        end
    ]], var_name, code_block)
    
    -- 加载代码并传入过滤后的集合
    load(exec_code)(target_col)
end

-- 使用示例
local creature = {
    {name="wolf", is_alive=true},
    {name="bear", is_alive=false},
    {name="fox", is_alive=true}
}

foreach(creature, "cre", "ps", [[
    if cre.name ~= "wolf" then
        print("符合条件的对象：", cre.name)
    end
]])

2. 完全贴合示例语法：需语法扩展 #

如果想要完全实现foreach ... do这种无字符串包裹代码块的语法，纯Lua做不到——因为这属于自定义语法，需要两种方式：

• 预处理器：写一个脚本，把自定义的foreach语法转换成标准Lua代码（比如把foreach creature cre "ps" do ... end转成foreach(creature, "cre", "ps", function() ... end)），再执行转换后的代码。
• C扩展：修改Lua解释器的C源码，添加foreach作为关键字，自定义语法解析逻辑，适合需要深度定制的场景。

3. 简化实用版：放弃字符串过滤标识 #

如果不需要用字符串"ps"来指代过滤规则，直接传递过滤函数，语法会更简洁，纯Lua完全支持：

function foreach(col, filter, fn)
    -- 默认不过滤
    filter = filter or function() return true end
    for _, item in ipairs(col) do
        if filter(item) then
            fn(item)
        end
    end
end

-- 使用示例
foreach(creature, 
    function(cre) return cre.is_alive end, 
    function(cre)
        if cre.name ~= "wolf" then
            print("符合条件的对象：", cre.name)
        end
    end
)

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