Ruby 1.9枚举更改

时间:2010-02-08 12:29:20

标签: ruby ruby-1.9

所以,我不太了解Ruby 1.9对枚举器的更改,但我想要的是对集合的惰性过滤/映射,所以例如我如何创建一个等效的惰性枚举器这...

(1..100).find_all{|x| x % 2 == 1}.map{|x| x * 2}.find_all{|x| x % 3 == 0}

我尝试将lambda传递给#enum_for,但它在我的机器上不起作用。

2 个答案:

答案 0 :(得分:3)

您可以找到lazy_select和lazy_map in this blog的实现。您只需要通过这两种方法扩展Enumerator模块。然后你应该可以使用

 (1..100).lazy_select{|x| x % 2 == 1}.lazy_map{|x| x * 2}.lazy_select{|x| x % 3 == 0}

答案 1 :(得分:0)

可悲的是,你不能用enum_for做到这一点。你必须使用这样的东西来做懒惰的枚举:

class LazyEnum
  include Enumerable

  attr_accessor :enum, :operations

  def initialize enum
    @enum = enum
    @operations = []
  end

  def each
    enum.each do |x|
      filtered = false
      operations.each do |type, op|
        if type == :filter
          unless op[x]
            filtered = true
            break
          end
        else
          x = op[x]
        end
      end
      yield x unless filtered
    end
  end

  def map! &blk
    @operations << [:transform, blk]
    self
  end

  def select! &blk
    @operations << [:filter, blk]
    self
  end

  def reject!
    select! {|x| !(yield x)}
  end

  def dup
    LazyEnum.new self
  end

  def map &blk
    dup.map! &blk
  end

  def select &blk
    dup.select! &blk
  end

  def reject &blk
    dup.reject! &blk
  end
end

LazyEnum.new(1..100).select{|x| x % 2 == 1}.map{|x| x * 2}.select{|x| x % 3 == 0}
#=> #<LazyEnum:0x7f7e11582000 @enum=#<LazyEnum:0x7f7e115820a0 @enum=#<LazyEnum:0x7f7e11582140 @enum=#<LazyEnum:0x7f7e115822d0 @enum=1..100, @operations=[]>, @operations=[[:filter, #<Proc:0x00007f7e11584058@(irb):348>]]>, @operations=[[:transform, #<Proc:0x00007f7e11583a90@(irb):348>]]>, @operations=[[:filter, #<Proc:0x00007f7e115823c0@(irb):348>]]>
irb(main):349:0> LazyEnum.new(1..100).select{|x| x % 2 == 1}.map{|x| x * 2}.select{|x| x % 3 == 0}.to_a
#=> [6, 18, 30, 42, 54, 66, 78, 90, 102, 114, 126, 138, 150, 162, 174, 186, 198]