9

我正在尝试在 dotty 中学习元编程。专门编译时代码生成。我认为通过构建一些东西来学习将是一个好方法。所以我决定制作一个 CSV 解析器，它将行解析为案例类。我想使用 dotty 宏来生成解码器

trait Decoder[T]{
  def decode(str:String):Either[ParseError, T]
}

object Decoder {
  inline given stringDec as Decoder[String] = new Decoder[String] {
    override def decode(str: String): Either[ParseError, String] = Right(str)
  }

  inline given intDec as Decoder[Int] = new Decoder[Int] {
    override def decode(str: String): Either[ParseError, Int] =
      str.toIntOption.toRight(ParseError(str, "value is not valid Int"))
  }
  
  inline def forType[T]:Decoder[T] = ${make[T]}

  def make[T:Type](using qctx: QuoteContext):Expr[Decoder[T]] = ???
}

我已经为Int&提供了基本的解码器String，现在我正在寻找def make[T:Type]方法的指导。如何T在此方法中迭代案例类的参数列表？有没有推荐的方法或模式来做到这一点？

4

1 回答 1

15

在 Dotty 中使用标准类型类派生

object TypeClassParser extends App {
    import scala.deriving._
    import scala.compiletime._

case class ParseError(str: String, msg: String)

trait Decoder[T]{
    def decode(str:String): Either[ParseError, T]
}

object Decoder {
    inline given stringDec: Decoder[String] = new Decoder[String] {
        override def decode(str: String): Either[ParseError, String] = Right(str)
    }

    inline given intDec: Decoder[Int] = new Decoder[Int] {
        override def decode(str: String): Either[ParseError, Int] =
        str.toIntOption.toRight(ParseError(str, "value is not valid Int"))
    }

    inline def derived[T](using m: Mirror.Of[T]): Decoder[T] = {
            val elemInstances = summonAll[m.MirroredElemTypes]
            inline m match {
            case p: Mirror.ProductOf[T] => productDecoder(p, elemInstances)
            case s: Mirror.SumOf[T]     => ???
        }
    }

    inline def summonAll[T <: Tuple]: List[Decoder[_]] = inline erasedValue[T] match {
        case _: EmptyTuple => Nil
        case _: (t *: ts) => summonInline[Decoder[t]] :: summonAll[ts]
    }

    def productDecoder[T](p: Mirror.ProductOf[T], elems: List[Decoder[_]]): Decoder[T] =
        new Decoder[T] {
            def decode(str: String): Either[ParseError, T] = {
                val what = elems.zip(str.split(','))
                .map(_.decode(_).map(_.asInstanceOf[AnyRef]))

                sequence(what)
                .map(ts => p.fromProduct(new ArrayProduct(ts.toArray)))
            }
    }

    def sequence[E,A](es: List[Either[E,A]]): Either[E,List[A]] =
        traverse(es)(x => x)

    def traverse[E,A,B](es: List[A])(f: A => Either[E, B]): Either[E, List[B]] =
        es.foldRight[Either[E, List[B]]](Right(Nil))((h, tRes) => map2(f(h), tRes)(_ :: _))

    def map2[E, A, B, C](a: Either[E, A], b: Either[E, B])(f: (A, B) => C): Either[E, C] =
        for { a1 <- a; b1 <- b } yield f(a1,b1)
    }

    case class A(i: Int, s: String) derives Decoder

    println(summon[Decoder[A]].decode("10,abc"))//Right(A(10,abc))
    println(summon[Decoder[A]].decode("xxx,abc"))//Left(ParseError(xxx,value is not valid Int))
    // println(summon[Decoder[A]].decode(","))

}

在 3.0.0-M3 中测试。

使用Shapeless-3

import shapeless.{K0, Typeable}

case class ParseError(str: String, msg: String)

trait Decoder[T]{
  def decode(str:String): Either[ParseError, T]
}

object Decoder {
  inline given stringDec as Decoder[String] = new Decoder[String] {
    override def decode(str: String): Either[ParseError, String] = Right(str)
  }

  inline given intDec as Decoder[Int] = new Decoder[Int] {
    override def decode(str: String): Either[ParseError, Int] =
      str.toIntOption.toRight(ParseError(str, "value is not valid Int"))
  }

  inline def derived[A](using gen: K0.Generic[A]): Decoder[A] =
    gen.derive(productDecoder, null)

  given productDecoder[T](using inst: K0.ProductInstances[Decoder, T], typeable: Typeable[T]) as Decoder[T] = new Decoder[T] {
    def decode(str: String): Either[ParseError, T] = {
      type Acc = (List[String], Option[ParseError])
      inst.unfold[Decoder, T, Acc](str.split(',').toList, None)([t] => (acc: Acc, dec: Decoder[t]) => 
        acc._1 match {
          case head :: tail => dec.decode(head) match {
            case Right(t) => ((tail, None), Some(t))
            case Left(e)  => ((Nil, Some(e)), None)
          }
          case Nil => (acc, None)
        }
      ) match {
        case ((_, Some(e)), None) => Left(e)
        case ((_, None), None)    => Left(ParseError(str, s"value is not valid ${typeable.describe}"))
        case (_, Some(t))         => Right(t)
      }
    }
  }
}

case class A(i: Int, s: String) derives Decoder

@main def test = {
  println(summon[Decoder[A]].decode("10,abc")) //Right(A(10,abc))
  println(summon[Decoder[A]].decode("xxx,abc")) //Left(ParseError(xxx,value is not valid Int))
  println(summon[Decoder[A]].decode(",")) //Left(ParseError(,,value is not valid A))
}

构建.sbt

scalaVersion := "0.24.0"

libraryDependencies += "org.typelevel" %% "shapeless-core" % "3.0.0-M1"

项目/plugins.sbt

addSbtPlugin("ch.epfl.lamp" % "sbt-dotty" % "0.4.1")

在dotty -macro-examples/macroTypeclassDerivation中使用Dotty 宏+ TASTy 反射（这种方法比使用 的方法更底层）scala.deriving.Mirror

import scala.quoted._
  
case class ParseError(str: String, msg: String)

trait Decoder[T]{
  def decode(str:String): Either[ParseError, T]
}

object Decoder {
  inline given stringDec as Decoder[String] = new Decoder[String] {
    override def decode(str: String): Either[ParseError, String] = Right(str)
  }

  inline given intDec as Decoder[Int] = new Decoder[Int] {
    override def decode(str: String): Either[ParseError, Int] =
      str.toIntOption.toRight(ParseError(str, "value is not valid Int"))
  }

  inline def derived[T]: Decoder[T] = ${ derivedImpl[T] }

  def derivedImpl[T](using qctx: QuoteContext, tpe: Type[T]): Expr[Decoder[T]] = {
    import qctx.tasty._
    val tpeSym = tpe.unseal.symbol
    if (tpeSym.flags.is(Flags.Case)) productDecoder[T]
    else if (tpeSym.flags.is(Flags.Trait & Flags.Sealed)) ???
    else sys.error(s"Unsupported combination of flags: ${tpeSym.flags.show}")
  }

  def productDecoder[T](using qctx: QuoteContext, tpe: Type[T]): Expr[Decoder[T]] = {
    import qctx.tasty._    
    val fields: List[Symbol]             = tpe.unseal.symbol.caseFields
    val fieldTypeTrees: List[TypeTree]   = fields.map(_.tree.asInstanceOf[ValDef].tpt)
    val fieldTypes: List[Type]           = fieldTypeTrees.map(_.tpe)
    val decoderTerms: List[Term]         = fieldTypes.map(lookupDecoderFor(_))
    val decoders: Expr[List[Decoder[_]]] = Expr.ofList(decoderTerms.map(_.seal.cast[Decoder[_]]))

    def mkT(fields: Expr[List[_]]): Expr[T] = {
      Apply(
        Select.unique(New(tpe.unseal), "<init>"),
        fieldTypeTrees.zipWithIndex.map((fieldType, i) =>
          TypeApply(
            Select.unique(
              Apply(
                Select.unique(
                  fields.unseal,
                  "apply"),
                List(Literal(Constant(i)))
              ), "asInstanceOf"),
            List(fieldType)
          )
        )
      ).seal.cast[T]
    } 
      
    '{
      new Decoder[T]{
        override def decode(str: String): Either[ParseError, T] = {
          str.split(',').toList.zip($decoders).map((str, decoder) =>
            decoder.decode(str)
          ).sequence.map(fields =>
            ${mkT('fields)}
          )
        }
      }
    }
  }

  def lookupDecoderFor(using qctx: QuoteContext)(t: qctx.tasty.Type): qctx.tasty.Term = {
    import qctx.tasty._
    val tpe = AppliedType(Type(classOf[Decoder[_]]), List(t))
    searchImplicit(tpe) match {
      case res: ImplicitSearchSuccess => res.tree
    }
  }

  def [E,A](es: List[Either[E,A]]) sequence: Either[E,List[A]] =
    traverse(es)(x => x)

  def traverse[E,A,B](es: List[A])(f: A => Either[E, B]): Either[E, List[B]] =
    es.foldRight[Either[E, List[B]]](Right(Nil))((h, tRes) => map2(f(h), tRes)(_ :: _))

  def map2[E, A, B, C](a: Either[E, A], b: Either[E, B])(f: (A, B) => C): Either[E, C] =
    for { a1 <- a; b1 <- b } yield f(a1,b1)
}

case class A(i: Int, s: String) derives Decoder

@main def test = {
  println(summon[Decoder[A]].decode("10,abc"))//Right(A(10,abc))
  println(summon[Decoder[A]].decode("xxx,abc"))//Left(ParseError(xxx,value is not valid Int))
 // println(summon[Decoder[A]].decode(","))
}

在 0.24.0 中测试。

用于比较 Scala 2 中的派生类型类

使用类型类中的最低子类型？

于 2020-07-12T13:05:16.660 回答