scaldi.Lifecycle.scala Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of scaldi_2.11 Show documentation

Scaldi - Scala Dependency Injection Library

There is a newer version: 0.6.2

package scaldi

import scaldi.util.Util._
import scala.util.{Success, Failure, Try}
import scala.util.control.Breaks._

case class BindingLifecycle[-T](
  initialize: Option[T => Unit] = None,
  destroy: Option[T => Unit] = None
) {
  def initializeObject(obj: T): Unit = obj <| (o => initialize foreach (_(o)))
  def destroyObject(obj: T): Unit = obj <| (o => destroy foreach (_(o)))
}

object BindingLifecycle {
  def empty[T] = BindingLifecycle[T]()
}

trait LifecycleManager {
  val IgnoringErrorHandler =
    (e: Throwable) => {e.printStackTrace(); true}

  def addDestroyable(fn: () => Unit): Unit

  def destroy(errorHandler: Throwable => Boolean = IgnoringErrorHandler): Unit
}

trait ShutdownHookLifecycleManager extends LifecycleManager {
  private var toDestroy: List[() => Unit] = Nil

  def addDestroyable(fn: () => Unit) = this.synchronized {
    if (toDestroy.isEmpty) addShutdownHook()
    toDestroy = toDestroy :+ fn
  }

  private def addShutdownHook() = {
    sys.addShutdownHook(doDestroyAll(IgnoringErrorHandler))
  }

  def destroy(errorHandler: Throwable => Boolean = IgnoringErrorHandler) =
    doDestroyAll(errorHandler)

  private def doDestroyAll(errorHandler: Throwable => Boolean) = this.synchronized {
    breakable {
      toDestroy.reverse.foreach { d =>
        Try(d()) match {
          case Failure(error) =>
            if (!errorHandler(error)) break()
          case Success(_) => // just continue
        }
      }
    }

    toDestroy = Nil
  }
}