获取包含模块的类的列表

Question

我有一个 mixin，我想获取包含它的所有类的列表。在 mixin 模块中，我做了以下操作：

module MyModule
  def self.included(base)
    @classes ||= []
    @classes << base.name
  end

  def self.classes
    @classes
  end
end

class MyClass
  include MyModule
end

这工作得很好：

> MyModule.classes #=> nil
> MyClass.new #=> #<MyClass ...>
> MyModule.classes #=> ["MyClass"]

现在，我想将这部分提取到一个单独的模块中，该模块可以包含在我的其他 mixin 中。所以，我想出了以下方法：

module ListIncludedClasses
  def self.included(base)
    p "...adding #{base.name} to #{self.name}.classes"

    @classes ||= []
    @classes << base.name

    base.extend(ClassMethods)
  end

  def self.classes
    @classes
  end

  module ClassMethods
    def included(module_base)
      p "...adding #{module_base.name} to #{self.name}.classes"

      @module_classes ||= []
      @module_classes << module_base.name
      super(module_base)
    end
    def classes
      @module_classes
    end
  end

end

module MyModule
  include ListIncludedClasses
end

但这不起作用，因为从 ListIncludedClasses 添加到 MyModule 的 #included(module_base) 方法永远不会运行。有趣的是，它确实成功地将 #classes 添加到 MyModule 中。

> MyModule.classes #=> 
  "...adding Rateable to ListIncludedClasses.classes"
  => nil 
> ListIncludedClasses #=> ["MyModule"]
> MyClass.new #=> #<MyClass ...>
# ^^ THIS SHOULD BE ADDING MyClass TO MyModule.classes ^^
> MyModule.classes #=> nil

我缺少什么？

Answer 1

module MyMod; end

class A; include MyMod; end
class B < A; end
class C; end

ObjectSpace.each_object(Class).select { |c| c.included_modules.include? MyMod }
  #=> [B, A]

请参阅 ObjectSpace::
每个_对象。

Answer 2

对 Cary 的答案（这太棒了！）的一个警告是，它只会选择已经由虚拟机评估过的类。

因此，如果您在开发 Rails 控制台等设置中运行该代码，则需要在检查该模块是否已包含之前显式 require 您感兴趣的文件。

你可以这样做：

Dir[Rails.root.join("app/models/**/*.rb")].each { |f| require f }

Answer 3

实际上，您的模块扩展模块可以工作。问题出在您的测试中：当您使用 Class.new 创建随机未命名类时，您忘记包含 MyModule。附带说明一下，您可以为包含该模块的类使用只读访问器，并使用有用的 Module#attr_reader 方法。

Answer 4

您可能应该使用扩展而不是包含，因为前者添加了类级别方法，而后者添加了实例级别方法（为什么您可以访问@classes）。

试试这个：

module MyModule
  extend ListIncludedClasses::ClassMethods
end