Маршал akka `HttpResponse` как аргонавт `Json`

Я пытаюсь маршалировать akka HttpResponse как таковой:

{
  "code": 200,
  "headers": [],
  "body": "{\"data\": \"Yes!\"}"
}

Если я напишу Argonaut EncodeJson для этого экземпляра, это может выглядеть так:

implicit def httpResponseEncodeJson: EncodeJson[HttpResponse] =
  EncodeJson(
    (res: HttpResponse) ⇒ {
      ("code" := res._1.value) ->:
      ("headers" := res._2.toList) ->:
      ("body" := res._3) ->: jEmptyObject
    }
  )

Мне удалось маршалировать заголовки как json. Проблема только с корпусом, т.е. ResponseEntity. Поскольку это поток akka, он может возвращать будущее только в том случае, если я использую .toStrict.

Может ли кто-нибудь указать мне, как я могу его маршалировать?


person Yash Srivastav    schedule 04.12.2016    source источник


Ответы (2)


Если возможно, я бы сохранил упорядоченное значение как Future, чтобы сохранить асинхронность извлечения сущности.

Я бы начал с чего-то вроде

  case class StrictHttpResponse(code: String, headers: List[HttpHeader], body: String)

  def toStrictResponse(response: HttpResponse): Future[StrictHttpResponse] = response.entity.dataBytes.runFold(ByteString(""))(_ ++ _).map { bs =>
    StrictHttpResponse(response.status.value, response.headers.toList, bs.utf8String)
  }

  implicit def httpResponseEncodeJson: EncodeJson[StrictHttpResponse] =
    EncodeJson(
      (res: StrictHttpResponse) ⇒ {
        ("code" := res.code) ->:
          ("headers" := res.headers) ->:
          ("body" := res.body) ->: jEmptyObject
      }
    )

  def encodeResponse(response: HttpResponse): Future[Json] = toStrictResponse(response).map(_.jencode)

а потом - напр. - обработать результат encodeResponse, предоставив обратный вызов.

person Stefano Bonetti    schedule 19.01.2017
comment
Большое спасибо за ответ. Я думаю, что я сделал почти то же самое. Можете ли вы просмотреть мой ответ? - person Yash Srivastav; 20.01.2017

В конечном итоге я использовал это:

  implicit def httpResponseListMarshal: ToEntityMarshaller[List[HttpResponse]] =
Marshaller { implicit ec ⇒ (responses: List[HttpResponse]) ⇒

    // Sink for folding Source of ByteString into 1 single huge ByteString
    val sink = Sink.fold[ByteString, ByteString](ByteString.empty)(_ ++ _)

    // A List of Future Json obtained by folding Source[ByteString]
    // and mapping appropriately
    val listFuture: List[Future[Json]] = for {
      res ← responses
    } yield for {
      byteString ← res._3.dataBytes runWith sink
      string = byteString.utf8String
    } yield ("code" := res._1.intValue) ->:
      ("headers" := res._2.toList) ->:
      ("body" := string) ->: jEmptyObject


    // Convert List[Future[Json]] to Future[List[Json]]
    val futureList: Future[List[Json]] = Future.sequence(listFuture)

    // ToEntityMarshaller is essentially a   Future[List[Marshalling[RequestEntity]]]
    for {
      list ← futureList
      json = jArray(list).nospaces
    } yield List(
      Marshalling.Opaque[RequestEntity](() ⇒
        HttpEntity(`application/json`, json)
    ).asInstanceOf[Marshalling[RequestEntity]]
  )
}

Полный код и пример использования можно найти здесь: https://github.com/yashsriv/akka-http-batch-api/blob/argonaut/src/main/scala/org.yashsriv/json/Batch.scala

person Yash Srivastav    schedule 20.01.2017