ruby - 你如何创建一个枚举器包装类？

Question

我有这个功能：

def file_parser (filename)
  Enumerator.new do |yielder|
    File.open(filename, "r:ISO-8859-1") do |file|
      csv = CSV.new(file, :col_sep => "\t", :headers => true, :quote_char => "\x07")                            
      csv.each do |row|
        yielder.yield map_fields(clean_data(row.to_hash))
      end
    end
  end
end

我可以这样使用它：

parser = file_parser("data.tab")
parser.each do { |data| do_profitable_things_with data }

相反，我想把它放在自己的类中并像这样使用它：

parser = SpecialParser.new("data.tab")
parser.each do { |data| do_profitable_things_with data }

我已经尝试了一些我没想到会起作用的事情，比如只是将枚举数从initialize(), 和self = file_parser().

我也试过了super do |yielder|。

出于某种原因，我没有办法做到这一点。

Answer 1

您可以将Enumerable模块包含到您的类中，并定义一个each调用yield.

您仍然可以免费获得所有Enumerable方法，如map,reduce等。

class SpecialParser
  include Enumerable

  def initialize(n)
    @n = n
  end

  def each
    0.upto(@n) { |i| yield i }
  end
end

sp = SpecialParser.new 4
sp.each { |i| p i }
p sp.map { |i| i }

输出：

0
1
2
3
4
[0, 1, 2, 3, 4]

Answer 2

file_parser在SpecialParser. _

然后像这样设置类的其余部分：

class SpecialParser
  include Enumerable  # needed to provide the other Enumerable methods

  def initialize(filename)
    @filename = filename
    @enum = file_parser(filename)
  end

  def each
    @enum.each do |val|
      yield val
    end
  end
end

编辑：

如果您想要免费的其他 Enumerable 方法，您还必须include Enumerable在课堂上。