如何在 Elixir 的同一个文件中加载另一个模块之前加载一个模块？

Question

我正在尝试实现 Protobuf 的一些功能，我有一个案例：一个模块需要在编译时调用另一个模块。它们在同一个文件中，很难确保它们的顺序。

defmodule FakeProtobuf do
  defmacro __using__(opts) do
    quote do
      # register_attribute fields and set fields
      @before_compile FakeProtobuf
    end
  end

  defmacro __before_compile__(_) do
    # result = Enum.map(fields, fn ...)

    # even Code.ensure_loaded doesn't work
    result = %{bar: Bar.default}
    quote do
      def default do
        unquote(Macro.escape(result))
      end
    end
  end
end

defmodule Foo do
  use FakeProtobuf
  # field :bar, type: Bar
end

defmodule Bar do
  def default do
    "bar"
  end
end

这段代码忽略了一些与字段相关的宏细节，但主要思想就像我上面说的一样。此代码无法编译，因为即使在调用 Code.ensure_loaded(Bar) 时，编译 Foo 时 Bar 也不可用。我需要这个，因为我想在编译时而不是运行时运行一些代码来节省一些时间。

如果 Bar 在 Foo 之前或在另一个文件中定义，它会起作用。但在 protobuf 生成的文件中很难确保这一点。

有没有办法解决这个问题？

Answer 1

我可以通过在其他模块之前简单地定义Bar 来编译您的代码。

当编译器开始编译您的模块FakeProtoBuf 时，它不知道该模块内的定义。

它开始编译文件并遇到您的__before__compile/1 宏。扩展该宏时，它需要调用模块 B 中的一个函数，编译器尚未看到该函数，因此会出错。

** (UndefinedFunctionError) function Bar.default/0 is undefined (module Bar is not available)

为了解决这个问题，我们将Bar 模块移到FakeProtoBuf 模块的定义之前。这样编译器可以安全地调用该函数，因为Bar 将被完全编译。

最后编译器将编译Foo，它将执行FakeProtoBuf中的__using__/1，这将注入@before_compile，从而调用FakeProtoBuf模块中的宏__before__compile__。

  def default do
    "bar"
  end
end

defmodule FakeProtobuf do
  defmacro __using__(_opts) do
    quote do
      # register_attribute fields and set fields
      @before_compile FakeProtobuf
    end
  end

  defmacro __before_compile__(_) do
    result = %{bar: Bar.default()}

    quote do
      def default do
        unquote(Macro.escape(result))
      end
    end
  end
end

defmodule Foo do
  use FakeProtobuf
  # field :bar, type: Bar
end