webdevqa.jp.net

指定された順序でIDによってActiveRecordオブジェクトを見つけるクリーンな方法

IDの配列を指定してActiveRecordオブジェクトの配列を取得したいと思います。

私は

Object.find([5,2,3])

オブジェクト5、オブジェクト2、オブジェクト3の順に配列を返しますが、代わりに、オブジェクト2、オブジェクト3、オブジェクト5の順に配列を取得します。

ActiveRecord Base find method API は、提供された順序でそれを期待すべきではないと述べています(他のドキュメントではこの警告は示されていません)。

1つの潜在的な解決策が 同じ順序でIDの配列で検索しますか? で与えられましたが、順序オプションはSQLiteでは有効ではないようです。

いくつかのRubyコードを記述して、オブジェクトを自分で並べ替えることができます(多少単純でスケーリングが不十分であるか、スケーリングが優れていてより複雑です)が、より良い方法はありますか?

26
Andrew Grimm

MySQLや他のDBが独自にソートするのではなく、ソートしないということです。 Model.find([5, 2, 3])を呼び出すと、生成されるSQLは次のようになります。

_SELECT * FROM models WHERE models.id IN (5, 2, 3)
_

これは順序を指定するのではなく、返されるレコードのセットのみを指定します。通常、MySQLはデータベースの行を_'id'_の順序で返しますが、これを保証するものではありません。

データベースが保証された順序でレコードを返すようにする唯一の方法は、order句を追加することです。レコードが常に特定の順序で返される場合は、並べ替え列をデータベースに追加してModel.find([5, 2, 3], :order => 'sort_column')を実行できます。そうでない場合は、コードで並べ替えを行う必要があります。

_ids = [5, 2, 3]
records = Model.find(ids)
sorted_records = ids.collect {|id| records.detect {|x| x.id == id}} 
_
22
tomafro

Jeroen van Dijkへの私の以前のコメントに基づいて、これをより効率的に、each_with_objectを使用して2行で行うことができます。

result_hash = Model.find(ids).each_with_object({}) {|result,result_hash| result_hash[result.id] = result }
ids.map {|id| result_hash[id]}

参考までに、私が使用したベンチマークは

ids = [5,3,1,4,11,13,10]
results = Model.find(ids)

Benchmark.measure do 
  100000.times do 
    result_hash = results.each_with_object({}) {|result,result_hash| result_hash[result.id] = result }
    ids.map {|id| result_hash[id]}
  end
end.real
#=>  4.45757484436035 seconds

今もう一方

ids = [5,3,1,4,11,13,10]
results = Model.find(ids)
Benchmark.measure do 
  100000.times do 
    ids.collect {|id| results.detect {|result| result.id == id}}
  end
end.real
# => 6.10875988006592

更新

ほとんどの場合、orderステートメントとcaseステートメントを使用してこれを行うことができます。これは、使用できるクラスメソッドです。

def self.order_by_ids(ids)
  order_by = ["case"]
  ids.each_with_index.map do |id, index|
    order_by << "WHEN id='#{id}' THEN #{index}"
  end
  order_by << "end"
  order(order_by.join(" "))
end

#   User.where(:id => [3,2,1]).order_by_ids([3,2,1]).map(&:id) 
#   #=> [3,2,1]
10
Schneems

どうやらmySQLと他のDB管理システムはそれ自身で物事を分類します。私はあなたがそれをすることをバイパスすることができると思います:

ids = [5,2,3]
@things = Object.find( ids, :order => "field(id,#{ids.join(',')})" )
6
marcgg

移植可能な解決策は、ORDERBYでSQLCASEステートメントを使用することです。 ORDER BYではほとんどすべての式を使用でき、CASEはインラインルックアップテーブルとして使用できます。たとえば、必要なSQLは次のようになります。

select ...
order by
    case id
    when 5 then 0
    when 2 then 1
    when 3 then 2
    end

これは、Rubyを少し使用することで非常に簡単に生成できます。

ids = [5, 2, 3]
order = 'case id ' + (0 .. ids.length).map { |i| "when #{ids[i]} then #{i}" }.join(' ') + ' end'

上記は、idsで数値またはその他の安全な値を使用していることを前提としています。そうでない場合は、使用する必要があります connection.quote または ActiveRecord SQLサニタイザーメソッド の1つで、idsを適切に引用します。

次に、注文条件としてorder文字列を使用します。

Object.find(ids, :order => order)

または現代の世界では:

Object.where(:id => ids).order(order)

これは少し冗長ですが、どのSQLデータベースでも同じように機能するはずであり、醜さを隠すことはそれほど難しくありません。

6
mu is too short

here と答えたので、次のようなネイティブSQLの順序付けを可能にするgem( order_as_specified )をリリースしました。

Object.where(id: [5, 2, 3]).order_as_specified(id: [5, 2, 3])

テストしたばかりで、SQLiteで動作します。

4
JacobEvelyn

Justin Weissが書いた この問題に関するブログ記事 ちょうど2日前。

データベースに優先順位を通知し、その順序でソートされたすべてのレコードをデータベースから直接ロードすることは、良いアプローチのようです。彼の例 ブログ記事

# in config/initializers/find_by_ordered_ids.rb
module FindByOrderedIdsActiveRecordExtension
  extend ActiveSupport::Concern
  module ClassMethods
    def find_ordered(ids)
      order_clause = "CASE id "
      ids.each_with_index do |id, index|
        order_clause << "WHEN #{id} THEN #{index} "
      end
      order_clause << "ELSE #{ids.length} END"
      where(id: ids).order(order_clause)
    end
  end
end

ActiveRecord::Base.include(FindByOrderedIdsActiveRecordExtension)

それはあなたが書くことを可能にします:

Object.find_ordered([2, 1, 3]) # => [2, 1, 3]
3
spickermann

方法として、パフォーマンスの高い(O(n)検出のように配列検索ではありません!)ワンライナーではありません!)

def find_ordered(model, ids)
  model.find(ids).map{|o| [o.id, o]}.to_h.values_at(*ids)
end

# We get:
ids = [3, 3, 2, 1, 3]
Model.find(ids).map(:id)          == [1, 2, 3]
find_ordered(Model, ids).map(:id) == ids
2
Kanat Bolazar

これが私が思いつくことができる最も簡単なことです:

ids = [200, 107, 247, 189]
results = ModelObject.find(ids).group_by(&:id)
sorted_results = ids.map {|id| results[id].first }
1
jasongarber

Rubyでそれを行う別の(おそらくより効率的な)方法:

ids = [5, 2, 3]
records_by_id = Model.find(ids).inject({}) do |result, record| 
  result[record.id] = record
  result
end
sorted_records = ids.map {|id| records_by_id[id] }
1
Jeroen van Dijk