我试图创建一个scalameta注释,它将在现有对象中定义一个新的case类。
我的目标是改造:
object MyObj {
case class A()
}
成:
object MyObj {
case class A()
case class B(b: Int, bb: Int)
}
为此我创建了以下注释:
import scala.annotation.StaticAnnotation
import scala.meta._
class schema extends StaticAnnotation {
inline def apply(defn: Any): Any = meta {
val q"..$mods object $ename extends $template" = defn
val generatedClass: Defn.Class =
q"""
case class B(b: Int, bb: Int)
"""
//the stats that were defined in the annotated object
val existingStats: scala.collection.immutable.Seq[Stat] = template.stats.get
//new stats
val stats = Some(existingStats :+ generatedClass)
//the new template
val newT = template.copy(stats=stats)
val res =
q"""
..$mods object $ename extends $newT
"""
println("============== result ==============")
println("res: " +res)
println("====================================")
res
}
}
和这个主要类来测试它:
object Main {
def main(args: Array[String]): Unit = {
println("Hello")
val b = new MyObj.B(2,22)
val q = MyObj.A()
println(b)
}
}
@schema
object MyObj {
case class A()
}
当我编译并运行主类时,代码的行为与预期的一样 在编译期间它打印:
============== result ==============
res: object MyObj {
case class A()
case class B(b: Int, bb: Int)
}
====================================
运行代码后:
Hello
B(2,22)
我的问题是当我尝试使用intellij这个代码时
代码从intellij编译并运行,但新类B
无法被识别,因此代码完成和提示不起作用
我正在使用最新的intellij和scala插件
我已将整个项目上传到git: https://github.com/lev112/scalameta_intellij/tree/master
这是我第一次尝试使用scalameta,我试图了解它是否是intellij中的错误,或者我做错了什么
答案 0 :(得分:0)
就像老式宏一样,需要在单独的编译单元中定义scalameta代码,以便首先对其进行编译,然后在编译其他源代码时将其调用。
感谢https://www.michaelpollmeier.com/2016/12/01/scalameta-code-generation-tutorial