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

=htc replace printEvent with Flow[T].log #20288 (#20384)

Browse source
This commit is contained in:
Bernard Leach 2016-04-26 01:48:26 +10:00 • committed by Konrad Malawski
parent 0e11ec2057
commit 3af02c3bf9
2 changed files with 1 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -43,30 +43,6 @@ package object util {
private[http] implicit def enhanceByteStringsMat[Mat](byteStrings: Source[ByteString, Mat]): EnhancedByteStringSource[Mat] =
new EnhancedByteStringSource(byteStrings)
private[http] def printEvent[T](marker: String): Flow[T, T, NotUsed] =
Flow[T].transform(() ⇒ new PushPullStage[T, T] {
override def onPush(element: T, ctx: Context[T]): SyncDirective = {
println(s"$marker: $element")
ctx.push(element)
}
override def onPull(ctx: Context[T]): SyncDirective = {
println(s"$marker: PULL")
ctx.pull()
}
override def onUpstreamFailure(cause: Throwable, ctx: Context[T]): TerminationDirective = {
println(s"$marker: Error $cause")
super.onUpstreamFailure(cause, ctx)
}
override def onUpstreamFinish(ctx: Context[T]): TerminationDirective = {
println(s"$marker: Complete")
super.onUpstreamFinish(ctx)
}
override def onDownstreamFinish(ctx: Context[T]): TerminationDirective = {
println(s"$marker: Cancel")
super.onDownstreamFinish(ctx)
}
})
private[this] var eventStreamLogger: ActorRef = _
private[http] def installEventStreamLoggerFor(channel: Class[_])(implicit system: ActorSystem): Unit = {
synchronized {

2
akka-http-core/src/test/scala/akka/http/impl/engine/ws/MessageSpec.scala
View file

@ -975,6 +975,6 @@ class MessageSpec extends FreeSpec with Matchers with WithMaterializerSpec {
val trace = false // set to `true` for debugging purposes
def printEvent[T](marker: String): Flow[T, T, NotUsed] =
if (trace) akka.http.impl.util.printEvent(marker)
if (trace) Flow[T].log(marker)
else Flow[T]
}