mill.define.Cacher.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mill-main-define_2.13 Show documentation

mill-main-define

The newest version!

package mill.define

import scala.collection.mutable
import scala.reflect.macros.blackbox.Context

private[mill] trait Cacher extends mill.moduledefs.Cacher {
  private[this] lazy val cacherLazyMap = mutable.Map.empty[sourcecode.Enclosing, Any]

  override protected[this] def cachedTarget[T](t: => T)(implicit c: sourcecode.Enclosing): T =
    synchronized {
      cacherLazyMap.getOrElseUpdate(c, t).asInstanceOf[T]
    }
}

private[mill] object Cacher {
  def impl0[T: c.WeakTypeTag](c: Context)(t: c.Expr[T]): c.Expr[T] = {
    c.Expr[T](wrapCached[T](c)(t.tree))
  }
  def wrapCached[R: c.WeakTypeTag](c: Context)(t: c.Tree): c.universe.Tree = {

    import c.universe._
    val owner = c.internal.enclosingOwner
    val ownerIsCacherClass =
      owner.owner.isClass &&
        owner.owner.asClass.baseClasses.exists(_.fullName == "mill.define.Cacher")

    if (ownerIsCacherClass && owner.isMethod) q"this.cachedTarget[${weakTypeTag[R]}]($t)"
    else c.abort(
      c.enclosingPosition,
      "Task{} members must be defs defined in a Module class/trait/object body"
    )
  }
}