• Home
• io.getquill
• quill-core_2.11
• 0.6.0
• source code
• CollectAst.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

io.getquill.ast.CollectAst.scala Maven / Gradle / Ivy

package io.getquill.ast

class CollectAst[T](p: PartialFunction[Ast, T], val state: List[T])
  extends StatefulTransformer[List[T]] {

  override def apply(a: Ast) =
    a match {
      case d if (p.isDefinedAt(d)) => (d, new CollectAst(p, state :+ p(d)))
      case other                   => super.apply(other)
    }
}

object CollectAst {
  def apply[T](a: Ast)(p: PartialFunction[Ast, T]) =
    (new CollectAst(p, List()).apply(a)) match {
      case (_, transformer) =>
        transformer.state
    }
}