Использование named_scope с количеством дочерних моделей - PullRequest
Новая
       18

Использование named_scope с количеством дочерних моделей

4 голосов
/ 25 мая 2010

У меня есть простой родительский объект, имеющий много детей. Я пытаюсь выяснить, как использовать именованную область для возвращения только родителей с определенным количеством детей.

Возможно ли это? 

class Foo < ActiveRecord::Base
    has_many :bars
    named_scope :with_no_bars, ... # count of bars == 0
    named_scope :with_one_bar, ... # count of bars == 1
    named_scope :with_more_than_one_bar, ... # count of bars > 1
end

class Bar < ActiveRecord::Base
    belongs_to :foo
end

Я надеюсь сделать что-то вроде Foo.with_one_bar

Я мог бы написать методы в родительском классе для чего-то подобного, но я бы предпочел силу именованной области видимости

Ответы [ 2 ]

10 голосов
/ 25 мая 2010 
class Foo < ActiveRecord::Base
  has_many :bars

  # I don't like having the number be part of the name, but you asked for it.
  named_scope :with_one_bar, :joins => :bars, :group => "bars.foo_id", :having => "count(bars.foo_id) = 1"

  # More generically...
  named_scope :with_n_bars, lambda {|n| {:joins => :bars, :group => "bars.foo_id", :having => ["count(bars.foo_id) = ?", n]}}
  named_scope :with_gt_n_bars, lambda {|n| {:joins => :bars, :group => "bars.foo_id", :having => ["count(bars.foo_id) > ?", n]}}

end

Вызывается так: 

Foo.with_n_bars(2)
6 голосов
/ 25 мая 2010

Я бы использовал для этого кеш счетчика . Поэтому вам нужна следующая миграция: 

class AddBarCount < ActiveRecord::Migration
  def self.up  
    add_column :foos, :bars_count, :integer, :default => 0  

    Foo.reset_column_information  
    Foo.all.each do |p|  
      p.update_attribute :bars_count, p.bars.length  
    end  
  end  

  def self.down  
    remove_column :foos, :bars_count  
  end
end

Чем тебе тоже нужно поменять Bar модель вот так: 

class Bar < ActiveRecord::Base
  belongs_to :foo, :counter_cache => true
end

Теперь счет bars кэшируется в модели foo, что ускорит ваши запросы для счетчика bars.

Тогда ваши named_scopes тоже выглядят так: 

#rails 2
named_scope :with_no_bars, :conditions => { :bars_count => 0 }
named_scope :with_one_bar, :conditions => { :bars_count => 1 }
named_scope :with_more_than_one_bar, :conditions => ["bars_count > 1"]

#rails 3 & ruby 1.9+
scope :with_no_bars, where(bars_count: 0)
scope :with_one_bar, where(bars_count: 1)
scope :with_more_than_on_bar, where("bars_count > 1")

#rails 4* & ruby 1.9+
scope :with_no_bars, -> { where(bars_count: 0) }
scope :with_one_bar, -> { where(bars_count: 1) }
scope :with_more_than_one_bar, -> { where("bars_count > 1") }

Таким образом, вы можете сэкономить время, считая bars для каждого foo каждый раз, когда вы делаете такой запрос.

Мне пришла в голову мысль посмотреть Railscast о счетчике кеша: http://railscasts.com/episodes/23-counter-cache-column

* Что нового в Active Record [Rails 4 Countdown to 2013]

...