Scala——如何通过Await.result使用定时器而不阻塞期货

Question

我有一个由akka-http提供的Rest API。在某些情况下，我需要从外部数据库（Apache HBase）获取数据，如果数据库传递数据的时间太长，我希望查询失败。

一种简单的方法是将调用包装在Future中，然后使用具有所需持续时间的Await.result来阻止它。

import scala.concurrent.duration._
import scala.concurrent.{Await, Future}

object AsyncTest1 extends App {

  val future = Future {
    getMyDataFromDB()
  }

  val myData = Await.result(future, 100.millis)
}

似乎效率低下，因为这个实现需要两个线程。有没有有效的方法来做到这一点？

我还有另一个用例，我想并行发送多个查询，然后聚合结果，具有相同的延迟限制。

val future1 = Future {
    getMyDataFromDB1()
  }

  val future2 = Future {
    getMyDataFromDB2()
  }

  val foldedFuture = Future.fold(
    Seq(future1, future2))(MyAggregatedData)(myAggregateFunction)
  )

  val myData = Await.result(foldedFuture, 100.millis)

同样的问题，最有效的实现方式是什么？

多谢你的帮忙

Answer 1

一种解决方案是使用Akka的after函数，该函数将允许您传递一个持续时间，在该持续时间之后，future会抛出一个异常或任何您想要的异常。

看看这里。它演示了如何实现这一点。

编辑：我想我会在这里发布代码，以防将来链接被破坏：

import scala.concurrent._
import scala.concurrent.duration._
import ExecutionContext.Implicits.global
import scala.util.{Failure, Success}
import akka.actor.ActorSystem
import akka.pattern.after

val system = ActorSystem("theSystem")

lazy val f = future { Thread.sleep(2000); true }
lazy val t = after(duration = 1 second, using = system.scheduler)(Future.failed(new TimeoutException("Future timed out!")))

val fWithTimeout = Future firstCompletedOf Seq(f, t)

fWithTimeout.onComplete {
   case Success(x) => println(x)
   case Failure(error) => println(error)
}