fix: Cancel any replays when the filter stage stops

akka · Sep 17, 2024 · 361e8f9 · 361e8f9
1 parent 227047e
commit 361e8f9
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/akka-projection-grpc/src/main/scala/akka/projection/grpc/internal/FilterStage.scala b/akka-projection-grpc/src/main/scala/akka/projection/grpc/internal/FilterStage.scala
@@ -463,6 +463,11 @@ import org.slf4j.LoggerFactory
           pullInEnvOrReplay()
         }
       })
+
+      override def postStop(): Unit = {
+        replayInProgress.values.foreach(_.queue.cancel())
+        replayInProgress = Map.empty
+      }
     }
 
 }