Lua，自定义迭代器 - 定义的正确方式？

local function phones(d)
   local cn, c, tn, t, i
   return
      function()
         local a
         repeat
            if tn then
               a, i = t[i], i+1
               if not a then
                  i, tn, t = 1, next(c, tn)
               end
            else
               cn, c = next(d, cn)
               i, tn, t = 1, next(c or {})
            end
         until a or not cn
         return cn, tn, a
      end
end

-- 遍历 data 表中的电话信息
for country, town, abonent in phones(data) do
   print(country, town, abonent.name, abonent.number)
end

你可以在Lua中创建自己的自定义迭代器，无需使用协程。迭代器是函数，当被调用时，返回你的结构中的下一个元素（你可以使用任何结构）

针对你的示例，迭代器可能如下所示：

function corIter(data)
    local itOut = {}

    for country,l1 in pairs(data) do
        for city,l2 in pairs(l1) do
            for _,entry in pairs(l2) do
                -- 拷贝条目
                local out = {}
                for k,v in pairs(entry) do
                    out[k] = v
                end

                out.country = country
                out.city = city

                table.insert(itOut,out)
            end
        end
    end

    local i = 0
    return function()
        i = i + 1
        return itOut[i]
    end
end

'corIter'返回的匿名函数将从你的数据中返回下一个元素。请注意，当我们使用'pairs'将条目复制到另一个表中进行迭代时，没有任何保证条目的顺序会保持原始顺序。

因此，现在您可以使用以下代码打印出条目：

for entry in corIter(data) do
    print(entry) -- 这是一个表
    for k,v in pairs(entry) do
        print(k,v) -- 这些是实际值
    end
end