在没有迭代器的情况下，如何收集Ruby中的块值

Question

这是一些代码：

i = 0
collection = []
loop do
  i += 1
  break if complicated_predicate_of(i)
  collection << i
end
collection

我不知道我需要迭代多少次；这取决于complicated_predicate_of(i)。我可以做类似 0.upto(Float::INFINITY).times.collect do ... end的事情，但这很丑陋。

我想这样做：

i = 0
collection = loop.collect do
  i += 1
  break if complicated_predicate_of(i)
  i
end

但是，尽管由于某种原因，这不是语法错误，但loop.collect似乎没有收集任何内容。（loop.reduce也没有）。collection在语句结尾处为零。

换句话说，我想在没有明确迭代器的情况下收集loop语句的值。有什么方法可以实现这一目标吗？

Answer 1

你可以写

collection = 1.step.take_while do |i| 
  i <= 3 # this block needs to return *false* to stop the taking
end

---

最终选择的任何解决方案，请记住，您总是可以选择以自我解释的名称引入辅助方法。特别是如果您需要在源代码的许多地方收集这样的数字。

说您想隐藏上面解决方案的错综复杂的肠子，然后这可能是您的帮助方法：

def numbers_until(&block)
  i = 0
  collection = []
  loop do
    i += 1
    break if yield i
    collection << i
  end
  collection
end
collection = numbers_until do |i|
   i > 3 # this block needs to return *true* to stop the taking
end

Answer 2

你可以写

def complicated_predicate_of(i)
  i > 3
end
1.step.with_object([]) { |i,collection| complicated_predicate_of(i) ?
  (break collection) : collection << i }
  #=> [1, 2, 3]