无限枚举器倒带

Question

我有一个以下列方式生成枚举器的函数：

def create_example_enumerator(starting_value)
  current = starting_value
  e = Enumerator.new do |y|
    loop do
      current += 1
      y << current
    end
  end
end

当前的行为非常简单。

> e = create_example_enumerator(0)
#<Enumerator: details>
> e.next
1
> e.next
2
> e.rewind
#<Enumerator: details>
> e.next
3

我希望e.rewind 将枚举器重置为其初始值。 有没有一种很好的方法可以在仍然使用无限枚举器的同时做到这一点？

Answer 1

这应该可行：

n = Enumerator.new do |y|
  number = 1
  loop do
    y.yield number
    number += 1
  end
end

n.next #=> 1
n.next #=> 2
n.next #=> 3
n.rewind
n.next #=> 1