com.zink.scala.fly.kit.Wait.scala Maven / Gradle / Ivy

Go to download
package com.zink.scala.fly.kit

import scala.annotation.tailrec

class Wait(millis: Long, sleep:Long = 0) {

  def forCondition(f: => Boolean): Boolean = untilSome(if (f) Some(true) else None).isDefined
    
  def untilSome[T](f: => Option[T]): Option[T] = {
    val end = System.currentTimeMillis() + millis

    def done(last:Option[T]) = last.isDefined || System.currentTimeMillis() >= end
    
    @tailrec
    def loop(last: Option[T]): Option[T] = if (done(last)) last else {Thread.sleep(sleep); loop(f)}

    loop(f)
  }
}

object Wait {
  def waitUpTo(millis: Long, sleep:Long = 0) = new Wait(millis, sleep)
}