From 8b0b63625ed17599dcb3ff9c83be997c353da8e1 Mon Sep 17 00:00:00 2001 From: Viktor Klang Date: Mon, 7 Dec 2015 12:41:38 +0100 Subject: [PATCH] =str - 19082 - Moves the `allow` var into the Logic of DropWithin --- akka-stream/src/main/scala/akka/stream/impl/fusing/Ops.scala | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/akka-stream/src/main/scala/akka/stream/impl/fusing/Ops.scala b/akka-stream/src/main/scala/akka/stream/impl/fusing/Ops.scala index 9fcff50e7a..2dc1437da3 100644 --- a/akka-stream/src/main/scala/akka/stream/impl/fusing/Ops.scala +++ b/akka-stream/src/main/scala/akka/stream/impl/fusing/Ops.scala @@ -880,9 +880,10 @@ private[stream] class TakeWithin[T](timeout: FiniteDuration) extends SimpleLinea } private[stream] class DropWithin[T](timeout: FiniteDuration) extends SimpleLinearGraphStage[T] { - private var allow = false - override def createLogic(inheritedAttributes: Attributes): GraphStageLogic = new TimerGraphStageLogic(shape) { + + private var allow = false + setHandler(in, new InHandler { override def onPush(): Unit = if (allow) push(out, grab(in))