有没有一种聪明的方式编写Lua对象以使其作为迭代器?

假设我在其他地方定义了一些“对象”。也许它代表了一组项目,但比简单的表格更复杂。无论它是什么,迭代它会很合理。

因此,它定义了一个iterator方法。所以我可以这样写:

local myObject = AbstractObject:new()

for obj in myObject:iterator() do
    obj:foo()
end

我想知道的是,是否有一些元方法技巧可以使我写成这样:

local myObject = AbstractObject:new()

for obj in myObject do
    obj:foo()
end

那么是不是有这样的技巧呢?

原文链接 https://stackoverflow.com/questions/6113475

点赞
stackoverflow用户600500
stackoverflow用户600500

in 后面使用的对象必须是一个函数,这个函数将在 for 循环中被重复调用。

我不确定是否可以将表格或用户对象像函数一样被调用,但即使可以,问题也在于对象只能有一个内部迭代器状态——也就是说,它不允许多次迭代同一对象(无论是并发还是顺序迭代),除非你明确地重置它。

就像 Stuart 的回答一样,你可以适当地使用 __call 元方法来返回迭代器,但这样你就不得不写成下面这样:

for obj in myObject() do
    obj:foo()
end

这并不是我们想要的。

PiL 中进一步阅读,我发现在 for 循环中使用了更多的组件:不变的循环状态和控制变量的当前值,在每次调用迭代器函数时将它们传递进去。如果我们在 in 表达式中没有提供它们,它们将被初始化为 nil

因此,我的想法是使用这些值来区分各个调用。

如果你可以为你的集合创建一个 next(element) 函数,它会为每个元素返回下一个元素,那么实现就很简单了:

metatable.__call = function(_state, _last)
    if(_last == nil) then
       return obj:first()
    else
       return obj:next(_last)
   end
end

但通常情况下我们没有这样的东西,那么它就变得更加复杂。


我考虑在这里使用协程,但这些仍然需要一个工厂方法(我们想要避免这个)。 它会导致与 Stuart 写的类似的结果(即在对象本身或与对象相关的某个变量中保存迭代器状态),并使用参数和/或迭代器结果来决定何时创建/清除迭代器对象/状态。

没有一方获胜。

2011-05-24 16:24:32
stackoverflow用户34799
stackoverflow用户34799

一个小改动可以使语义变得更简单:

local myObject = AbstractObject:new()

for obj in myObject() do
    obj:foo()
end

这样,您可以使用元表来定义 __call 元方法以返回myObject:iterator(),具体如下所示:

setmetatable(newobject, {__call = function() return newobject:iterator() end})

如果没有迭代器构造,您将会使用一个迭代器来进行多次迭代,这意味着您需要在对象/创建闭包中保持迭代器状态,并在其结束时重置它,以便下一个调用将重新开始迭代。如果您真的想这么做,最好的解决方案确实是针对特定迭代实现编写一些东西,但这将执行通用迭代:

local iterator

--table.pack is planned for 5.2
local pack = table.pack or function(...)
  local t = {...}
  t.n = select('#',...)
  return t
end

--in 5.1 unpack isn't in table
local unpack = table.unpack or unpack

function metamethods.__call(...)
  if not iterator then
    iterator = newobject:iterator()
  end

  local returns = pack(iterator(...))

  if returns[1] == nil then
    --iteration is finished: next call will restart iteration
    iterator = nil
  end
  return unpack(returns, 1, returns.n)
end

再次强调:这应该 真的 根据您的用例进行调整。

2011-05-24 16:25:29