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

窃取使用quasiquotes from this answer的想法，我们首先添加宏天堂插件：

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

用Scala宏拼接符号

1 个答案: