Scala 宏注解 - 带类型参数的案例类答案

【问题标题】：Scala macro annotation - case class with type parametersScala 宏注解 - 带类型参数的案例类
【发布时间】：2016-07-05 21:10:27
【问题描述】：

我正在尝试为案例类编写一个简单的宏注释，它向伴随对象添加一个方法。问题是新方法必须考虑带注释的案例类的类型参数。

这是我需要通过的测试

package my.macros

import org.scalatest._

class DefaultApplyTest extends FlatSpec with Matchers {

  @defaultApply case class Generic[A, B](a: A, b: B)

  it should "define defaultApply method in companion object" in {
    assert(Generic.defaultApply("str", 1) == Generic("str", 1))
  }
}

这是我为此编写的代码

package my.macros

import scala.reflect.macros._
import scala.language.experimental.macros
import scala.annotation.StaticAnnotation

class defaultApply extends StaticAnnotation {
  def macroTransform(annottees: Any*): Any = macro DefaultApply.impl
}

object DefaultApply {

  def impl(c: blackbox.Context)(annottees: c.Expr[Any]*): c.Expr[Any] = {
    import c.universe._

    def defaultApplyCompanion(classDecl: ClassDef) = {
      val (name, typeParams, valueParams) = try {
        val q"case class ${name: TypeName}[..${typeParams: Seq[TypeDef]}](..${valueParams: Seq[ValDef]}) extends ..$bases { ..$body }" = classDecl
        (name, typeParams, valueParams)
      } catch {
        case e: MatchError =>
          c.warning(c.enclosingPosition, e.toString)
          c.abort(c.enclosingPosition, "Annotation is only supported on case class")
      }

      val applyDef = q"""${name.toTermName}.apply[..$typeParams]"""
      c.warning(c.enclosingPosition, showRaw(applyDef))

      q"""
        object ${name.toTermName} {
          def defaultApply: (..${valueParams.map(_.tpt)}) => $name[..$typeParams] = $applyDef
        }
      """
    }

    def modifiedDeclaration(classDecl: ClassDef) = {
      val compDecl = defaultApplyCompanion(classDecl)

      c.Expr(q"""
        $classDecl
        $compDecl
      """)
    }

    annottees.map(_.tree) match {
      case (classDecl: ClassDef) :: Nil => modifiedDeclaration(classDecl)
      case _ => c.abort(c.enclosingPosition, "Invalid annottee")
    }
  }
}

据我了解，问题是当我尝试将类型参数列表lift 放入生成的语法树中时，它们不会被识别为与原始树中相同的类型参数。

所以我关注的是宏的这一部分

  val applyDef = q"""${name.toTermName}.apply[..$typeParams]"""
  c.warning(c.enclosingPosition, showRaw(applyDef))

原始语法树发出为

TypeApply(Select(Ident(TermName("Generic")), TermName("apply")), List(TypeDef(Modifiers(PARAM), TypeName("A"), List(), TypeBoundsTree(EmptyTree, EmptyTree)), TypeDef(Modifiers(PARAM), TypeName("B"), List(), TypeBoundsTree(EmptyTree, EmptyTree))))

但编译器对此并不满意

type arguments [<notype>,<notype>] do not conform to method apply's type parameter bounds [A,B]

最终用例是生成涉及超过 1k 行代码的可缓存类型类的实例。非参数化版本已经可以工作了，这只是锦上添花。 scalac 的底层有些东西我不明白，但我想。非常感谢您花时间阅读本文。

我正在使用带有 macro paradise 2.1.0 的 Scala 2.11.8

【问题讨论】：

标签： scala macros scala-2.11 scala-quasiquotes scala-macro-paradise

【解决方案1】：

问题似乎是您使用类型参数代替类型参数。这似乎可行（我还必须将类型参数添加到defaultApply 方法声明）：

  val typeArgs = typeParams.map(_.name)
  val applyDef = q"""${name.toTermName}.apply[..$typeArgs]"""
  c.warning(c.enclosingPosition, showRaw(applyDef))

  q"""
    object ${name.toTermName} {
      def defaultApply[..$typeParams]:
          (..${valueParams.map(_.tpt)}) => $name[..$typeArgs] = $applyDef
    }
  """

【讨论】：