ruby-on-rails - 在 Rails 中通过多态类型查找所有内容？

Question

有没有办法在 Rails 中找到特定多态类型的所有多态模型？因此，如果我的 Group、Event 和 Project 都带有如下声明：

has_many :assignments, :as => :assignable

我可以做类似的事情：

Assignable.all

...或者

BuiltInRailsPolymorphicHelper.all("assignable")

那样就好了。

编辑：

...这样Assignable.all返回[Event, Group, Product]（类数组）

Answer 1

没有直接的方法。我为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']

Answer 2

您也可以尝试这种方式。因为上面的解决方案对我不起作用，因为我有一些 mongo 的模型。

def get_has_many_associations_for_model(associations_name, polymorphic=nil)

  associations_name = associations_name.to_s.parameterize.underscore.pluralize.to_sym
  active_models = ActiveRecord::Base.descendants
  get_model = []
  active_models.each do |model|

    has_many_associations =model.reflect_on_all_associations(:has_many).select{|a|a.name==associations_name }
    has_many_associations = has_many_associations.select{ |a| a.options[:as] == polymorphic.to_s.to_sym} if polymorphic.present?
    get_model << model if has_many_associations.present?

  end
  get_model.map{|a| a.to_s}
end

Anb 称之为

get_has_many_associations_for_model("assignments", "assignable")

如果您想要多态记录而不是传递它，则此处的第二个参数是可选的，否则将其保留为空白。

它将模型名称数组作为字符串返回。

Answer 3

Harish Shetty 的解决方案不适用于未直接存储在 Rails.root/app/models 中而是在子目录中的命名空间模型文件。尽管它正确地将文件放在子目录中，但在将文件名转换为常量时却无法包含子目录。这样做的原因是，命名空间子目录被此行删除：

klass = File.basename(file, ".rb").camelize.constantize rescue nil

这是我为保留命名空间子目录所做的工作：

file.sub!(File.join(Rails.root, "app", "models"), '')
file.sub!('.rb', '')
klass = file.classify.constantize rescue nil

这是完整的修改解决方案：

  def self.all_polymorphic_types(name)
    @poly_hash ||= {}.tap do |hash|
      Dir.glob(File.join(Rails.root, "app", "models", "**", "*.rb")).each do |file|
        file.sub!(File.join(Rails.root, "app", "models"), '')
        file.sub!('.rb', '')
        klass = file.classify.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

现在，该方法会将 /models/test/tensile.rb 正确转换为 Test::Tensile，然后再反映其关联。

只是一个小小的改进，所有的功劳都归于 Harish！

Answer 4

我创建了一个带有“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

Answer 5

您应该能够只使用关联的集合：

model.assignments.all