Scala Macros - scalamela 1.x,inline-meta annotations

2024-04-09 04:58

本文主要是介绍Scala Macros - scalamela 1.x,inline-meta annotations,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

  在上期讨论中我们介绍了Scala Macros,它可以说是工具库编程人员不可或缺的编程手段,可以实现编译器在编译源代码时对源代码进行的修改、扩展和替换,如此可以对用户屏蔽工具库复杂的内部细节,使他们可以用简单的声明方式,通过编译器自动产生铺垫代码来实现工具库中各种复杂的类型、对象及方法函数的构建。虽然Def Macros可能具备超强的编程功能,但同时使用者也普遍认为它一直存有着一些严重的诟病:包括用法复杂、容易犯错、运算行为难以预测以及没用完善的集成开发环境工具(IDE)支持等。这些恶评主要是因为Def Macros和编译器scalac捆绑的太紧,使用者必须对编译器的内部运作原理和操作函数有比较深刻的了解。加之Def Macros向用户提供的api比较复杂且调用繁琐,其中比较致命的问题就是与scalac的紧密捆绑了:因为Def Macros还只是一项实验性功能,没有scala语言规范文件背书,肯定会面临升级换代。而且scala本身也面临着向2.12版本升级的情况,其中dotty就肯定是scalac的替代编译器。Scalameta是根据scala语言规范SIP-28-29-Inline-Macros由零重新设计的Macros编程工具库。主要目的就是为了解决Def Macros所存在的问题,而且Jetbrains的IntelliJ IDEA 2016.3 EAP对Scalameta已经有了比较好的支持,能为使用者带来更简单、安全的Macros编程工具。

  我在介绍了Slick之后立即转入Scala Macros是有一些特别目的的。研究FRM Slick乃至学习泛函编程的初衷就是希望能为传统的OOP编程人员提供更简单易用的泛函库应用帮助,使他们无须对函数式编程模式有太深刻了解也能使用由函数式编程模式所开发的函数库。实现这个目标的主要方式就是Macros了。希望通过Macros的产生代码功能把函数库的泛函特性和模式屏蔽起来,让用户能用他们习惯的方式来定义函数库中的类型对象、调用库中的方法函数。

  Macros功能实现方式(即编译时的源代码扩展compile time expansion)由两个主要部分组成:一是在调用时扩展(on call expansion),二是申明时扩展即注释(annotation)。这两种方式我们在上一篇讨论里都一一做了示范。通过测试发现,Scalameta v1.x只支持注释方式。这事动摇了我继续探讨的意愿:试想如果没了”Implicit Macros“,“Extractor Macros“这些模式,会损失多少理想有趣的编码方式。通过与Scalameta作者沟通后得知他们将会在Scalameta v2.x中开始支持全部两种模式,因此决定先介绍一下Scalameta v1.x,主要目的是让大家先熟悉了解Scalameta新的api和使用模式。我们可以把上次Def Macros的Macros Annotations示范例子在Scalameta里重新示范一遍来达到这样的目的。

  虽然Scalameta是从头设计的,但是它还是保留了许多Def Macros的思想,特别是沿用了大部分scala-reflect的quasiquote模式。与Def Macros运算原理相同,Scalameta的Macros扩展也是基于AST(abstract syntax tree)由编译器运算产生的,因此Macros申明必须先完成编译,所以我们还是沿用了上一篇讨论中的build.sbt,保留项目结构,及demos对macros的这种依赖关系。

name := "learn-scalameta"val commonSettings = Seq(version := "1.0" ,scalaVersion := "2.11.8",scalacOptions ++= Seq("-deprecation", "-feature"),resolvers += Resolver.sonatypeRepo("snapshots"),addCompilerPlugin("org.scalameta" % "paradise" % "3.0.0-M5" cross CrossVersion.full),scalacOptions += "-Xplugin-require:macroparadise")
val macrosSettings = Seq(libraryDependencies += "org.scalameta" %% "scalameta" % "1.3.0",libraryDependencies +=  "org.scalatest" %% "scalatest" % "3.0.1" % "test"
)
lazy val root = (project in file(".")).aggregate(macros, demos)lazy val macros  = project.in(file("macros")).settings(commonSettings : _*).settings(macrosSettings : _*)lazy val demos  = project.in(file("demos")).settings(commonSettings : _*).dependsOn(macros)


下面我们先用一个最简单的例子来开始了解Scalameta Macros Annotations:

object MacroAnnotDemo extends App {@Greetings object Greet {def add(x: Int, y: Int) = println(x + y)}Greet.sayHello("John")Greet.add(1,2)
}


这里的注释@Greetings代表被注释对象Greet将会被扩展增加一个sayHello的函数。我们看看这个注释的实现方式:

import scala.meta._class Greetings extends scala.annotation.StaticAnnotation {inline def apply(defn: Any): Any = meta {defn match {case q"object $name {..$stats}" => {q"""object $name {def sayHello(msg: String): Unit = println("Hello," + msg)..$stats}"""}case _ => abort("annottee must be object!")}}
}


首先,我们看到这段源代码表达方式直接了许多:只需要import scala.meta,没有了blackbox、whitebox、universe这些imports。特别是避免了对blackbox.Context和whitebox.Context这些复杂运算域的人为判定。quasiquote的使用没有什么变化。直观上Macros编程简单了,实际上编写的Macros程序能更安全稳定的运行。

我们再重复演示方法注释(method annotation)的实现方法:

class Benchmark extends scala.annotation.StaticAnnotation {inline def apply(defn: Any): Any = meta {defn match {case q"..$mod def $name[..$tparams](...$args): $rtpe = $body" =>q"""..$mod def $name[..$tparams](...$args): $rtpe = {val start = System.nanoTime()val result = $bodyval end = System.nanoTime()println(${name.toString} + " elapsed time = " + (end - start) + "ns")result}"""case _ => abort("Fail to expand annotation Benchmark!")}}
}


还是固定格式。只是quasiquote的调用组合变化。用下面方法调用测试:

  @Benchmarkdef calcPow(x: Double, y: Double) = {val z = x + ymath.pow(z,z)}println(calcPow(4.2, 8.9))


在下面这个例子里我们在注释对象中增加main方法(未extends App的对象):

import scala.meta.Ctor.Call
class main extends scala.annotation.StaticAnnotation {inline def apply(defn: Any): Any = meta {def abortIfObjectAlreadyExtendsApp(ctorcalls: scala.collection.immutable.Seq[Call], objectName: Term) = {val extendsAppAlready = ctorcalls.map(_.structure).contains(ctor"App()".structure)if (extendsAppAlready){abort(s"$objectName already extends App")}}defn match {case q"..$mods object $name extends $template" => template match {case template"{ ..$stats1 } with ..$ctorcalls { $param => ..$stats2 }" =>abortIfObjectAlreadyExtendsApp(ctorcalls, name)val mainMethod = q"def main(args: Array[String]): Unit = { ..$stats2 }"val newTemplate = template"{ ..$stats1 } with ..$ctorcalls { $param => $mainMethod }"q"..$mods object $name extends $newTemplate"}case _ => abort("@main can be annotation of object only")}}
}


下面这个是case class的注释示例:效果是添加一个从case class转Map的类型转换函数toMap:

@compileTimeOnly("@Mappable not expanded")
class Mappable extends StaticAnnotation {inline def apply(defn: Any): Any = meta {defn match {case q"..$mods class $tname[..$tparams] (...$paramss) extends $template" =>template match {case template"{ ..$stats } with ..$ctorcalls { $param => ..$body }" => {val expr = paramss.flatten.map(p => q"${p.name.toString}").zip(paramss.flatten.map{case param"..$mods $paramname: $atpeopt = $expropt" => paramname}).map{case (q"$paramName", paramTree) => {q"${Term.Name(paramName.toString)} -> ${Term.Name(paramTree.toString)}"}}val resultMap = q"Map(..$expr)"val newBody = body :+ q"""def toMap: Map[String, Any] = $resultMap"""val newTemplate = template"{ ..$stats } with ..$ctorcalls { $param => ..$newBody }"q"..$mods class $tname[..$tparams] (...$paramss) extends $newTemplate"}}case _ => throw new Exception("@Mappable can be annotation of class only")}}
}


可以用下面的数据进行测试:

  @Mappablecase class Car(color: String, model: String, year: Int, owner: String){def turnOnRadio = {"playing"}}val newCarMap = Car("Silver", "Ford", 1998, "John Doe").toMapprintln(newCarMap)


在下面这个例子里示范了如何使用注释参数:

import scala.util.Try
@compileTimeOnly("@RetryOnFailure not expanded")
class RetryOnFailure(repeat: Int) extends scala.annotation.StaticAnnotation {inline def apply(defn: Any): Any = meta {defn match {case q"..$mods def $name[..$tparams](...$paramss): $tpeopt = $expr" => {val q"new $_(${arg})" = thisval repeats = Try(arg.toString.toInt).getOrElse(abort(s"Retry on failure takes number as parameter"))val newCode =q"""..$mods def $name[..$tparams](...$paramss): $tpeopt = {import scala.util.Tryfor( a <- 1 to $repeats){val res = Try($expr)if(res.isSuccess){return res.get}}throw new Exception("Method fails after "+$repeats + " repeats")}"""newCode}case _ => abort("@RetryOnFailure can be annotation of method only")}}
}

具体使用方法如下:

 object utils {def methodThrowingException(random: Int): Unit = {if(random%2 == 0){throw new Exception(s"throwing exception for ${random}")}}}import scala.util.Random@RetryOnFailure(20) def failMethod[String](): Unit = {val random = Random.nextInt(10)println("Retrying...")utils.methodThrowingException(random)}

顺便也把上次的那个TalkingAnimal重新再写一下:

class TalkingAnimal(voice: String) extends StaticAnnotation {inline def apply(defn: Any): Any = meta {defn match {case q"..$mods class $tname[..$tparams] (...$paramss) extends $template" =>template match {case template"{ ..$stats } with ..$ctorcalls { $param => ..$body }" => {val q"new $_(${arg})" = thisval sound = arg.toString()val animalType = tname.toString()val newBody = body :+q""" def sayHello: Unit =println("Hello, I'm a " + $animalType +" and my name is " + name + " " + $sound+ "...")"""val newTemplate =template"{ ..$stats } with ..$ctorcalls { $param => ..$newBody }"q"..$mods class $tname[..$tparams] (...$paramss) extends $newTemplate"}}case _ => abort("Error: expanding TalkingAnimal!")}}
}


对比旧款Def Macros可以发现quasiquote的语法还是有变化的,比如拆分class定义就需要先拆出template。Scalameta重新定义了新的quasiquote,另外注释对象参数的运算方法也有所不同,这是因为Scalameta的AST新设计的表达结构。

测试运算如下:

  trait Animal {val name: String}@TalkingAnimal("wangwang")case class Dog(val name: String) extends Animal@TalkingAnimal("miaomiao")case class Cat(val name: String) extends Animal//@TalkingAnimal("")//case class Carrot(val name: String)//Error:(12,2) Annotation TalkingAnimal only apply to Animal inherited! @TalingAnimalDog("Goldy").sayHelloCat("Kitty").sayHello

下面是本次讨论中的完整示范源代码:

注释实现源代码:

import scala.meta._
class Greetings extends scala.annotation.StaticAnnotation {inline def apply(defn: Any): Any = meta {defn match {case q"object $name {..$stats}" => {q"""object $name {def sayHello(msg: String): Unit = println("Hello," + msg)..$stats}"""}case q"object $name extends $parent {..$stats}" => {q"""object $name extends $parent {def sayHello(msg: String): Unit = println("Hello," + msg)..$stats}"""}case _ => abort("annottee must be object!")}}
}class Benchmark extends scala.annotation.StaticAnnotation {inline def apply(defn: Any): Any = meta {defn match {case q"..$mod def $name[..$tparams](...$args): $rtpe = $body" =>q"""..$mod def $name[..$tparams](...$args): $rtpe = {val start = System.nanoTime()val result = $bodyval end = System.nanoTime()println(${name.toString} + " elapsed time = " + (end - start) + "ns")result}"""case _ => abort("Fail to expand annotation Benchmark!")}}
}import scala.meta.Ctor.Call
class main extends scala.annotation.StaticAnnotation {inline def apply(defn: Any): Any = meta {def abortIfObjectAlreadyExtendsApp(ctorcalls: scala.collection.immutable.Seq[Call], objectName: Term) = {val extendsAppAlready = ctorcalls.map(_.structure).contains(ctor"App()".structure)if (extendsAppAlready){abort(s"$objectName already extends App")}}defn match {case q"..$mods object $name extends $template" => template match {case template"{ ..$stats1 } with ..$ctorcalls { $param => ..$stats2 }" =>abortIfObjectAlreadyExtendsApp(ctorcalls, name)val mainMethod = q"def main(args: Array[String]): Unit = { ..$stats2 }"val newTemplate = template"{ ..$stats1 } with ..$ctorcalls { $param => $mainMethod }"q"..$mods object $name extends $newTemplate"}case _ => abort("@main can be annotation of object only")}}
}
import scala.annotation.{StaticAnnotation, compileTimeOnly}
@compileTimeOnly("@Mappable not expanded")
class Mappable extends StaticAnnotation {inline def apply(defn: Any): Any = meta {defn match {case q"..$mods class $tname[..$tparams] (...$paramss) extends $template" =>template match {case template"{ ..$stats } with ..$ctorcalls { $param => ..$body }" => {val expr = paramss.flatten.map(p => q"${p.name.toString}").zip(paramss.flatten.map{case param"..$mods $paramname: $atpeopt = $expropt" => paramname}).map{case (q"$paramName", paramTree) => {q"${Term.Name(paramName.toString)} -> ${Term.Name(paramTree.toString)}"}}val resultMap = q"Map(..$expr)"val newBody = body :+ q"""def toMap: Map[String, Any] = $resultMap"""val newTemplate = template"{ ..$stats } with ..$ctorcalls { $param => ..$newBody }"q"..$mods class $tname[..$tparams] (...$paramss) extends $newTemplate"}}case _ => throw new Exception("@Mappable can be annotation of class only")}}
}
import scala.util.Try
@compileTimeOnly("@RetryOnFailure not expanded")
class RetryOnFailure(repeat: Int) extends scala.annotation.StaticAnnotation {inline def apply(defn: Any): Any = meta {defn match {case q"..$mods def $name[..$tparams](...$paramss): $tpeopt = $expr" => {val q"new $_(${arg})" = thisval repeats = Try(arg.toString.toInt).getOrElse(abort(s"Retry on failure takes number as parameter"))val newCode =q"""..$mods def $name[..$tparams](...$paramss): $tpeopt = {import scala.util.Tryfor( a <- 1 to $repeats){val res = Try($expr)if(res.isSuccess){return res.get}}throw new Exception("Method fails after "+$repeats + " repeats")}"""newCode}case _ => abort("@RetryOnFailure can be annotation of method only")}}
}class TalkingAnimal(voice: String) extends StaticAnnotation {inline def apply(defn: Any): Any = meta {defn match {case q"..$mods class $tname[..$tparams] (...$paramss) extends $template" =>template match {case template"{ ..$stats } with ..$ctorcalls { $param => ..$body }" => {val q"new $_(${arg})" = thisval sound = arg.toString()val animalType = tname.toString()val newBody = body :+q""" def sayHello: Unit =println("Hello, I'm a " + $animalType +" and my name is " + name + " " + $sound+ "...")"""val newTemplate =template"{ ..$stats } with ..$ctorcalls { $param => ..$newBody }"q"..$mods class $tname[..$tparams] (...$paramss) extends $newTemplate"}}case _ => abort("Error: expanding TalkingAnimal!")}}
}


试运行代码:

object MacroAnnotDemo extends App {@Greetings object Greet {def add(x: Int, y: Int) = println(x + y)}@Greetings object Hi extends AnyRef {}Greet.sayHello("John")Greet.add(1,2)Hi.sayHello("Susana Wang")@Benchmarkdef calcPow(x: Double, y: Double) = {val z = x + ymath.pow(z,z)}println(calcPow(4.2, 8.9))@Mappablecase class Car(color: String, model: String, year: Int, owner: String){def turnOnRadio = {"playing"}}val newCarMap = Car("Silver", "Ford", 1998, "John Doe").toMapprintln(newCarMap)object utils {def methodThrowingException(random: Int): Unit = {if(random%2 == 0){throw new Exception(s"throwing exception for ${random}")}}}import scala.util.Random@RetryOnFailure(20) def failMethod[String](): Unit = {val random = Random.nextInt(10)println("Retrying...")utils.methodThrowingException(random)}trait Animal {val name: String}@TalkingAnimal("wangwang")case class Dog(val name: String) extends Animal@TalkingAnimal("miaomiao")case class Cat(val name: String) extends Animal//@TalkingAnimal("")//case class Carrot(val name: String)//Error:(12,2) Annotation TalkingAnimal only apply to Animal inherited! @TalingAnimalDog("Goldy").sayHelloCat("Kitty").sayHello}
















这篇关于Scala Macros - scalamela 1.x,inline-meta annotations的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/887227

相关文章

【scala 安装错误】错误: 找不到或无法加载主类 scala.tools.nsc.MainGenericRunner

错误: 找不到或无法加载主类 scala.tools.nsc.MainGenericRunner 原因: Scala安装路径中包含空格。 解决办法:scala 不要安装在E:\Program Files 这种有空格的目录下,简直坑

Vue3+vite中使用import.meta.glob

前言:         在vue2中支持require导入模块或文件但是在vue3中已经不支持require导入了,为此vite提供了一个全新的方法import.meta.glob方法来支持批量导入文件  import.meta.glob 匹配到的文件默认是懒加载的,通过动态导入实现,并会在构建时分离为独立的 chunk。如果你倾向于直接引入所有的模块(例如依赖于这些模块中的副作用首先被应用

Scala模式匹配下提取器构造

示例代码: object :> {def unapply[A] (list:List[A]) = {Some( (list.init,list.last) )}}object Extractor_Advanced {def main(args: Array[String]): Unit = {(1 to 9).toList match{ case _ :> 9 => println(

从spark源码的角度思考scala中的模式匹配

1.scala中模式匹配 2.spark源码中的模式匹配思考 spark中master会收到worker发过来的akka的消息, 此消息是case class即(Master.class中): case class RegisterWorker(id:String,host:String,port:Int,cores:Int,memory:Int,webUiPort:int

Scala界面事件处理

示例代码: import scala.swing.SimpleSwingApplicationimport scala.swing.MainFrameimport scala.swing.Buttonimport scala.swing.Labelimport scala.swing.Orientationimport scala.swing.BoxPanelimpo

Scala界面Panel、Layout初探

示例代码: package com.dt.scala.guiimport scala.swing.SimpleSwingApplicationimport scala.swing.MainFrameimport scala.swing.Buttonimport scala.swing.Labelimport scala.swing.Orientationimport scal

scala界面GUI编程实战初步了解

示例代码: import scala.swing._//SimpleSwingApplication继承自SwingApplication类(此类中有main方法,因此可以运行显示界面)object Hello_GUI extends SimpleSwingApplication {def top = new MainFrame{ //顶级容器title = "Hello GUI"co

Scala并发编程react、loop代码实战详解

示例代码及注释: //scala并发编程中的react和loop,共同特点://通过线程存用的方式让性能有所提升。//Actor本身的运行,被actor子系统管理的时候,会有一个或者多个远程的线程让当前的actor使用//一般情况下每个Actor都有自己的线程。只有有自己的线程时,我们的Actor中的actor方法才会执行。//但是,这样线程的开销会非常大,所以为了共用线

scala并发编程原生线程Actor、Case Class下的消息传递和偏函数实战

参考代码: import scala.actors._case class Person(name:String,age:Int)class HelloActor extends Actor{def act(){while(true){receive{case Person(name,age)=>{ //偏函数println("Name: "+ name + ":" +"Age:"

LEAN 类型理论之注解(Annotations of LEAN Type Theory)—— 定义上相等(Definitional Equality)

定义上相等(Definitional Equality)指的是意义上相等,即同义,包括了,定义的缩写(Abbreviatory Definition),alpha转换,相同替代(substituting equals for equals)等。下面是LEAN给定的何谓 定义上相等。          注:罗列的推演规则中,如自明其义的,则不多加解析其前提、结果、或特定注解。