raboof/pekko
1
0
Fork 0
Code Issues Pull requests Projects Releases 6 Packages Wiki Activity Actions 10

=htc #16055 refactoring: make headAndTail a simple flow

Browse source
This commit is contained in:
Johannes Rudolph 2015-04-28 16:14:42 +02:00
parent 0f15cb6d78
commit 444a0deef5
6 changed files with 9 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

2
akka-http-core/src/main/scala/akka/http/impl/engine/client/OutgoingConnectionBlueprint.scala
View file

@ -72,7 +72,7 @@ private[http] object OutgoingConnectionBlueprint {
.transform(StreamUtils.recover { case x: ResponseParsingError ⇒ x.error :: Nil }) // FIXME after #16565
.mapConcat(identityFunc)
.splitWhen(x ⇒ x.isInstanceOf[MessageStart] || x == MessageEnd)
.headAndTail
.via(headAndTailFlow)
.collect {
case (ResponseStart(statusCode, protocol, headers, createEntity, _), entityParts) ⇒
HttpResponse(statusCode, headers, createEntity(entityParts), protocol)

2
akka-http-core/src/main/scala/akka/http/impl/engine/server/HttpServerBluePrint.scala
View file

@ -62,7 +62,7 @@ private[http] object HttpServerBluePrint {
val requestPreparation =
Flow[RequestOutput]
.splitWhen(x ⇒ x.isInstanceOf[MessageStart] || x == MessageEnd)
.headAndTail
.via(headAndTailFlow)
.map {
case (RequestStart(method, uri, protocol, headers, createEntity, _, _), entityParts) ⇒
val effectiveUri = HttpRequest.effectiveUri(uri, headers, securedConnection = false, defaultHostHeader)

2
akka-http-core/src/main/scala/akka/http/impl/engine/ws/Websocket.scala
View file

@ -47,7 +47,7 @@ private[http] object Websocket {
/** Collects user-level API messages from MessageDataParts */
val collectMessage: Flow[Source[MessageDataPart, Unit], Message, Unit] =
Flow[Source[MessageDataPart, Unit]]
.headAndTail
.via(headAndTailFlow)
.map {
case (TextMessagePart(text, true), remaining) ⇒
TextMessage.Strict(text)

19
akka-http-core/src/main/scala/akka/http/impl/util/package.scala
View file

@ -43,25 +43,14 @@ package object util {
private[http] implicit def enhanceByteStrings[Mat](byteStrings: Source[ByteString, Mat]): EnhancedByteStringSource[Mat] =
new EnhancedByteStringSource(byteStrings)
private[http] implicit class SourceWithHeadAndTail[T, Mat](val underlying: Source[Source[T, Any], Mat]) extends AnyVal {
def headAndTail: Source[(T, Source[T, Unit]), Mat] =
underlying.map {
private[http] def headAndTailFlow[T]: Flow[Source[T, Any], (T, Source[T, Unit]), Unit] =
Flow[Source[T, Any]]
.map {
_.prefixAndTail(1)
.filter(_._1.nonEmpty)
.map { case (prefix, tail) ⇒ (prefix.head, tail) }
}
.flatten(FlattenStrategy.concat)
}
private[http] implicit class FlowWithHeadAndTail[In, Out, Mat](val underlying: Flow[In, Source[Out, Any], Mat]) extends AnyVal {
def headAndTail: Flow[In, (Out, Source[Out, Unit]), Mat] =
underlying.map {
_.prefixAndTail(1)
.filter(_._1.nonEmpty)
.map { case (prefix, tail) ⇒ (prefix.head, tail) }
}
.flatten(FlattenStrategy.concat)
}
.flatten(FlattenStrategy.concat)
private[http] def printEvent[T](marker: String): Flow[T, T, Unit] =
Flow[T].transform(() ⇒ new PushPullStage[T, T] {

2
akka-http-core/src/test/scala/akka/http/impl/engine/parsing/RequestParserSpec.scala
View file

@ -453,7 +453,7 @@ class RequestParserSpec extends FreeSpec with Matchers with BeforeAndAfterAll {
.map(ByteString.apply)
.transform(() ⇒ parser.stage).named("parser")
.splitWhen(x ⇒ x.isInstanceOf[MessageStart] || x.isInstanceOf[EntityStreamError])
.headAndTail
.via(headAndTailFlow)
.collect {
case (RequestStart(method, uri, protocol, headers, createEntity, _, close), entityParts) ⇒
closeAfterResponseCompletion :+= close

2
akka-http-core/src/test/scala/akka/http/impl/engine/parsing/ResponseParserSpec.scala
View file

@ -264,7 +264,7 @@ class ResponseParserSpec extends FreeSpec with Matchers with BeforeAndAfterAll {
.map(ByteString.apply)
.transform(() ⇒ newParserStage(requestMethod)).named("parser")
.splitWhen(x ⇒ x.isInstanceOf[MessageStart] || x.isInstanceOf[EntityStreamError])
.headAndTail
.via(headAndTailFlow)
.collect {
case (ResponseStart(statusCode, protocol, headers, createEntity, close), entityParts) ⇒
closeAfterResponseCompletion :+= close