scala - 使用 Scala 宏将符号拼接在一起

Question

我正在尝试从通用 Scala 代码调用专门的集合库，例如 FastUtil 或 Trove。我想实现类似的东西

def openHashMap[@specialized K, @specialized V]: ${K}2${V}OpenHashMap = 
   new ${K}2${V}OpenHashMap()

其中 ${X} 显然不是有效的 Scala，而只是我用于文本替换的元符号， 这样 openHashMap[Long, Double] 将返回一个 Long2DoubleOpenHashMap，其类型在编译时就已知。这可以用 Scala 宏实现吗？如果是的话，是什么味道？我知道有 def 宏、隐式宏、fundep 物化、宏注释、类型宏(现已停产)……我认为这些在普通的 Scala-2.10、2.10 宏天堂和 Scala-2.11 中是不同的。其中哪些(如果有的话)适合于此？

或者还有其他东西可以做到这一点，比如字节码操作、语言虚拟化……？但是，我不相信我刚才提到的替代方案可以。

Answer 1

窃取使用准引号的想法 from this answer ，我们首先添加macro-paradise插件:

// build.sbt
scalaVersion := "2.10.2"

resolvers += Resolver.sonatypeRepo("snapshots")

addCompilerPlugin("org.scala-lang.plugins" % "macro-paradise" % "2.0.0-SNAPSHOT" 
  cross CrossVersion.full)

然后宏看起来像这样:

// src/main/scala/Foo.scala
import reflect.macros.Context
import language.experimental.macros

trait Foo[A]
class IntFoo() extends Foo[Int]
class AnyFoo() extends Foo[Any]

object Foo {
  def apply[A]: Foo[A] = macro applyImpl[A]

  def applyImpl[A](c: Context)(t: c.WeakTypeTag[A]): c.Expr[Foo[A]] = {
    import c.universe._
    val aTpe    = t.tpe
    val prefix  = if (aTpe =:= typeOf[Int]) "Int" else "Any"
    val clazz   = newTypeName(s"${prefix}Foo")
    c.Expr(q"new $clazz()")
  }
}

测试用例:

// src/test/scala/Test.scala
object Test extends App {
  val fooInt = Foo[Int]
  val fooAny = Foo[Any]

  println(fooInt)
  println(fooAny)
}

---

如果没有macro-paradise-plugin，您需要手动构建树，例如New(clazz, ???)，我无法让它工作，所以放弃了，但这当然也是可能的。