From 06e2ae994af39f77d8ee2b7ed367cba0936f35c4 Mon Sep 17 00:00:00 2001 From: adamw Date: Sat, 15 Jul 2017 21:16:28 +0200 Subject: Response as parameters --- .../com/softwaremill/sttp/akkahttp/AkkaHttpSttpHandler.scala | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'akka-http-handler') diff --git a/akka-http-handler/src/main/scala/com/softwaremill/sttp/akkahttp/AkkaHttpSttpHandler.scala b/akka-http-handler/src/main/scala/com/softwaremill/sttp/akkahttp/AkkaHttpSttpHandler.scala index a22c7cf..866aaad 100644 --- a/akka-http-handler/src/main/scala/com/softwaremill/sttp/akkahttp/AkkaHttpSttpHandler.scala +++ b/akka-http-handler/src/main/scala/com/softwaremill/sttp/akkahttp/AkkaHttpSttpHandler.scala @@ -58,17 +58,23 @@ class AkkaHttpSttpHandler(actorSystem: ActorSystem) .runFold(ByteString(""))(_ ++ _) .map(_.toArray[Byte]) + def asString(enc: String) = + asByteArray.map(new String(_, enc)) + rr match { case IgnoreResponse => hr.discardEntityBytes() Future.successful(()) case ResponseAsString(enc) => - asByteArray.map(new String(_, enc)) + asString(enc) case ResponseAsByteArray => asByteArray + case r @ ResponseAsParams(enc) => + asString(enc).map(r.parse) + case r @ ResponseAsStream() => Future.successful(r.responseIsStream(hr.entity.dataBytes)) } -- cgit v1.2.3