跟踪“使用”另一个模块的模块

Question

我正在尝试在编译时跟踪 use 另一个模块的所有模块，并在运行时获取所有这些模块的列表：

defmodule BaseModule do
  defmacro __using__(_) do
    quote do
      # common behaviour
    end
  end
end


defmodule A do
  use BaseModule
end

defmodule B do
  use BaseModule
end

defmodule C do
  use BaseModule
end

并在运行时通过调用类似这样的方法获取它们：

BaseModule.children()
#=> [A, B, C]

---

我一直在尝试找到一种方法来实现这一点，但仍然完全不知道如何去做。通过this thread on the elixir-lang mailing list、@josevalim 建议使用Protocols 来执行此操作。但是在为此苦苦挣扎了大约一个小时后，我也无法让它与协议一起使用。

我也一直在研究 Registry 模块，看看我是否可以使用它来完成此任务，但它看起来主要是为处理进程而设计的。

任何帮助将不胜感激。提前致谢！

Answer 1

如果您的基本模块使用行为，则可以使用以下解决方案。

defmodule BaseModule do
  @callback foo() :: any()

  defmacro __using__(_opts) do
    quote do
      # Fairly standard to use a behaviour and gives us something to detect later.
      @behaviour unquote(__MODULE__)

      # Any other setup required
    end
  end

  def children() do
    (for {module, _} <- :code.all_loaded(), do: module)
    |> Enum.filter(&is_child?/1)
  end

  def is_child?(module) do
    module.module_info[:attributes]
    |> Keyword.get(:behaviour, [])
    |> Enum.member?(__MODULE__)
  end
end

defmodule A do
  use BaseModule
end

BaseModule.children()
#=> [A]