rails モデルで作成したメソッドがtrueのものを一括で取得

Question

userモデルで、userが持つitemの中でメソッドにかけて、trueのものだけを取得したく、以下のように実装しました。

  def exam_items
    item_ids = []
    items.each do |item|
      if item.has_exam? # itemモデルで定義した、true or falseを返すメソッド
        item_ids << item.id
      end
    end
    Item.where(id: item_ids.uniq)
  end

もう少しスマートな書き方はないでしょうか？・・・

Answer 1

has_exam? で返ってくる値は、DBの中には無いという前提でよいでしょうか？もしあるのであれば、普通にitems.where()で取れると思います。

DBにはないという前提ですと、書いていただいたコードだとitemsを取得するSQLが2回走ってしまうのが懸念点です。

  def exam_items
    item_ids = []
    items.each do |item| ## NOTE ここでselectが走る
      if item.has_exam? # itemモデルで定義した、true or falseを返すメソッド
        item_ids << item.id
      end
    end
    Item.where(id: item_ids.uniq) ## NOTE ここでもう一回走る
  end

こうするとクエリは一回で、かつシンプルに取得出来るかと思います。

def exam_items
  items.select {|item| item.has_exam? }
end

Answer 2

item_ids = []
items.each {|item| item_ids << item.id if item.has_exam? }
Item.where(id: item_ids.uniq)

スマートかどうか不明ですが、このように書くことはできます
が、短くなっただけで可読性を考えるとスマートとは言えないですね