Timeout Future

timeoutFuture.scala

import java.util.concurrent.{Executors, TimeoutException}

import scala.concurrent.duration._
import scala.concurrent.{ExecutionContext, Future, Promise}
import scala.concurrent.ExecutionContext.Implicits.global

def timeoutFuture[T](f: Future[T], after: Duration)(implicit ec: ExecutionContext) = {
  val p = Promise[T]()
  p tryCompleteWith f
  val action = new Runnable {
    override def run(): Unit = p tryFailure new TimeoutException()
  }
  Executors.newScheduledThreadPool(1).schedule(action, after.toMillis, MILLISECONDS)
  Future.firstCompletedOf(Seq(f, p.future))
}