Question

考虑一个基于Scala宏的注释，例如@memoise中的macmemo。批注需要两个参数：最大缓存大小和生存时间，例如

@memoize(maxSize = 20000, expiresAfter = 2 hours)

假设您要创建一个与@cacheall等效的@memoize(maxSize = Int.MaxValue, expiresAfter = 100 days)注释，以减少样板并具有单个参数化点。

这种重用是否有标准模式？显然，

class cacheall extends memoize(Int.MaxValue, 100 days)

由于宏中的编译时参数解析而无法使用。

Answer 1

标准模式是使您的注释成为宏注释，将其展开后会使用必要的参数打开必要的注释。

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

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

object cacheallMacro {
  def impl(c: blackbox.Context)(annottees: c.Tree*): c.Tree = {
    import c.universe._

    val memoize = q"""
      new _root_.com.softwaremill.macmemo.memoize(
        _root_.scala.Int.MaxValue, {
          import _root_.scala.concurrent.duration._
          100.days 
      })"""

    annottees match {
      case q"${mods: Modifiers} def $tname[..$tparams](...$paramss): $tpt = $expr" :: _ =>
        q"${mods.mapAnnotations(memoize :: _)} def $tname[..$tparams](...$paramss): $tpt = $expr"
    }
  }
}

基于Scala宏的注释重用

1 个答案: