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

#17717 handle OnComplete in ActorSubscriber doc samples

Browse source
This commit is contained in:
Martynas Mickevičius 2016-10-17 12:07:40 +03:00 • committed by Johan Andrén
parent 0e60020c58
commit 1e85e1720c
2 changed files with 43 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

12
akka-docs/rst/scala/code/docs/stream/ActorSubscriberDocSpec.scala
View file

@ -16,6 +16,7 @@ import akka.stream.actor.MaxInFlightRequestStrategy
import akka.stream.scaladsl.Sink
import akka.stream.scaladsl.Source
import akka.testkit.AkkaSpec
import scala.concurrent.duration._
object ActorSubscriberDocSpec {
//#worker-pool
@ -54,6 +55,13 @@ object ActorSubscriberDocSpec {
case Reply(id) =>
queue(id) ! Done(id)
queue -= id
if (canceled && queue.isEmpty) {
context.stop(self)
}
case OnComplete =>
if (queue.isEmpty) {
context.stop(self)
}
}
}
@ -79,11 +87,13 @@ class ActorSubscriberDocSpec extends AkkaSpec {
//#actor-subscriber-usage
val N = 117
Source(1 to N).map(WorkerPool.Msg(_, replyTo))
val worker = Source(1 to N).map(WorkerPool.Msg(_, replyTo))
.runWith(Sink.actorSubscriber(WorkerPool.props))
//#actor-subscriber-usage
watch(worker)
receiveN(N).toSet should be((1 to N).map(WorkerPool.Done).toSet)
expectTerminated(worker, 10.seconds)
}
}