Question

我正在努力实现一个StringContext扩展，这将允许我写这个：

val tz = zone"Europe/London" //tz is of type java.util.TimeZone

但是，如果提供的时区无效（假设可以在编译时确定），那么它应该无法编译。

这是辅助函数：

def maybeTZ(s: String): Option[java.util.TimeZone] = java.util.TimeZone.getAvailableIDs collectFirst { case id if id == s => java.util.TimeZone.getTimeZone(id) }

我可以非常轻松地创建非宏实现：

scala> implicit class TZContext(val sc: StringContext) extends AnyVal { | def zone(args: Any *) = { | val s = sc.raw(args.toSeq : _ *) | maybeTZ(s) getOrElse sys.error(s"Invalid zone: $s") | } | }

然后：

scala> zone"UTC" res1: java.util.TimeZone = sun.util.calendar.ZoneInfo[id="UTC",offset=0,...

到目前为止，这么好。如果时区是无意义的（例如zone"foobar"），则编译不会失败;代码在运行时失效。我想将它扩展到一个宏，但是，尽管阅读了docs，我真的很挣扎于细节（所有细节，确切地说。）

任何人都可以帮我开始吗？全唱，全舞的解决方案应该查看StringContext是否定义了任何参数和（如果是），将计算推迟到运行时，否则尝试在编译时解析区域

我尝试了什么？

嗯，宏defs似乎必须在静态可访问的对象中。所以：

package object oxbow { implicit class TZContext(val sc: StringContext) extends AnyVal { def zone(args: Any *) = macro zoneImpl //zoneImpl cannot be in TZContext } def zoneImpl(c: reflect.macros.Context) (args: c.Expr[Any] *): c.Expr[java.util.TimeZone] = { import c.universe._ //1. How can I access sc from here? /// ... if I could, would this be right? if (args.isEmpty) { val s = sc.raw() reify(maybeTZ(s) getOrElse sys.error(s"Not valid $s")) } else { //Ok, now I'm stuck. What goes here? } } }

根据som-snytt的建议，这是最新的尝试：

def zoneImpl(c: reflect.macros.Context) (args: c.Expr[Any] *): c.Expr[java.util.TimeZone] = { import c.universe._ val z = c.prefix.tree match { case Apply(_, List(Apply(_, List(Literal(Constant(const: String)))))) => gsa.shared.datetime.XTimeZone.getTimeZone(const) case x => ??? //not sure what to put here } c.Expr[java.util.TimeZone](Literal(Constant(z))) //this compiles but doesn't work at the use-site ^^^^^^^^^^^^^^^^^^^ this is wrong. What should it be? }

在use-site，有效的zone"UTC"无法使用错误进行编译：

java.lang.Error: bad constant value: sun.util.calendar.ZoneInfo[id="UTC",offset=0,dstSavings=0,useDaylight=false,transitions=0,lastRule=null] of class class sun.util.calendar.ZoneInfo

据推测，我不应该使用Literal(Constant( .. ))来封闭它。我该怎么用？

最后一个例子 - 基于Travis Brown的答案

def zoneImpl(c: reflect.macros.Context) (args: c.Expr[Any] *): c.Expr[java.util.TimeZone] = { import c.universe._ import java.util.TimeZone val tzExpr: c.Expr[String] = c.prefix.tree match { case Apply(_, Apply(_, List(tz @ Literal(Constant(s: String)))) :: Nil) if TimeZone.getAvailableIDs contains s => c.Expr(tz) case Apply(_, Apply(_, List(tz @ Literal(Constant(s: String)))) :: Nil) => c.abort(c.enclosingPosition, s"Invalid time zone! $s") case _ => ??? // ^^^ What do I do here? I do not want to abort, I merely wish to // "carry on as you were". I've tried ... // c.prefix.tree.asInstanceOf[c.Expr[String]] // ...but that does not work } c.universe.reify(TimeZone.getTimeZone(tzExpr.splice)) }

Answer 1

这是处理时区插值的“歌舞”解决方案：

package object timezone {
  import scala.language.implicitConversions
  implicit def zoned(sc: StringContext) = new ZoneContext(sc)
}

package timezone {
  import scala.language.experimental.macros
  import scala.reflect.macros.Context
  import java.util.TimeZone

  class ZoneContext(sc: StringContext) {

    def tz(args: Any*): TimeZone = macro TimeZoned.tzImpl

    // invoked if runtime interpolation is required
    def tz0(args: Any*): TimeZone = {
      val s = sc.s(args: _*)
      val z = TimeZoned maybeTZ s getOrElse (throw new RuntimeException(s"Bad timezone $s"))
      TimeZone getTimeZone z
    }
  }
  object TimeZoned {
    def maybeTZ(s: String): Option[String] =
      if (TimeZone.getAvailableIDs contains s) Some(s) else None

    def tzImpl(c: Context)(args: c.Expr[Any]*): c.Expr[TimeZone] = {
      import c.universe._
      c.prefix.tree match {
        case Apply(_, List(Apply(_, List(tz @Literal(Constant(const: String)))))) =>
          maybeTZ(const) map (
            k => reify(TimeZone getTimeZone c.Expr[String](tz).splice)
          ) getOrElse c.abort(c.enclosingPosition, s"Bad timezone $const")
        case x =>
          val rts = x.tpe.declaration(newTermName("tz0"))
          val rt = treeBuild.mkAttributedSelect(x, rts)
          c.Expr[TimeZone](Apply(rt, args.map(_.tree).toList))
      }
    }
  }
}

用法：

package tztest 

import timezone._

object Test extends App {

  val delta = 8
  //Console println tz"etc/GMT+$delta"  //java.lang.RuntimeException: Bad timezone etc/GMT+8
  Console println tz"Etc/GMT+$delta"
  Console println tz"US/Hawaii"
  //Console println tz"US/Nowayi"     //error: Bad timezone US/Nowayi
}

Answer 2

问题是您无法将TimeZone的编译时实例移植到宏生成的代码中。但是，您可以滑动字符串文字，这样您就可以生成代码，这些代码将在运行时构造您想要的TimeZone，同时仍然在编译时检查以确保标识符是可用。

以下是一个完整的工作示例：

object TimeZoneLiterals {
  import java.util.TimeZone
  import scala.language.experimental.macros
  import scala.reflect.macros.Context

  implicit class TZContext(val sc: StringContext) extends AnyVal {
    def zone() = macro zoneImpl
  }

  def zoneImpl(c: reflect.macros.Context)() = {
    import c.universe._

    val tzExpr = c.prefix.tree match {
      case Apply(_, Apply(_, List(tz @ Literal(Constant(s: String)))) :: Nil)
        if TimeZone.getAvailableIDs contains s => c.Expr(tz)
      case _ => c.abort(c.enclosingPosition, "Invalid time zone!")
    }

    reify(TimeZone.getTimeZone(tzExpr.splice))
  }
}

reify的参数将是生成方法的主体 - 字面上，不是在任何类型的评估之后，除了tzExpr.slice位将被编译时字符串文字替换（如果当然，您已在可用标识符列表中找到它 - 否则会出现编译时错误。）

StringContext和宏：一个简单的例子

2 个答案: