lihua.mongo.ShutdownHook.scala Maven / Gradle / Ivy

Go to download

/*
* Copyright [2017] [iHeartMedia Inc]
* All rights reserved
*/
package lihua.mongo

trait ShutdownHook {
  def onShutdown[T](code: ⇒ T): Unit
}


object ShutdownHook {

  /**
    * a shutdownhook
    */
  object ignore extends ShutdownHook {
    override def onShutdown[T](code: => T): Unit = ()
  }

  object manual extends ShutdownHook {
    @volatile
    private var callbacks: List[() => _] = Nil
    override def onShutdown[T](code: => T): Unit = {
      callbacks = (() => code) :: callbacks
    }
    def shutdown(): Unit = callbacks.foreach(_())
  }
}