2

我试图从通用 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 中是不同的。如果有的话，哪一个适合这个？

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

4

1 回答 1

2

从这个答案中窃取使用准引号的想法，我们首先添加宏天堂插件：

// 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, ???)，我无法让它工作，所以放弃了，但它当然也是可能的。

于 2013-09-29T11:52:10.733 回答