mapやselectメソッドで、配列以外を返す書き方

Question

コードレビューの中で以下のようなコード（あくまでサンプル）を見つけました。

class HistoryCollection
  def initialize(array)
    @array = array
  end

  def select_by_name(name)
    arr = @array.select { |history| history.name == name }
    HistoryCollection.new(arr)
  end
end

ここで、一度selectメソッドで絞った後に、同じCollectionとして返すためにHistoryCollection.newをするのを見て、自分は「気持ち悪いな」と思いました。
なので、これを無くせないかコードをいじろうとしてみました。

そこで、パッと思いついたのが、Enumerableをincludeすればいけるかなと思ったのですが
想定と違っていましたので少し見てください。

class HistoryCollection
  include Enumerable

  delegate :each, to: :@array

  def initialize(array)
    @array = array
  end

  def select_by_name(name)
    select { |history| history.name == name }
  end
end

実際やってみるとこの時のselect_by_nameの返り値はArrayクラス。

Enumerableをincludeしても、mapやselectメソッドで返ってくる値がいつの間にはArrayに変わってしまいます。

集合を作るつもりでCollectionの名前を冠するクラスを用意したのに、知らない間にクラスが変わってしまうのは辛いので、mapやselectなどを実行した時に、同じクラスの集合を返すような書き方というのは無いのでしょうか？

Answer 1

Enumerable#selectはArrayを返すので、当然じゃないでしょうか。

Answer 2

Enumerableをincludeしても、mapやselectメソッドで返ってくる値がいつの間にはArrayに変わってしまいます。

はい、Enumerable#mapやEnumerable#selectの返り値はArrayです。

集合を作るつもりでCollectionの名前を冠するクラスを用意したのに、知らない間にクラスが変わってしまうのは辛いので

でしたら自分で実装してください。

Comments

[maisumakun]

むしろ、「mapやselect」といったメソッドでは、Arrayを返すほうが自然な実装です（Hash#mapすらArrayを返します）。