转义 Julia 中宏生成宏中的表达式

Question

我想在 Julia 中编写一个宏来生成许多宏

@register_attribute foo

应该生成宏

@set_foo
@get_foo

反过来应该定义为

@set_foo x 5 == set_attr!(x, :foo, 5)

我已经成功地解决了这个问题的第一部分，但是@macroexpand显示内部变量x没有正确转义，如果我使用局部变量调用该方法，则会导致UndefVarErrors 。

@macroexpand @set_foo x 5 =
    :(Main.set_attr!(Main.x, (Symbol)("foo"), 5))

而我想要的是

@macroexpand @set_foo x 5 =
    :(set_attr!(x, (Symbol)("foo"), 5))

我的代码是

macro register_attribute(name)
    setn = Symbol("set_", string(name))
    getn = Symbol("get_", string(name))
    arg = gensym()
    arg2 = gensym()
    nsym = string(name)
    return Expr(:block, 
    esc(:(
        macro $(setn)($arg, $arg2)
            :(set_attr!($$arg, $$Symbol($$nsym), $$arg2)) 
        end)
        ),
    esc(:(
        macro $(getn)($arg, $arg2)
            :(get_attr!($$arg, $$Symbol($$nsym), $$arg2)) 
        end)
        ))
    e2
end

Answer 1

这是我将如何解决整个问题：

# for purely generative tasks like this, where the generated code is
# (presumably?) destined to be evalled at top level, I prefer using @eval
for attr in [:foo, :bar]
    get_attr_name = Symbol("get_", attr)
    set_attr_name = Symbol("set_", attr)
    quoted_attr   = QuoteNode(attr)

    @eval begin
        $get_attr_name(x)      = getproperty( x, $quoted_attr)
        $set_attr_name(x, val) = setproperty!(x, $quoted_attr, val)
    end
end

# should generate code like:
#   get_foo(x) = getproperty(x, :foo)

julia> nt = (foo=1, bar=2);

julia> get_foo(nt)
1

julia> get_bar(nt)
2

# Let's test in local scope too
julia> let nt2 = (foo=42,)
           get_foo(nt2)
       end
42