au.csiro.pbdava.ssparkle.common.utils.Timed.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of variant-spark_2.11 Show documentation

Genomic variants interpretation toolkit

The newest version!

package au.csiro.pbdava.ssparkle.common.utils

case class Timer() {
  val startTime: Long = System.currentTimeMillis()
  lazy val duration: Long = System.currentTimeMillis() - startTime
  def durationInSec: Double = duration / 1000.0
}

case class TimedResult[T](result: T, elapsedTime: Long) {
  def withResult(f: T => Unit): TimedResult[T] = {
    f(result)
    this
  }
  def withResultAndTime(f: (T, Long) => Unit): TimedResult[T] = {
    f(result, elapsedTime)
    this
  }

  def report(msg: => String): TimedResult[T] = {
    println(s"${msg} time: ${elapsedTime}")
    this
  }
}

object Timed {
  def time[T](c: => T): TimedResult[T] = {
    val startTime = System.currentTimeMillis()
    val result = c
    val elapsedTime = System.currentTimeMillis() - startTime
    TimedResult(result, elapsedTime)
  }
}