Найти все по полиморфному типу в Rails? - PullRequest
10 голосов
/ 23 февраля 2010

Есть ли способ найти все полиморфные модели определенного полиморфного типа в Rails? Поэтому, если у меня есть Group, Event и Project, все с объявлением вроде:

has_many :assignments, :as => :assignable

Могу ли я сделать что-то вроде:

Assignable.all

... или

BuiltInRailsPolymorphicHelper.all("assignable")

Это было бы неплохо.

Edit:

... такой, что Assignable.all возвращает [Event, Group, Product] (массив классов)

Ответы [ 3 ]

16 голосов
/ 23 февраля 2010

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

class ActiveRecord::Base

  def self.all_polymorphic_types(name)
    @poly_hash ||= {}.tap do |hash|
      Dir.glob(File.join(Rails.root, "app", "models", "**", "*.rb")).each do |file|
        klass = File.basename(file, ".rb").camelize.constantize rescue nil
        next unless klass.ancestors.include?(ActiveRecord::Base)

        klass.
          reflect_on_all_associations(:has_many).
          select{ |r| r.options[:as] }.
          each do |reflection|
            (hash[reflection.options[:as]] ||= []) << klass
          end
      end
    end
    @poly_hash[name.to_sym]
  end

end

Теперь вы можете сделать следующее:

Assignable.all_polymorphic_types(:assignable).map(&:to_s)
# returns ['Project', 'Event', 'Group']
0 голосов
/ 23 февраля 2010

Я создал класс полиморфной модели с методом 'all', чтобы проверить это.

class Profile
  # Return all profile instances
  # For class return use 'ret << i' instead of 'ret << i.all'
  def self.all
    ret = []
    subclasses_of(ActiveRecord::Base).each do |i|
      unless i.reflect_on_all_associations.select{|j| j.options[:as] == :profile}.empty?
        ret << i
      end
    end
    ret.flatten
  end

  def self.all_associated
    User.all.map{|u| u.profile }.flatten
  end
end

Вот мои настройки приложения:

User < ActiveRecord::Base
  belongs_to :profile, :polymorphic => true
end

Student < ActiveRecord::Base
  has_one :user, :as => :profile
end
0 голосов
/ 23 февраля 2010

Вы можете просто использовать связанную коллекцию:

model.assignments.all
...