pekko/akka-actor/src/main/scala/akka/dispatch/Future.scala

811 lines
27 KiB
Scala
Raw Normal View History

/**
* Copyright (C) 2009-2011 Scalable Solutions AB <http://scalablesolutions.se>
*/
package akka.dispatch
import akka.AkkaException
import akka.event.EventHandler
import akka.actor.{ Actor, Channel }
import akka.util.Duration
import akka.japi.{ Procedure, Function JFunc }
2010-10-31 19:27:55 +01:00
import scala.util.continuations._
import java.util.concurrent.locks.ReentrantLock
import java.util.concurrent.{ ConcurrentLinkedQueue, TimeUnit, Callable }
import java.util.concurrent.TimeUnit.{ NANOSECONDS NANOS, MILLISECONDS MILLIS }
import java.util.concurrent.atomic.{ AtomicBoolean }
import java.lang.{ Iterable JIterable }
import java.util.{ LinkedList JLinkedList }
2011-04-27 00:38:10 +02:00
import scala.annotation.tailrec
import scala.collection.mutable.Stack
2011-04-29 17:15:00 +02:00
class FutureTimeoutException(message: String, cause: Throwable = null) extends AkkaException(message, cause)
object Futures {
/**
* Java API, equivalent to Future.apply
*/
2011-02-28 16:29:03 -07:00
def future[T](body: Callable[T]): Future[T] =
Future(body.call)
/**
* Java API, equivalent to Future.apply
*/
2011-02-28 16:29:03 -07:00
def future[T](body: Callable[T], timeout: Long): Future[T] =
Future(body.call, timeout)
/**
* Java API, equivalent to Future.apply
*/
def future[T](body: Callable[T], dispatcher: MessageDispatcher): Future[T] =
Future(body.call)(dispatcher)
/**
* Java API, equivalent to Future.apply
*/
def future[T](body: Callable[T], timeout: Long, dispatcher: MessageDispatcher): Future[T] =
Future(body.call, timeout)(dispatcher)
/**
* Returns a Future to the result of the first future in the list that is completed
*/
def firstCompletedOf[T](futures: Iterable[Future[T]], timeout: Long = Long.MaxValue): Future[T] = {
val futureResult = new DefaultPromise[T](timeout)
val completeFirst: Future[T] Unit = _.value.foreach(futureResult complete _)
for (f futures) f onComplete completeFirst
futureResult
}
/**
2011-03-30 21:19:26 +02:00
* Java API.
* Returns a Future to the result of the first future in the list that is completed
*/
def firstCompletedOf[T <: AnyRef](futures: java.lang.Iterable[Future[T]], timeout: Long): Future[T] =
firstCompletedOf(scala.collection.JavaConversions.iterableAsScalaIterable(futures), timeout)
2011-01-24 13:10:52 +01:00
/**
* A non-blocking fold over the specified futures.
* The fold is performed on the thread where the last future is completed,
* the result will be the first failure of any of the futures, or any failure in the actual fold,
* or the result of the fold.
2011-03-30 21:19:26 +02:00
* Example:
* <pre>
* val result = Futures.fold(0)(futures)(_ + _).await.result
* </pre>
2011-01-24 13:10:52 +01:00
*/
def fold[T, R](zero: R, timeout: Long = Actor.TIMEOUT)(futures: Iterable[Future[T]])(foldFun: (R, T) R): Future[R] = {
if (futures.isEmpty) {
new KeptPromise[R](Right(zero))
} else {
val result = new DefaultPromise[R](timeout)
val results = new ConcurrentLinkedQueue[T]()
val allDone = futures.size
val aggregate: Future[T] Unit = f if (!result.isCompleted) { //TODO: This is an optimization, is it premature?
f.value.get match {
case r: Right[Throwable, T]
results add r.b
if (results.size == allDone) { //Only one thread can get here
try {
result completeWithResult scala.collection.JavaConversions.collectionAsScalaIterable(results).foldLeft(zero)(foldFun)
} catch {
case e: Exception
EventHandler.error(e, this, e.getMessage)
result completeWithException e
}
finally {
results.clear
}
}
case l: Left[Throwable, T]
result completeWithException l.a
results.clear
}
}
futures foreach { _ onComplete aggregate }
result
}
}
2011-01-24 16:37:08 +01:00
/**
* Java API
* A non-blocking fold over the specified futures.
* The fold is performed on the thread where the last future is completed,
* the result will be the first failure of any of the futures, or any failure in the actual fold,
* or the result of the fold.
*/
def fold[T <: AnyRef, R <: AnyRef](zero: R, timeout: Long, futures: java.lang.Iterable[Future[T]], fun: akka.japi.Function2[R, T, R]): Future[R] =
fold(zero, timeout)(scala.collection.JavaConversions.iterableAsScalaIterable(futures))(fun.apply _)
2011-01-24 16:37:08 +01:00
/**
* Initiates a fold over the supplied futures where the fold-zero is the result value of the Future that's completed first
2011-03-30 21:19:26 +02:00
* Example:
* <pre>
* val result = Futures.reduce(futures)(_ + _).await.result
* </pre>
2011-01-24 16:37:08 +01:00
*/
def reduce[T, R >: T](futures: Iterable[Future[T]], timeout: Long = Actor.TIMEOUT)(op: (R, T) T): Future[R] = {
2011-01-24 17:12:56 +01:00
if (futures.isEmpty)
new KeptPromise[R](Left(new UnsupportedOperationException("empty reduce left")))
else {
val result = new DefaultPromise[R](timeout)
val seedFound = new AtomicBoolean(false)
val seedFold: Future[T] Unit = f {
if (seedFound.compareAndSet(false, true)) { //Only the first completed should trigger the fold
f.value.get match {
case r: Right[Throwable, T]
2011-03-18 11:18:03 +01:00
result.completeWith(fold(r.b, timeout)(futures.filterNot(_ eq f))(op))
case l: Left[Throwable, T]
result.completeWithException(l.a)
}
}
2011-01-24 16:37:08 +01:00
}
for (f futures) f onComplete seedFold //Attach the listener to the Futures
result
2011-01-24 16:37:08 +01:00
}
}
/**
2011-03-30 21:19:26 +02:00
* Java API.
* Initiates a fold over the supplied futures where the fold-zero is the result value of the Future that's completed first
*/
def reduce[T <: AnyRef, R >: T](futures: java.lang.Iterable[Future[T]], timeout: Long, fun: akka.japi.Function2[R, T, T]): Future[R] =
reduce(scala.collection.JavaConversions.iterableAsScalaIterable(futures), timeout)(fun.apply _)
/**
* Java API.
* Simple version of Futures.traverse. Transforms a java.lang.Iterable[Future[A]] into a Future[java.lang.Iterable[A]].
* Useful for reducing many Futures into a single Future.
*/
def sequence[A](in: JIterable[Future[A]], timeout: Long): Future[JIterable[A]] =
scala.collection.JavaConversions.iterableAsScalaIterable(in).foldLeft(Future(new JLinkedList[A]()))((fr, fa)
for (r fr; a fa) yield {
r add a
r
})
2011-01-24 16:37:08 +01:00
2011-03-30 21:19:26 +02:00
/**
* Java API.
* Simple version of Futures.traverse. Transforms a java.lang.Iterable[Future[A]] into a Future[java.lang.Iterable[A]].
* Useful for reducing many Futures into a single Future.
2011-03-30 21:19:26 +02:00
*/
def sequence[A](in: JIterable[Future[A]]): Future[JIterable[A]] = sequence(in, Actor.TIMEOUT)
2011-03-30 21:19:26 +02:00
/**
* Java API.
* Transforms a java.lang.Iterable[A] into a Future[java.lang.Iterable[B]] using the provided Function A => Future[B].
* This is useful for performing a parallel map. For example, to apply a function to all items of a list
* in parallel.
2011-03-30 21:19:26 +02:00
*/
def traverse[A, B](in: JIterable[A], timeout: Long, fn: JFunc[A, Future[B]]): Future[JIterable[B]] =
scala.collection.JavaConversions.iterableAsScalaIterable(in).foldLeft(Future(new JLinkedList[B]())) { (fr, a)
val fb = fn(a)
for (r fr; b fb) yield {
r add b
r
}
}
/**
* Java API.
* Transforms a java.lang.Iterable[A] into a Future[java.lang.Iterable[B]] using the provided Function A => Future[B].
* This is useful for performing a parallel map. For example, to apply a function to all items of a list
* in parallel.
*
* def traverse[A, B, M[_] <: Traversable[_]](in: M[A], timeout: Long = Actor.TIMEOUT)(fn: A => Future[B])(implicit cbf: CanBuildFrom[M[A], B, M[B]]): Future[M[B]] =
* in.foldLeft(new DefaultPromise[Builder[B, M[B]]](timeout).completeWithResult(cbf(in)): Future[Builder[B, M[B]]]) { (fr, a) =>
* val fb = fn(a.asInstanceOf[A])
* for (r <- fr; b <-fb) yield (r += b)
* }.map(_.result)
*/
def traverse[A, B](in: JIterable[A], fn: JFunc[A, Future[B]]): Future[JIterable[B]] = traverse(in, Actor.TIMEOUT, fn)
}
object Future {
2011-04-27 01:06:08 +02:00
/**
* This method constructs and returns a Future that will eventually hold the result of the execution of the supplied body
* The execution is performed by the specified Dispatcher.
*/
def apply[T](body: T, timeout: Long = Actor.TIMEOUT)(implicit dispatcher: MessageDispatcher): Future[T] =
dispatcher.dispatchFuture(() body, timeout)
/**
* Construct a completable channel
*/
def channel(timeout: Long = Actor.TIMEOUT) = new Channel[Any] {
2011-04-23 11:49:03 +02:00
val future = empty[Any](timeout)
def !(msg: Any) = future completeWithResult msg
}
2011-04-23 11:35:54 +02:00
/**
* Create an empty Future with default timeout
*/
def empty[T](timeout: Long = Actor.TIMEOUT) = new DefaultPromise[T](timeout)
2011-04-23 11:35:54 +02:00
import scala.collection.mutable.Builder
import scala.collection.generic.CanBuildFrom
/**
* Simple version of Futures.traverse. Transforms a Traversable[Future[A]] into a Future[Traversable[A]].
* Useful for reducing many Futures into a single Future.
*/
def sequence[A, M[_] <: Traversable[_]](in: M[Future[A]], timeout: Long = Actor.TIMEOUT)(implicit cbf: CanBuildFrom[M[Future[A]], A, M[A]]): Future[M[A]] =
in.foldLeft(new DefaultPromise[Builder[A, M[A]]](timeout).completeWithResult(cbf(in)): Future[Builder[A, M[A]]])((fr, fa) for (r fr; a fa.asInstanceOf[Future[A]]) yield (r += a)).map(_.result)
/**
* Transforms a Traversable[A] into a Future[Traversable[B]] using the provided Function A => Future[B].
* This is useful for performing a parallel map. For example, to apply a function to all items of a list
* in parallel:
* <pre>
* val myFutureList = Futures.traverse(myList)(x => Future(myFunc(x)))
* </pre>
*/
def traverse[A, B, M[_] <: Traversable[_]](in: M[A], timeout: Long = Actor.TIMEOUT)(fn: A Future[B])(implicit cbf: CanBuildFrom[M[A], B, M[B]]): Future[M[B]] =
in.foldLeft(new DefaultPromise[Builder[B, M[B]]](timeout).completeWithResult(cbf(in)): Future[Builder[B, M[B]]]) { (fr, a)
val fb = fn(a.asInstanceOf[A])
for (r fr; b fb) yield (r += b)
}.map(_.result)
/**
* Captures a block that will be transformed into 'Continuation Passing Style' using Scala's Delimited
* Continuations plugin.
*
* Within the block, the result of a Future may be accessed by calling Future.apply. At that point
* execution is suspended with the rest of the block being stored in a continuation until the result
* of the Future is available. If an Exception is thrown while processing, it will be contained
* within the resulting Future.
*
* This allows working with Futures in an imperative style without blocking for each result.
*
* Completing a Future using 'Promise << Future' will also suspend execution until the
* value of the other Future is available.
*
* The Delimited Continuations compiler plugin must be enabled in order to use this method.
*/
def flow[A](body: A @cps[Future[Any]], timeout: Long = Actor.TIMEOUT): Future[A] = {
val future = Promise[A](timeout)
(reset(future.asInstanceOf[Promise[Any]].completeWithResult(body)): Future[Any]) onComplete { f
val opte = f.exception
if (opte.isDefined) future completeWithException (opte.get)
}
future
}
}
sealed trait Future[+T] {
/**
* For use only within a Future.flow block or another compatible Delimited Continuations reset block.
*
* Returns the result of this Future without blocking, by suspending execution and storing it as a
* continuation until the result is available.
*
* If this Future is untyped (a Future[Nothing]), a type parameter must be explicitly provided or
* execution will fail. The normal result of getting a Future from an ActorRef using !!! will return
* an untyped Future.
*/
def apply[A >: T](): A @cps[Future[Any]] = shift(this flatMap (_: A Future[Any]))
/**
2011-04-26 10:49:52 +02:00
* Blocks awaiting completion of this Future, then returns the resulting value,
* or throws the completed exception
*
* Scala & Java API
*
* throws FutureTimeoutException if this Future times out when waiting for completion
*/
def get: T = this.await.resultOrException.get
2011-02-23 19:57:42 -07:00
/**
* Blocks the current thread until the Future has been completed or the
* timeout has expired. In the case of the timeout expiring a
* FutureTimeoutException will be thrown.
*/
def await: Future[T]
2010-10-31 19:27:55 +01:00
/**
* Blocks the current thread until the Future has been completed or the
* timeout has expired. The timeout will be the least value of 'atMost' and the timeout
* supplied at the constructuion of this Future.
* In the case of the timeout expiring a FutureTimeoutException will be thrown.
*/
def await(atMost: Duration): Future[T]
2011-02-23 19:57:42 -07:00
/**
* Blocks the current thread until the Future has been completed. Use
* caution with this method as it ignores the timeout and will block
* indefinitely if the Future is never completed.
*/
2011-05-06 11:55:35 +12:00
@deprecated("Will be removed after 1.1, it's dangerous and can cause deadlocks, agony and insanity.", "1.1")
def awaitBlocking: Future[T]
2010-10-31 19:27:55 +01:00
2011-02-23 19:57:42 -07:00
/**
* Tests whether this Future has been completed.
*/
final def isCompleted: Boolean = value.isDefined
2010-10-31 19:27:55 +01:00
2011-02-23 19:57:42 -07:00
/**
* Tests whether this Future's timeout has expired.
*
* Note that an expired Future may still contain a value, or it may be
* completed with a value.
*/
def isExpired: Boolean
2010-10-31 19:27:55 +01:00
2011-02-23 19:57:42 -07:00
/**
* This Future's timeout in nanoseconds.
*/
def timeoutInNanos: Long
2010-10-31 19:27:55 +01:00
2011-02-23 19:57:42 -07:00
/**
* The contained value of this Future. Before this Future is completed
* the value will be None. After completion the value will be Some(Right(t))
* if it contains a valid result, or Some(Left(error)) if it contains
* an exception.
*/
def value: Option[Either[Throwable, T]]
2011-02-23 19:57:42 -07:00
/**
* Returns the successful result of this Future if it exists.
*/
final def result: Option[T] = {
val v = value
if (v.isDefined) v.get.right.toOption
else None
}
2010-10-31 19:27:55 +01:00
2011-02-23 19:57:42 -07:00
/**
* Returns the contained exception of this Future if it exists.
*/
final def exception: Option[Throwable] = {
val v = value
if (v.isDefined) v.get.left.toOption
else None
}
2010-10-31 19:27:55 +01:00
2011-02-23 19:57:42 -07:00
/**
* When this Future is completed, apply the provided function to the
* Future. If the Future has already been completed, this will apply
* immediately.
2011-02-23 19:57:42 -07:00
*/
def onComplete(func: Future[T] Unit): Future[T]
/**
* When the future is completed with a valid result, apply the provided
2011-02-23 19:57:42 -07:00
* PartialFunction to the result.
2011-03-30 21:19:26 +02:00
* <pre>
* val result = future receive {
* case Foo => "foo"
* case Bar => "bar"
* }.await.result
* </pre>
*/
final def receive(pf: PartialFunction[Any, Unit]): Future[T] = onComplete { f
val optr = f.result
if (optr.isDefined) {
val r = optr.get
if (pf.isDefinedAt(r)) pf(r)
}
}
/**
* Creates a new Future by applying a PartialFunction to the successful
* result of this Future if a match is found, or else return a MatchError.
* If this Future is completed with an exception then the new Future will
* also contain this exception.
2011-03-30 21:19:26 +02:00
* Example:
* <pre>
* val future1 = for {
* a <- actor !!! Req("Hello") collect { case Res(x: Int) => x }
* b <- actor !!! Req(a) collect { case Res(x: String) => x }
* c <- actor !!! Req(7) collect { case Res(x: String) => x }
* } yield b + "-" + c
* </pre>
*/
final def collect[A](pf: PartialFunction[Any, A]): Future[A] = {
val fa = new DefaultPromise[A](timeoutInNanos, NANOS)
onComplete { ft
2011-04-25 16:14:07 -06:00
val v = ft.value.get
fa complete {
if (v.isLeft) v.asInstanceOf[Either[Throwable, A]]
else {
try {
val r = v.right.get
if (pf isDefinedAt r) Right(pf(r))
else Left(new MatchError(r))
} catch {
case e: Exception
2011-04-25 16:14:07 -06:00
EventHandler.error(e, this, e.getMessage)
Left(e)
}
}
}
}
fa
}
2011-04-25 16:14:07 -06:00
/**
* Creates a new Future that will handle any matching Throwable that this
* Future might contain. If there is no match, or if this Future contains
* a valid result then the new Future will contain the same.
* Example:
* <pre>
* Future(6 / 0) failure { case e: ArithmeticException => 0 } // result: 0
* Future(6 / 0) failure { case e: NotFoundException => 0 } // result: exception
* Future(6 / 2) failure { case e: ArithmeticException => 0 } // result: 3
* </pre>
*/
final def failure[A >: T](pf: PartialFunction[Throwable, A]): Future[A] = {
val fa = new DefaultPromise[A](timeoutInNanos, NANOS)
onComplete { ft
2011-04-25 16:14:07 -06:00
val opte = ft.exception
fa complete {
if (opte.isDefined) {
val e = opte.get
try {
if (pf isDefinedAt e) Right(pf(e))
else Left(e)
} catch {
case x: Exception Left(x)
2011-04-25 16:14:07 -06:00
}
} else ft.value.get
}
}
fa
}
2011-02-23 19:57:42 -07:00
/**
* Creates a new Future by applying a function to the successful result of
* this Future. If this Future is completed with an exception then the new
* Future will also contain this exception.
2011-03-30 21:19:26 +02:00
* Example:
* <pre>
* val future1 = for {
* a: Int <- actor !!! "Hello" // returns 5
* b: String <- actor !!! a // returns "10"
* c: String <- actor !!! 7 // returns "14"
* } yield b + "-" + c
* </pre>
2011-02-23 19:57:42 -07:00
*/
final def map[A](f: T A): Future[A] = {
val fa = new DefaultPromise[A](timeoutInNanos, NANOS)
onComplete { ft
2011-02-22 20:32:59 -07:00
val optv = ft.value
if (optv.isDefined) {
val v = optv.get
if (v.isLeft)
fa complete v.asInstanceOf[Either[Throwable, A]]
else {
fa complete (try {
Right(f(v.right.get))
} catch {
case e: Exception
EventHandler.error(e, this, e.getMessage)
Left(e)
2011-02-22 20:32:59 -07:00
})
}
}
}
fa
}
2011-02-23 19:57:42 -07:00
/**
* Creates a new Future by applying a function to the successful result of
* this Future, and returns the result of the function as the new Future.
* If this Future is completed with an exception then the new Future will
* also contain this exception.
2011-03-30 21:19:26 +02:00
* Example:
* <pre>
* val future1 = for {
* a: Int <- actor !!! "Hello" // returns 5
* b: String <- actor !!! a // returns "10"
* c: String <- actor !!! 7 // returns "14"
* } yield b + "-" + c
* </pre>
2011-02-23 19:57:42 -07:00
*/
final def flatMap[A](f: T Future[A]): Future[A] = {
val fa = new DefaultPromise[A](timeoutInNanos, NANOS)
onComplete { ft
2011-02-22 20:32:59 -07:00
val optv = ft.value
if (optv.isDefined) {
val v = optv.get
if (v.isLeft)
fa complete v.asInstanceOf[Either[Throwable, A]]
else {
try {
fa.completeWith(f(v.right.get))
2011-02-22 20:32:59 -07:00
} catch {
case e: Exception
EventHandler.error(e, this, e.getMessage)
fa completeWithException e
2011-02-22 20:32:59 -07:00
}
}
}
}
fa
}
final def foreach(f: T Unit): Unit = onComplete { ft
val optr = ft.result
if (optr.isDefined)
f(optr.get)
}
final def filter(p: Any Boolean): Future[Any] = {
val f = new DefaultPromise[T](timeoutInNanos, NANOS)
onComplete { ft
2011-02-22 20:32:59 -07:00
val optv = ft.value
if (optv.isDefined) {
val v = optv.get
if (v.isLeft)
f complete v
else {
val r = v.right.get
f complete (try {
if (p(r)) Right(r)
else Left(new MatchError(r))
} catch {
case e: Exception
EventHandler.error(e, this, e.getMessage)
Left(e)
2011-02-22 20:32:59 -07:00
})
}
}
}
f
}
/**
2011-03-30 21:19:26 +02:00
* Returns the current result, throws the exception is one has been raised, else returns None
*/
final def resultOrException: Option[T] = {
val v = value
if (v.isDefined) {
val r = v.get
if (r.isLeft) throw r.left.get
else r.right.toOption
} else None
}
/* Java API */
final def onComplete[A >: T](proc: Procedure[Future[A]]): Future[T] = onComplete(proc(_))
final def map[A >: T, B](f: JFunc[A, B]): Future[B] = map(f(_))
final def flatMap[A >: T, B](f: JFunc[A, Future[B]]): Future[B] = flatMap(f(_))
final def foreach[A >: T](proc: Procedure[A]): Unit = foreach(proc(_))
final def filter(p: JFunc[Any, Boolean]): Future[Any] = filter(p(_))
}
object Promise {
def apply[A](timeout: Long): Promise[A] = new DefaultPromise[A](timeout)
def apply[A](): Promise[A] = apply(Actor.TIMEOUT)
2011-05-23 11:54:41 +02:00
private[akka] val callbacksPendingExecution = new ThreadLocal[Option[Stack[() Unit]]]() {
override def initialValue = None
}
}
/**
2011-03-30 21:19:26 +02:00
* Essentially this is the Promise (or write-side) of a Future (read-side).
*/
trait Promise[T] extends Future[T] {
/**
2011-03-30 21:19:26 +02:00
* Completes this Future with the specified result, if not already completed.
* @return this
*/
def complete(value: Either[Throwable, T]): Future[T]
/**
2011-03-30 21:19:26 +02:00
* Completes this Future with the specified result, if not already completed.
* @return this
*/
final def completeWithResult(result: T): Future[T] = complete(Right(result))
/**
2011-03-30 21:19:26 +02:00
* Completes this Future with the specified exception, if not already completed.
* @return this
*/
final def completeWithException(exception: Throwable): Future[T] = complete(Left(exception))
/**
* Completes this Future with the specified other Future, when that Future is completed,
2011-03-30 21:19:26 +02:00
* unless this Future has already been completed.
* @return this.
*/
final def completeWith(other: Future[T]): Future[T] = {
other onComplete { f complete(f.value.get) }
this
}
final def <<(value: T): Future[T] @cps[Future[Any]] = shift { cont: (Future[T] Future[Any]) cont(complete(Right(value))) }
final def <<(other: Future[T]): Future[T] @cps[Future[Any]] = shift { cont: (Future[T] Future[Any])
val fr = new DefaultPromise[Any](Actor.TIMEOUT)
this completeWith other onComplete { f
try {
fr completeWith cont(f)
} catch {
case e: Exception
EventHandler.error(e, this, e.getMessage)
fr completeWithException e
}
}
fr
}
}
/**
2011-03-30 21:19:26 +02:00
* The default concrete Future implementation.
*/
class DefaultPromise[T](timeout: Long, timeunit: TimeUnit) extends Promise[T] {
2010-10-31 19:27:55 +01:00
def this() = this(0, MILLIS)
def this(timeout: Long) = this(timeout, MILLIS)
val timeoutInNanos = timeunit.toNanos(timeout)
private val _startTimeInNanos = currentTimeInNanos
private val _lock = new ReentrantLock
private val _signal = _lock.newCondition
private var _value: Option[Either[Throwable, T]] = None
private var _listeners: List[Future[T] Unit] = Nil
2011-04-26 17:19:07 +02:00
/**
* Must be called inside _lock.lock<->_lock.unlock
*/
@tailrec
private def awaitUnsafe(waitTimeNanos: Long): Boolean = {
if (_value.isEmpty && waitTimeNanos > 0) {
val start = currentTimeInNanos
val remainingNanos = try {
_signal.awaitNanos(waitTimeNanos)
} catch {
case e: InterruptedException
waitTimeNanos - (currentTimeInNanos - start)
}
awaitUnsafe(remainingNanos)
} else {
_value.isDefined
}
}
def await(atMost: Duration) = {
2011-05-18 08:37:58 +02:00
_lock.lock()
if (try { awaitUnsafe(atMost.toNanos min timeLeft()) } finally { _lock.unlock }) this
else throw new FutureTimeoutException("Futures timed out after [" + NANOS.toMillis(timeoutInNanos) + "] milliseconds")
}
2011-02-27 17:42:17 -07:00
def await = {
2011-05-18 08:37:58 +02:00
_lock.lock()
2011-04-26 17:19:07 +02:00
if (try { awaitUnsafe(timeLeft()) } finally { _lock.unlock }) this
else throw new FutureTimeoutException("Futures timed out after [" + NANOS.toMillis(timeoutInNanos) + "] milliseconds")
}
2011-02-27 17:42:17 -07:00
def awaitBlocking = {
_lock.lock
try {
while (_value.isEmpty) {
_signal.await
2011-02-27 17:42:17 -07:00
}
this
} finally {
_lock.unlock
}
}
2011-04-26 17:19:07 +02:00
def isExpired: Boolean = timeLeft() <= 0
def value: Option[Either[Throwable, T]] = {
_lock.lock
try {
_value
} finally {
_lock.unlock
}
}
def complete(value: Either[Throwable, T]): DefaultPromise[T] = {
_lock.lock
val notifyTheseListeners = try {
if (_value.isEmpty && !isExpired) { //Only complete if we aren't expired
_value = Some(value)
val existingListeners = _listeners
_listeners = Nil
existingListeners
} else Nil
} finally {
_signal.signalAll
_lock.unlock
2011-02-27 17:42:17 -07:00
}
2011-04-29 13:22:39 +02:00
if (notifyTheseListeners.nonEmpty) { // Steps to ensure we don't run into a stack-overflow situation
@tailrec
def runCallbacks(rest: List[Future[T] Unit], callbacks: Stack[() Unit]) {
2011-04-29 13:22:39 +02:00
if (rest.nonEmpty) {
notifyCompleted(rest.head)
while (callbacks.nonEmpty) { callbacks.pop().apply() }
runCallbacks(rest.tail, callbacks)
}
2011-04-28 21:14:18 -06:00
}
2011-05-23 11:54:41 +02:00
val pending = Promise.callbacksPendingExecution.get
2011-04-29 13:22:39 +02:00
if (pending.isDefined) { //Instead of nesting the calls to the callbacks (leading to stack overflow)
pending.get.push(() { // Linearize/aggregate callbacks at top level and then execute
2011-04-29 13:22:39 +02:00
val doNotify = notifyCompleted _ //Hoist closure to avoid garbage
notifyTheseListeners foreach doNotify
})
} else {
try {
val callbacks = Stack[() Unit]() // Allocate new aggregator for pending callbacks
2011-05-23 11:54:41 +02:00
Promise.callbacksPendingExecution.set(Some(callbacks)) // Specify the callback aggregator
2011-04-29 13:22:39 +02:00
runCallbacks(notifyTheseListeners, callbacks) // Execute callbacks, if they trigger new callbacks, they are aggregated
2011-05-23 11:54:41 +02:00
} finally { Promise.callbacksPendingExecution.set(None) } // Ensure cleanup
}
}
this
}
def onComplete(func: Future[T] Unit): Promise[T] = {
_lock.lock
val notifyNow = try {
if (_value.isEmpty) {
if (!isExpired) { //Only add the listener if the future isn't expired
_listeners ::= func
false
} else false //Will never run the callback since the future is expired
} else true
} finally {
_lock.unlock
}
2011-04-29 13:22:39 +02:00
if (notifyNow) notifyCompleted(func)
this
}
private def notifyCompleted(func: Future[T] Unit) {
try {
func(this)
} catch {
case e EventHandler notify EventHandler.Error(e, this)
}
}
2010-10-31 19:27:55 +01:00
@inline
private def currentTimeInNanos: Long = MILLIS.toNanos(System.currentTimeMillis)
@inline
private def timeLeft(): Long = timeoutInNanos - (currentTimeInNanos - _startTimeInNanos)
}
/**
* An already completed Future is seeded with it's result at creation, is useful for when you are participating in
* a Future-composition but you already have a value to contribute.
*/
sealed class KeptPromise[T](suppliedValue: Either[Throwable, T]) extends Promise[T] {
val value = Some(suppliedValue)
def complete(value: Either[Throwable, T]): Promise[T] = this
def onComplete(func: Future[T] Unit): Future[T] = { func(this); this }
def await(atMost: Duration): Future[T] = this
def await: Future[T] = this
def awaitBlocking: Future[T] = this
def isExpired: Boolean = true
def timeoutInNanos: Long = 0
}