文章/答案/技术大牛

发布

问Scala 3宏特征参数
EN

Stack Overflow用户

提问于 2021-07-16 16:17:00

回答 1查看 222关注 0票数 2

以下是基本设置：

trait MyProduct[A, B](a: A, b:  B)
class X
class Y

class XxY(x: X, y: Y) extends MyProduct(x,y)

我试图检查MyProduct特性的参数。更具体地说，我想提取一些信息，例如字符串"x“和"y"，这些信息指示XxY的哪些字段被传递给MyProduct。

重要的是，我需要满足具有相同类型的多个字段(如class XxYxY(x: X, y1: Y, y2: Y) extends MyProduct3(x, y1, y2) )的情况，因此从类型参数进行推理是不够的。

我想问题可能是，我还没有找到一种方法来获取extends子句本身的符号。我可以找到ClassDef for XxY，从中提取parents并得到已经构造的MyProduct[X,Y]类型。我还查看了封闭模块的symbol.declarations和XxY.<init>函数，以确定是否有数据可用于查找参数，但这些数据似乎都没有我要查找的信息。

查看封装模块的树会让我想到，这些信息可能被删除了，需要从源代码中解析为文本，但我希望有人有一个更好的解决方案。

编辑自评论：

作为输入，我有一个封闭对象/模块的Type实例。例如：

inline def simpleProductTest[T <: Singleton]: String = ${simpleProductTestImpl[T]}

def simpleProductTestImpl[T](using q: Quotes)(using Type[T]): Expr[String] = ???

作为输出，我想要一些东西，允许我检查扩展子句中使用的特征的参数。

scala-macros

scala-3

scala

回答 1

Stack Overflow用户

回答已采纳

发布于 2021-07-16 23:33:10

如果你这样做了

import scala.quoted.*

object Macros {
  inline def simpleProductTest[T]: Unit = ${simpleProductTestImpl[T]}

  def simpleProductTestImpl[T](using Quotes)(using Type[T]): Expr[Unit] = {
    import quotes.reflect.*
    println(TypeRepr.of[T].typeSymbol.tree.asInstanceOf[TypeDef].rhs)
    '{()}
  }
}

你终究会明白的

object App {
  trait MyProduct[A, B](a: A, b:  B)
  class X
  class Y

  class XxY(x: X, y: Y, z: Int) extends MyProduct(x,y)

  Macros.simpleProductTest[XxY]
}

没有关于x,y的信息

Template(
  DefDef(
    <init>,
    List(List(
      ValDef(x,TypeTree[TypeRef(ThisType(TypeRef(ThisType(TypeRef(NoPrefix,module class <empty>)),module class App$)),class X)],EmptyTree),
      ValDef(y,TypeTree[TypeRef(ThisType(TypeRef(ThisType(TypeRef(NoPrefix,module class <empty>)),module class App$)),class Y)],EmptyTree),
      ValDef(z,TypeTree[TypeRef(TermRef(ThisType(TypeRef(NoPrefix,module class <root>)),object scala),class Int)],EmptyTree)
    )),
    TypeTree[TypeRef(ThisType(TypeRef(ThisType(TypeRef(NoPrefix,module class <empty>)),module class App$)),class XxY)],
    EmptyTree
  ),
  List(
    TypeTree[TypeRef(ThisType(TypeRef(NoPrefix,module class lang)),class Object)],
    TypeTree[AppliedType(TypeRef(ThisType(TypeRef(ThisType(TypeRef(NoPrefix,module class <empty>)),module class App$)),trait MyProduct),
      List(
        TypeRef(ThisType(TypeRef(ThisType(TypeRef(NoPrefix,module class <empty>)),module class App$)),class X),
        TypeRef(ThisType(TypeRef(ThisType(TypeRef(NoPrefix,module class <empty>)),module class App$)),class Y)
      )
    )]
  ),
  ValDef(_,EmptyTree,EmptyTree),
  List(
    ValDef(x,TypeTree[TypeRef(ThisType(TypeRef(ThisType(TypeRef(NoPrefix,module class <empty>)),module class App$)),class X)],EmptyTree),
    ValDef(y,TypeTree[TypeRef(ThisType(TypeRef(ThisType(TypeRef(NoPrefix,module class <empty>)),module class App$)),class Y)],EmptyTree),
    ValDef(z,TypeTree[TypeRef(TermRef(ThisType(TypeRef(NoPrefix,module class <root>)),object scala),class Int)],EmptyTree)
  )
)

Template不存在于scala.quoted.*中，它是dotty.tools.dotc.ast.Trees.Template。

但是如果用.tasty打印TASTy检查文件

libraryDependencies += "org.scala-lang" %% "scala3-tasty-inspector" % "3.0.0"

import scala.quoted.*
import scala.tasty.inspector.*

class MyInspector extends Inspector:
  def inspect(using Quotes)(tastys: List[Tasty[quotes.type]]): Unit =
    import quotes.reflect.*
    for tasty <- tastys do
      val tree = tasty.ast
      println(tree)

@main def test: Unit =
  val tastyFiles = List("target/scala-3.0.0/classes/App.tasty")
  TastyInspector.inspectTastyFiles(tastyFiles)(new MyInspector)

你终究会明白的

Template(
  DefDef(
    <init>,
    List(List(
      ValDef(x,Ident(X),EmptyTree),
      ValDef(y,Ident(Y),EmptyTree),
      ValDef(z,Ident(Int),EmptyTree)
    )),
    TypeTree[TypeRef(TermRef(ThisType(TypeRef(NoPrefix,module class <root>)),object scala),Unit)],
    EmptyTree
  ),
  List(
    Apply(Select(New(TypeTree[TypeRef(TermRef(ThisType(TypeRef(NoPrefix,module class java)),object lang),Object)]),<init>),List()),
    Apply(
      TypeApply(
        Select(New(Ident(MyProduct)),<init>),
        List(
          TypeTree[TypeRef(ThisType(TypeRef(ThisType(TypeRef(ThisType(TypeRef(NoPrefix,module class <root>)),module class <empty>)),module class App$)),class X)],
          TypeTree[TypeRef(ThisType(TypeRef(ThisType(TypeRef(ThisType(TypeRef(NoPrefix,module class <root>)),module class <empty>)),module class App$)),class Y)]
        )
      ),
      List(Ident(x), Ident(y))  <--- HERE!!!
    )
  ),
  ValDef(_,EmptyTree,EmptyTree),
  List(
    ValDef(x,TypeTree[TypeRef(ThisType(TypeRef(ThisType(TypeRef(ThisType(TypeRef(NoPrefix,module class <root>)),module class <empty>)),module class App$)),class X)],EmptyTree),
    ValDef(y,TypeTree[TypeRef(ThisType(TypeRef(ThisType(TypeRef(ThisType(TypeRef(NoPrefix,module class <root>)),module class <empty>)),module class App$)),class Y)],EmptyTree),
    ValDef(z,TypeTree[TypeRef(TermRef(ThisType(TypeRef(NoPrefix,module class <root>)),object scala),Int)],EmptyTree)
  )
)

这棵树略有不同，而且有关于x,y的信息。

在.tree of Symbol中编写了

     *  **Warning**: avoid using this method in macros.
     *
     *  **Caveat**: The tree is not guaranteed to exist unless the compiler
     *  option `-Yretain-trees` is enabled.

让我们打开-Yretain-trees

scalacOptions += "-Yretain-trees"

然后Macros.simpleProductTest[XxY]打印

Template(
  DefDef(
    <init>,
    List(List(
      ValDef(x,Ident(X),EmptyTree),
      ValDef(y,Ident(Y),EmptyTree),
      ValDef(z,Ident(Int),EmptyTree)
    )),
    TypeTree[TypeRef(ThisType(TypeRef(NoPrefix,module class scala)),class Unit)],
    EmptyTree
  ),
  List(
    Apply(Select(New(TypeTree[TypeRef(ThisType(TypeRef(NoPrefix,module class lang)),class Object)]),<init>),List()),
    Apply(
      TypeApply(
        Select(New(Ident(MyProduct)),<init>),
        List(
          TypeTree[TypeVar(TypeParamRef(A) -> TypeRef(ThisType(TypeRef(ThisType(TypeRef(NoPrefix,module class <empty>)),module class App$)),class X))],
          TypeTree[TypeVar(TypeParamRef(B) -> TypeRef(ThisType(TypeRef(ThisType(TypeRef(NoPrefix,module class <empty>)),module class App$)),class Y))]
        )
      ),
      List(Ident(x), Ident(y))   <----  AGAIN HERE !!!!
    )
  ),
  ValDef(_,EmptyTree,EmptyTree),
  List(
    ValDef(x,TypeTree[TypeRef(ThisType(TypeRef(ThisType(TypeRef(NoPrefix,module class <empty>)),module class App$)),class X)],EmptyTree),
    ValDef(y,TypeTree[TypeRef(ThisType(TypeRef(ThisType(TypeRef(NoPrefix,module class <empty>)),module class App$)),class Y)],EmptyTree),
    ValDef(z,TypeTree[TypeRef(TermRef(ThisType(TypeRef(NoPrefix,module class <root>)),object scala),class Int)],EmptyTree)
  )
)

也就是说，树也略有不同，但是出现了关于x,y的信息。所以现在的问题只是如何提取这些信息，因为它是dotty.tools.dotc.ast.*

case class Template[-T >: Untyped] private[ast](
  constr: DefDef[T], 
  parentsOrDerived: List[Tree[T]], 
  self: ValDef[T], 
  private var preBody: LazyTreeList[T @uncheckedVariance]
)(implicit @constructorOnly src: SourceFile)

所以最后

libraryDependencies += "org.scala-lang" %% "scala3-compiler" % "3.0.0"

import scala.quoted.*

inline def simpleProductTest[T]: List[String] = ${simpleProductTestImpl[T]}

def simpleProductTestImpl[T](using Quotes)(using Type[T]): Expr[List[String]] = {
  import quotes.reflect.*
  Expr(
    TypeRepr.of[T].typeSymbol.tree.asInstanceOf[TypeDef].rhs
      .asInstanceOf[dotty.tools.dotc.ast.Trees.Template[?]]
      .parentsOrDerived
      .asInstanceOf[List[Apply]]
      .tail.head.args.map(_.symbol.name)
  )
}

Macros.simpleProductTest[XxY] // List(x, y)

票数 5

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/68412253

复制

相似问题

问Scala 3宏特征参数
EN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问Scala 3宏特征参数EN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问Scala 3宏特征参数
EN