使用 Akka http 处理 httpResponse 404 上的重试逻辑

Question

我正在尝试实现一个向外部 REST 端点发出请求的代码，当该端点返回 404 时，它应该重试有限的时间。

HttpRequest 类似于

val responseFuture = Http().singleRequest(HttpRequest(method = requestMethod,
    uri = url,
    entity = HttpEntity(requestBody).withContentType(ContentTypes.`application/json`)
))

并且响应处理为

responseFuture.onComplete {
    case Success(r) =>
      if (r.status.isFailure()) Future.failed(new Exception("request failed with status 404"))
      else r
    case Failure(e) => throw e
}

我的重试逻辑是：

def retryFuture[T](retries: Int, delay: FiniteDuration = 1.second)(fn: => Future[T])(implicit ec: ExecutionContext, s: Scheduler): Future[T] = {
    fn.recoverWith {
      case _ if retries > 0 => after(delay, s)(retryFuture(retries - 1, delay)(fn))
    }
}

问题是当端点返回 404 时，它以 SUCCESS(HttpResponse(404,...( 的形式出现，因此重试不起作用。谁能指出可以做些什么来解决这个问题？

Answer 1

您应该将responseFuture定义为def而不是val，以便能够多次运行它。

然后，您只需要执行responseFuture.filter(_.status.isSuccess())即可对 404 返回代码(实际上是所有错误代码(进行Future.failed。

您现在可以对此使用恢复策略。

Answer 2

onComplete不会

改变未来的成功或失败;事实上，函数的返回值被丢弃。相反，要么在onComplete重试，要么使用 transform 获得新的失败的未来，如果它是 404 然后recoverWith。