Question

我正在Scala中构建DSL，为此，我需要存储一个类的“实例”（在这种情况下为Parent），除了这些“实例”必须在运行时重新创建几次。。因此，我改为存储“构造函数”-构成实例的lambda。

考虑以下代码-假设userVar可以在运行时更改，并且在构造实例时必须使用更新的值。

class Parent {
  def func(param: Any): Unit = { ... }
}

class User {
  def construct(constr: => Parent): ParentWrapper = { ... }

  var userVar = 13

  construct(new Parent {
    func(1)
    func(userVar)
  }

  construct(new Parent {
    func(userVar)
  }
}

表达我想要的一种更自然的方式是（使用上述定义）：

class User {
  var userVar = 13
  object ObjectA extends Parent {
    func(1)
    func(userVar)
  }

  construct(ObjectA)
}

但是，鉴于ObjectA是立即创建的并且没有“构造函数”，因此这似乎是不可能的。

我正在思考，通过创造性地使用宏，我可以改为：

class User {
  var userVar = 13

  constructMacro {
    func(1)
    func(userVar}
  }
}

，然后让constructMacro将代码转换为construct(new Parent {code block goes here})。

我该怎么做？

还是有更好的方法来避免尴尬的construct(new Parent{...})通话？我的要求是在User类的某个地方存储一个引用，我可以重复调用并获取Parent定义的新实例，这些实例反映了其构造中使用的新值-和{{1} }调用最好返回该引用的包装器对象。

Answer 1

不幸的是，宏将无济于事。

宏注释（在类型检查之前会展开）can't annotate code blocks：

@constructMacro {
  func(1)
  func(userVar)
}

是非法的。

Def宏（在类型检查期间扩展）have their arguments type checked before macros are expanded。所以

constructMacro {
  func(1)
  func(userVar)
}

不编译：

Error: not found: value func
      func(1)
Error: not found: value func
      func(userVar)

这就是宏shapeless.test.illTyped接受字符串而不是代码块的原因：

illTyped("""
  val x: Int = "a"
""")

而不是

illTyped {
  val x: Int = "a"
}

所以您可以实现的最接近的是

constructMacro("""
  func(1)
  func(userVar)
""")

def constructMacro(block: String): ParentWrapper = macro constructMacroImpl

def constructMacroImpl(c: blackbox.Context)(block: c.Tree): c.Tree = {
  import c.universe._
  val q"${blockStr: String}" = block
  val block1 = c.parse(blockStr)
  q"""construct(new Parent {
    ..$block1
  })"""
}

您可以注释变量

@constructMacro val x = {
  func(1)
  func(userVar)
}

//         becomes
// val x: ParentWrapper = construct(new Parent {
//   func(1)
//   func(userVar)
// })

@compileTimeOnly("enable macro paradise to expand macro annotations")
class constructMacro extends StaticAnnotation {
  def macroTransform(annottees: Any*): Any = macro constructMacroImpl.impl
}

object constructMacroImpl {
  def impl(c: whitebox.Context)(annottees: c.Tree*): c.Tree = {
    import c.universe._
    annottees match {
      case q"$mods val $tname: $tpt = { ..$stats }" :: Nil =>
        q"$mods val $tname: ParentWrapper = construct(new Parent { ..$stats })"

      case _ =>
        c.abort(c.enclosingPosition, "Not a val")
    }
  }
}

Answer 2

如果我理解正确，则只需在object块中将def更改为ObjectA：

class User {
  var userVar = 13
  def makeParent = new Parent {
    func(1)
    func(userVar)
  }

  construct(makeParent)
}

它会做你想要的。

Scala：从类主体创建实例的宏

2 个答案: