Question

我需要使用Akka的HTTP客户端（v2.0.2）来使用REST服务。逻辑方法是通过主机连接池执行此操作，因为我们期望大量的同时连接。此Flow消耗(HttpRequest, T)并返回(Try[HttpResponse, T)。 documentation表示需要一些任意类型T来管理对请求的潜在无序响应，但没有指出调用者应该对返回的T做什么。

我的第一次尝试是使用Int作为T的下面的功能。从许多地方调用它以确保连接使用单个池。

val pool = Http().cachedHostConnectionPool[Int]("127.0.0.1", 8888, ConnectionPoolSettings(system))

def pooledRequest(req: HttpRequest): Future[HttpResponse] = {
  val unique = Random.nextInt
  Source.single(req → unique).via(pool).runWith(Sink.head).flatMap {
    case (Success(r: HttpResponse), `unique`) ⇒ Future.successful(r)
    case (Failure(f), `unique`) ⇒ Future.failed(f)
    case (_, i) ⇒ Future.failed(new Exception("Return does not match the request"))
  }
}

问题是客户应该如何使用此T？有更清洁更有效的解决方案吗？最后，我的偏执是否有可能无法实现偏执？

Answer 1

我最初对自己有点困惑，直到我几次阅读文档。如果您要在池中使用单个请求，则无论共享同一个池的位置有多少，您提供的T（在您的情况下为Int）都无关紧要。因此，如果您一直使用Source.single，那么如果您真的需要，该密钥可以始终为1。

虽然它确实起作用，但是如果一段代码将使用池并一次向池中提交多个请求并且想要来自所有这些请求的响应。原因是响应按照从被调用的服务接收的顺序返回，而不是它们被提供给池的顺序。每个请求可能需要不同的时间，因此它们按照从池中收回的顺序向下游流向Sink。

假设我们有一项服务接受了GET个请求，其中包含以下格式的网址：

/product/123

123部分是您要查找的产品的ID。如果我想一次查找产品1-10，并且每个产品都有单独的请求，那么这就是标识符变得重要的地方，这样我就可以将每个HttpResponse与其所针对的产品ID相关联。此方案的简化代码示例如下：

val requests = for(id <- 1 until 10) yield (HttpRequest(HttpMethods.GET, s"/product/$id"), id)
val responsesMapFut:Future[Map[Int,HttpResponse]] = 
  Source(requests).
    via(pool).
    runFold(Map.empty[Int,HttpResponse]){
      case (m, (util.Success(resp), id)) => 
        m ++ Map(id -> resp)

      case (m, (util.Failure(ex), i)) =>
        //Log a failure here probably
          m
    }

当我在fold中收到回复时，我也方便地获得每个回复的ID，以便我可以将它们添加到由id键入的Map。如果没有这个功能，我可能不得不做一些事情，比如解析正文（如果它是json），试图找出哪个响应是哪个，哪个不理想，这不包括失败的情况。在这个解决方案中，我知道哪些请求失败了，因为我仍然得到了标识符。

我希望能为你澄清一些事情。

Answer 2

Akka HTTP连接池在使用基于HTTP的资源时是强大的盟友。如果您要一次执行单个请求，则解决方案是：

def exec(req: HttpRequest): Future[HttpResponse] = {
  Source.single(req → 1)
    .via(pool)
    .runWith(Sink.head).flatMap {
      case (Success(r: HttpResponse), _) ⇒ Future.successful(r)
      case (Failure(f), _) ⇒ Future.failed(f)
    }
}

由于您正在执行single请求，因此无需消除响应歧义。然而，Akka流是聪明的。您可以同时向池提交多个请求。在这种情况下，我们传入Iterable[HttpRequest]。返回的Iterable[HttpResponse]使用SortedMap重新排序，与原始请求的顺序相同。您可以执行request zip response排序：

def exec(requests: Iterable[HttpRequest]): Future[Iterable[Future[HttpResponse]]] = {
  Source(requests.zipWithIndex.toMap)
    .via(pool)
    .runFold(SortedMap[Int, Future[HttpResponse]]()) {
      case (m, (Success(r), idx)) ⇒ m + (idx → Future.successful(r))
      case (m, (Failure(e), idx)) ⇒ m + (idx → Future.failed(e))
    }.map(r ⇒ r.values)
}

如果您需要以自己的方式解开东西，可迭代期货的期货很棒。通过扁平化事物可以获得更简单的响应。

def execFlatten(requests: Iterable[HttpRequest]): Future[Iterable[HttpResponse]] = {
  Source(requests.zipWithIndex.toMap)
    .via(pool)
    .runFold(SortedMap[Int, Future[HttpResponse]]()) {
      case (m, (Success(r), idx)) ⇒ m + (idx → Future.successful(r))
      case (m, (Failure(e), idx)) ⇒ m + (idx → Future.failed(e))
    }.flatMap(r ⇒ Future.sequence(r.values))
}

我已经使用所有导入和包装器this gist来建立一个客户端来使用HTTP服务。

特别感谢@cmbaxter的简洁例子。

Answer 3

有一个改进akka-http文档的公开票据。请 check this example

val pool = Http().cachedHostConnectionPool[Promise[HttpResponse]](host = "google.com", port = 80)
val queue = Source.queue[(HttpRequest, Promise[HttpResponse])](10, OverflowStrategy.dropNew)
  .via(pool)
  .toMat(Sink.foreach({
     case ((Success(resp), p)) => p.success(resp)
    case ((Failure(e), p)) => p.failure(e)
  }))(Keep.left)
  .run


val promise = Promise[HttpResponse]
val request = HttpRequest(uri = "/") -> promise

val response = queue.offer(request).flatMap(buffered => {
  if (buffered) promise.future
  else Future.failed(new RuntimeException())
})

正确使用Akka http客户端连接池

3 个答案: