ActiveRecord нетерпеливый load multiple принадлежит ассоциациям

Проблема

У меня есть следующая модель ActiveRecord: 

class Person
  belongs_to :favourite_car, class_name: 'Car'
  belongs_to :business_car, class_name: 'Car'
  belongs_to :home_car, class_name: 'Car'
end

Когда я хочу получить доступ ко всем трем из этих ассоциаций, он генерирует три запроса select: 

SELECT * FROM cars WHERE cars.id = ?

Это, по существу, проблемаN+1 .

В идеале, я бы хотел, чтобы он генерировал только один запрос вида 

SELECT * FROM cars WHERE cars.id IN (?, ?, ?)

Возможное решение

Я мог бы переместить это в ассоциацию has_many :through => :join_table со столбцом в таблице join, чтобы указать тип ассоциации, а затем использовать includes([:join_table, :cars]) нетерпеливо загружать ассоциации. Однако в этом случае он только сокращает 3 запроса до 2 и вводит дополнительную таблицу.

Другое возможное решение

Другим возможным решением было бы вручную загрузить ассоциации следующим образом: 

module EagerLoader
  def eager_load(*associations)
    reflections = associations.map { |association| self.class.reflections[association.to_sym] }
    raise 'Not all are valid associations' if reflections.any?(&:nil?)

    reflections.group_by { |association| association.klass }.each do |klass, reflections|
      load_associations(klass, reflections)
    end

    self

  end

private

  def load_associations(klass, reflections)

    primary_key = klass.primary_key

    ids = reflections.map { |reflection| public_send(reflection.foreign_key) }

    records = klass.where(id: ids)


    reflections.each_with_index do |reflection, i|
      record = records.find do |record|
        record.public_send(primary_key) == ids[i]
      end

      public_send("#{reflection.name}=", record)
    end

  end


end
Я проверил его, и он работает. 
class Person
  include EagerLoader
end

Person.find(2).eager_load(:favorite_car, :business_car, :home_car)
Тем не менее, это все еще не помогает вам, когда вы хотите делать такие вещи, как 
Person.includes(:favourite_car, :business_car, :home_car)

Например, на странице индекса персон. Это уменьшает количество запросов от 3N+1 до 4, но на самом деле нужны только 2.

Есть ли лучшие решения этой проблемы?

3 8
ruby-on-rails rails-activerecord eager-loading

3 ответа:

Есть отличный пост о ручной нагрузке.

Http://mrbrdo.wordpress.com/2013/09/25/manually-preloading-associations-in-rails-using-custom-scopessql/

Я думаю, это то, что вы искали: 

owners = People.all
association_name = :photos

owners.each do |owner|
   record = whatever_you_want

   association = owner.association(association_name)
   association.target = record
   association.set_inverse_instance(record)
end
3

Попробуйте вот что: 

> person_id = 1
> person = Person.includes(:favorite_car, :business_car, :home_car).where("people.id" = ?", person_id).references(:favorites_car, :business_car, :home_car)
> person[0].favorite_car

Приложение / модели / персона.rb 

class Person < ActiveRecord::Base
  # columns: id, name, favorite_car_id, business_car_id, home_car_id
  belongs_to :favorite_car, class_name: 'Car'
  belongs_to :business_car, class_name: 'Car'
  belongs_to :home_car, class_name: 'Car'
end

Приложение / модели / автомобиль.rb 

class Car < ActiveRecord::Base
  # columns: id, name
  has_many :people
end

Доказательство того, что это работает: 

> Person.all
  Person Load (0.3ms)  SELECT "people".* FROM "people"
=> #<ActiveRecord::Relation [#<Person id: 1, favorite_car_id: 1, business_car_id: 2, home_car_id: 3, name: "Frankie", created_at: "2014-02-18 21:51:58", updated_at: "2014-02-18 21:53:34">]>

> Car.all
  Car Load (0.3ms)  SELECT "cars".* FROM "cars"
=> #<ActiveRecord::Relation [#<Car id: 1, name: "Mazda", created_at: "2014-02-18 21:52:16", updated_at: "2014-02-18 21:52:16">, #<Car id: 2, name: "Honda", created_at: "2014-02-18 21:52:20", updated_at: "2014-02-18 21:52:20">, #<Car id: 3, name: "BMW", created_at: "2014-02-18 21:52:24", updated_at: "2014-02-18 21:52:24">]>

> Person.includes(:favorite_car, :business_car, :home_car).where("people.id = ?", 1).references(:favorite_car, :business_car, :home_car)
  SQL (0.4ms)  SELECT "people"."id" AS t0_r0, "people"."favorite_car_id" AS t0_r1, "people"."business_car_id" AS t0_r2, "people"."home_car_id" AS t0_r3, "people"."name" AS t0_r4, "people"."created_at" AS t0_r5, "people"."updated_at" AS t0_r6, "cars"."id" AS t1_r0, "cars"."name" AS t1_r1, "cars"."created_at" AS t1_r2, "cars"."updated_at" AS t1_r3, "business_cars_people"."id" AS t2_r0, "business_cars_people"."name" AS t2_r1, "business_cars_people"."created_at" AS t2_r2, "business_cars_people"."updated_at" AS t2_r3, "home_cars_people"."id" AS t3_r0, "home_cars_people"."name" AS t3_r1, "home_cars_people"."created_at" AS t3_r2, "home_cars_people"."updated_at" AS t3_r3 FROM "people" LEFT OUTER JOIN "cars" ON "cars"."id" = "people"."favorite_car_id" LEFT OUTER JOIN "cars" "business_cars_people" ON "business_cars_people"."id" = "people"."business_car_id" LEFT OUTER JOIN "cars" "home_cars_people" ON "home_cars_people"."id" = "people"."home_car_id" WHERE (people.id = 1)
=> #<ActiveRecord::Relation [#<Person id: 1, favorite_car_id: 1, business_car_id: 2, home_car_id: 3, name: "Frankie", created_at: "2014-02-18 21:51:58", updated_at: "2014-02-18 21:53:34">]>

Важное примечание: Rails будет автоматически использовать people как множественное число person. Поэтому, когда вы создаете модель Person, она создаст таблицу базы данных people.

7

Я бы лично создал таблицу соединений. Тем не менее, вот еще один вариант: можно определить отношение на основе SQL-запросов: 

class Person
  belongs_to :favourite_car, class_name: 'Car'
  belongs_to :business_car, class_name: 'Car'
  belongs_to :home_car, class_name: 'Car'

  has_many :cars, class_name: 'Car', :finder_sql => %q(
   SELECT DISTINCT cars.*
   FROM cars
   WHERE cars.id IN (#{c_ids})
  )

  def c_ids
    [favourite_car_id, business_car_id, home_car_id].compact.uniq.join(',')
  end
end

Тогда вы можете сделать человека.включает в себя (: автомобили)

0