Question

我有一张地图，希望将单一的事实来源用于多个功能。可以说是

source_of_truth = %{a: 10, b: 20}

我希望该映射的键为EctoEnum的值。 EctoEnum提供了一个宏defenum，我应该这样使用：

  defenum(
    EnumModule,
    :enum_name,
    [:a, :b]
  )

我不想重复[:a, :b]部分。我想像这样使用地图上的键：

  defenum(
    EnumModule,
    :enum_name,
    Map.keys(source_of_truth)
  )

它不起作用，因为defenum宏需要一个普通列表。

我认为可以通过定义自己的宏来做到这一点：

 defmacro dynamic_enum(enum_module, enum_name, enum_values) do
   quote do
     defenum(
       unquote(enum_module),
       unquote(enum_name),
       unquote(enum_values)
     )
   end
 end

然后致电：

dynamic_enum(EnumModule, :enum_name, Map.keys(source_of_truth))

但是，它做的却不一样：enum_values不是预先计算的列表，而是Map.get的AST。我的下一个方法是：

 defmacro dynamic_enum(enum_module, enum_name, enum_values) do
   quote do
     values = unquote(enum_values)
     defenum(
       unquote(enum_module),
       unquote(enum_name),
       ?
     )
   end
 end

不确定我可以将?放在哪里。我不能只放values，因为它是变量而不是列表。我也不能放unquote(values)。

一种可行的解决方案是这样的：

defmacro dynamic_enum(enum_module, enum_name, enum_values) do
  {values, _} = Code.eval_quoted(enum_values)
  quote do
    defenum(
      unquote(enum_module),
      unquote(enum_name),
      unquote(values)
    )
  end
end

但是，文档说在宏中使用eval_quoted是一种不好的做法。

[编辑] 使用Macro.expand的解决方案也不起作用，因为它实际上没有评估任何内容。扩展在以下位置停止：

Expanded: {{:., [],
  [
    {:__aliases__, [alias: false, counter: -576460752303357631], [:Module]},
    :get_attribute
  ]}, [],
 [
   {:__MODULE__, [counter: -576460752303357631], Kernel},
   :keys,
   [
     {:{}, [],
      [
        TestModule,
        :__MODULE__,
        0,
        [
          file: '...',
          line: 16
        ]
      ]}
   ]
 ]}

因此它不会扩展到我们期望的列表中。

[\ EDIT]

什么是解决该问题的好方法？

Answer 1

如Macro.expand/2文档中所述

以下内容已展开：


宏（本地或远程）

扩展别名（如果可能）并返回原子

编译环境宏（__CALLER__/0，__DIR__/0，__ENV__/0和__MODULE__/0）

模块属性读取器（@foo）

重点是我的。因此，可以将模块属性与Macro.expand/2一起使用。

  defmacro dynamic_enum(enum_module, enum_name, enum_values) do
    IO.inspect(enum_values, label: "Passed")
    expanded = Macro.expand(enum_values, __CALLER__)
    IO.inspect(expanded, label: "Expanded")

    quote do
      defenum(
        unquote(enum_module),
        unquote(enum_name),
        unquote(expanded)
      )
    end
  end

并这样称呼：

  @source_of_truth %{a: 10, b: 20}
  @keys Map.keys(@source_of_truth)

  def test_attr do
    dynamic_enum(EnumModuleA, :enum_name_a, @keys)
  end

FWIW，完整代码：

$ \cat lib/eenum.ex

defmodule Eenum do
  import EctoEnum

  defmacro dynamic_enum(enum_module, enum_name, enum_values) do
    IO.inspect(enum_values, label: "Passed")
    expanded = Macro.expand(enum_values, __CALLER__)
    IO.inspect(expanded, label: "Expanded")

    quote do
      defenum(
        unquote(enum_module),
        unquote(enum_name),
        unquote(expanded)
      )
    end
  end
end

$ \cat lib/tester.ex

defmodule Tester do
  import Eenum

  @source_of_truth %{a: 10, b: 20}
  @keys Map.keys(@source_of_truth)

  def test_attr do
    dynamic_enum(EnumModuleA, :enum_name_a, @keys)
  end
end

FWIW2。要能够从模块作用域中如上所示调用dynamic_enum，您需要做的是（surprise :)另一个模块作用域，该作用域已经在宏的那一刻进行了编译调用：

defmodule Defs do
  @source_of_truth %{a: 10, b: 20}
  @keys Map.keys(@source_of_truth)

  defmacro keys, do: Macro.expand(@keys, __CALLER__)
end

defmodule Tester do
  import Defs
  import Eenum

  dynamic_enum(EnumModuleA, :enum_name_a, keys())
end

FWIW 3.后者（带有定义的显式模块）即使不需要具有模块属性也可以工作：

defmodule Defs do
  defmacro keys, do: Macro.expand(Map.keys(%{a: 10, b: 20}), __CALLER__)
end

defmodule Tester do
  import Defs
  import Eenum

  dynamic_enum(EnumModuleA, :enum_name_a, keys())
end

经验法则是，当您发现自己需要调用Code.eval_quoted/3并将此代码放入独立模块中并使编译器为您调用此代码编译时。对于功能来说，它是在模块级别上工作的，对于模块级别，应将其放入另一个模块中，以使模块上下文（即__CALLER__和__ENV__）可用。

Answer 2

前一段时间我在同一个问题上作斗争。基本上，您可以在quote中构建语法树，使用unquote注入动态值，然后使用Code.eval_quoted评估宏：

options = Map.keys(source_of_truth)

Code.eval_quoted(
  quote do
    EctoEnum.defenum(MyEnum, :type_name, unquote(options))
  end,
  [],
  __ENV__
)

将计算列表传递给Elixir宏

2 个答案: