Как я могу использовать макрос / квазицитату Scala для шаблонов кода? - PullRequest
Новая
       63

Как я могу использовать макрос / квазицитату Scala для шаблонов кода?

0 голосов
/ 21 января 2019

Я хочу создать группу объектов во время компиляции, которые следуют простому шаблону, поэтому я написал следующий макрос: 

object MyMacro {

  def readWrite[T](taName: String, readParse: String => T, label: String, format: T => String): Any = macro readWriteImpl[T]

  def readWriteImpl[T: c.WeakTypeTag](c: Context)(taName: c.Expr[String], readParse: c.Expr[String => T], label: c.Expr[String], format: c.Expr[T => String]): c.Expr[Any] = {
    import c.universe._

    def termName(s: c.Expr[String]): TermName = s.tree match {
      case Literal(Constant(s: String)) => TermName(s)
      case _ => c.abort(c.enclosingPosition, "Not a string literal")
    }

    c.Expr[Any](q"""
    object ${termName(taName)} extends TypeAdapter.=:=[${implicitly[c.WeakTypeTag[T]].tpe}] {
      def read[WIRE](path: Path, reader: Transceiver[WIRE], isMapKey: Boolean = false): ${implicitly[c.WeakTypeTag[T]].tpe} =
        reader.readString(path) match {
          case null => null.asInstanceOf[${implicitly[c.WeakTypeTag[T]].tpe}]
          case s => Try( $readParse(s) ) match {
            case Success(d) => d
            case Failure(u) => throw new ReadMalformedError(path, "Failed to parse "+${termName(label)}+" from input '"+s+"'", List.empty[String], u)
          }
        }

      def write[WIRE](t: ${implicitly[c.WeakTypeTag[T]].tpe}, writer: Transceiver[WIRE], out: Builder[Any, WIRE]): Unit =
        t match {
          case null => writer.writeNull(out)
          case _    => writer.writeString($format(t), out)
        }
    }
    """)
  }
}

Я не уверен, что у меня есть возвращаемое значение для readWrite и readWriteImplправильно - компилятор сильно жалуется на какой-то сбой утверждения!

Я также не уверен, как на самом деле использовать этот макрос.Сначала я попробовал (в отдельном модуле компиляции): 

object TimeFactories {
    MyMacro.readWrite[Duration](
      "DurationTypeAdapterFactory",
      (s: String) => Duration.parse(s),
      "Duration",
      (t: Duration) => t.toString)
}

Не сработало.Если я попытался сослаться на TimeFactories.DurationTypeAdapterFactory, я получил сообщение об ошибке, сказав, что он не найден.Затем я подумал, что попробую назначить его в val ... тоже не сработало: 

object Foo {
  val duration = MyMacro.readWrite[Duration](
    "DurationTypeAdapterFactory",
    (s: String) => Duration.parse(s),
    "Duration",
    (t: Duration) => t.toString).asInstanceOf[TypeAdapterFactory]
}

Как мне это подключить, чтобы сгенерировать сгенерированный код следующим образом: 

object TimeFactories{
    object DurationTypeAdapterFactory extends TypeAdapter.=:=[Duration] {
      def read[WIRE](path: Path, reader: Transceiver[WIRE], isMapKey: Boolean = false): Duration =
        reader.readString(path) match {
          case null => null.asInstanceOf[Duration]
          case s => Try( Duration.parse(s) ) match {
            case Success(d) => d
            case Failure(u) => throw new ReadMalformedError(path, "Failed to parse Duration from input 'Duration'", List.empty[String], u)
          }
        }

      def write[WIRE](t: Duration, writer: Transceiver[WIRE], out: Builder[Any, WIRE]): Unit =
        t match {
          case null => writer.writeNull(out)
          case _    => writer.writeString(t.toString, out)
        }
    }

// ... More invocations of the readWrite macro with other types for T
}

1 Ответ

0 голосов
/ 21 января 2019

Я не думаю, что вы можете генерировать новые идентификаторы, используя макросы, а затем использовать их публично.

Вместо этого попробуйте заменить object ${termName(taName)} extends TypeAdapter просто на new TypeAdapter и назначить вызов макроса дляval (как в вашем втором примере).Затем вы будете ссылаться на анонимный (и сгенерированный) класс, хранящийся в val.Параметр taName становится избыточным.

...