Question

我对scala宏有一些问题并且确定了构造函数的实现类型。不确定我在这里做错了什么，或者正确的电话会是什么。从文档中看，typeSignatureIn看起来应该返回正确的信息，例如ClassTag [Int]，但是当我运行宏时，实际上我得到的ClassTag [U]无法编译，因为U是类型参数，而不是实现的类型。

import scala.language.experimental.macros
import scala.reflect.ClassTag
import scala.reflect.macros.Context

def macroImpl[T: c.WeakTypeTag](c: Context) = {
  import c.universe._

  val typeToMock = weakTypeOf[T]

  val primaryConstructorOpt = typeToMock.members.collectFirst {
    case method: MethodSymbolApi if method.isPrimaryConstructor => method
  }

  val constructorArgumentsTypes = primaryConstructorOpt.map { 
    constructor =>
    val constructorTypeContext = constructor.typeSignatureIn(typeToMock)
    val constructorArguments = constructor.paramss
    constructorArguments.map { symbols =>
      symbols.map(_.typeSignatureIn(constructorTypeContext))
    }
  }

  println(typeToMock)
  println(constructorArgumentsTypes)

  c.literalUnit
}

def foo[T] = macro macroImpl[T]

class Foo[U: ClassTag]

foo[Foo[Int]]

运行它：

scala> foo[Foo[Int]]
Foo[Int]
Some(List(List(), List(scala.reflect.ClassTag[U]))

我需要以某种方式让ClassTag [Int]能够在以后生成正确的树，任何想法？

Answer 1

尝试在解决类型的每个地方使用dealias。 Scala保留了引用，即使它知道它们引用的内容。 dealias会为您提供已替换引用的类型的副本（？）。

您还应该选择blackbox或whitebox宏，而不仅仅是scala.reflect.macros.Context。

这似乎有效：

import scala.language.experimental.macros
import scala.reflect.ClassTag
import scala.reflect.macros.whitebox.Context

def macroImpl[T: c.WeakTypeTag](c: Context) = {
  import c.universe._

  val typeToMock = weakTypeOf[T].dealias

  val primaryConstructorOpt = typeToMock.members.collectFirst {
    case method: MethodSymbolApi if method.isPrimaryConstructor => method
  }

  val constructorArgumentsTypes = primaryConstructorOpt.map { constructor =>
    val constructorTypeContext = constructor.typeSignatureIn(typeToMock).dealias
    val constructorArguments = constructorTypeContext.paramLists
    constructorArguments.map { symbols =>
      symbols.map(_.typeSignatureIn(constructorTypeContext).dealias)
    }
  }

  println(typeToMock)
  println(constructorArgumentsTypes)

  q"()"
}

def foo[T]: Any = macro macroImpl[T]

class Foo[U: ClassTag]

foo[Foo[Int]]

结果

Foo[Int]
Some(List(List(), List(scala.reflect.ClassTag[Int])))

使用宏获取Scala类的实现类型

1 个答案: