Elixir 宏,引用位串模式匹配类型

Elixir macros, quoting bitstring patternmatch types

我正在从事一个项目,我可能会在其中编写大量以下形式的代码:

defmodule Kind
  defstruct [name1, name2, name3]
  @type t :: %Kind{name1: integer(), name2: integer(), name3: binary()}
  def unpack(input) do
    with <<name1::integer-8>> <- Enum.take(input, 1),
      <<name2::integer-little-32>> <- Enum.take(input, 4),
      <<name3::binary-10>> <- Enum.take(input, 10),
    do: %Kind{name1: name1, name2: name2, name3: name3>>
  end
end

(对于任意组的输入名称和类型,input 是一次产生一个字节的二进制流)

如果能够在宏中处理它会非常有用,这样我就可以简单地编写(例如)use Unpack quote([{name1, integer-8}, {name2, integer-little-32}, {name3, binary-10}]) 并自动生成必要的结构、typedef 和解包函数,用于任意固定大小的命名字段。甚至可以扩展它,在元组中添加第三个字段以传递一个函数来处理可变大小的类型。不幸的是,当我尝试做一个更简单的版本时(只采用一个大小的字段,并且只匹配 1 到它):

defmodule Unpack do
  defmacro testmacro({name, kind}) do
    quote do
      <<unquote(name)::unqote(kind)>> = 1
    end
  end
end

系统告诉我 quote/1 的参数无效。我认为这是因为在位串模式匹配中使用的 "types" 是一种特殊形式,一般情况下位串文字也是如此,并且这些特定项目不会在其他任何地方使用。

那么,我该如何解决这个问题?我有十几种打包结构需要解包,每种都有五到二十个不同的字段。如果我不这样做,我可能会求助于 Vim 宏来至少拯救我的双手……但这对维护大量极其重复的代码并没有真正帮助。

两件事:您在 unquote 中有一个拼写错误,并且 RHS 必须是二进制的,这样模式才能匹配。通过这些更改,您的代码对我有用:

defmodule Unpack do
  defmacro unpack({name, kind}) do
    quote do
      <<unquote(name)::unquote(kind)>> = "a"
    end
  end
end

defmodule Main do
  import Unpack

  def main do
    unpack({foo, integer-8})
    IO.inspect foo
  end
end

Main.main

输出:

97