54 lines
1.5 KiB
Scala
54 lines
1.5 KiB
Scala
package docs.stream.cookbook
|
|
|
|
import akka.stream.OverflowStrategy
|
|
import akka.stream.scaladsl._
|
|
import akka.stream.testkit.StreamTestKit.SubscriberProbe
|
|
|
|
import scala.collection.immutable
|
|
import scala.concurrent.Await
|
|
import scala.concurrent.duration._
|
|
|
|
class RecipeDroppyBroadcast extends RecipeSpec {
|
|
|
|
"Recipe for a droppy broadcast" must {
|
|
"work" in {
|
|
val myElements = Source(immutable.Iterable.tabulate(100)(_ + 1))
|
|
|
|
val sub1 = SubscriberProbe[Int]()
|
|
val sub2 = SubscriberProbe[Int]()
|
|
val futureSink = Sink.head[Seq[Int]]
|
|
val mySink1 = Sink(sub1)
|
|
val mySink2 = Sink(sub2)
|
|
val mySink3 = Flow[Int].grouped(200).toMat(futureSink)(Keep.right)
|
|
|
|
//#droppy-bcast
|
|
val graph = FlowGraph.closed(mySink1, mySink2, mySink3)((_, _, _)) { implicit b =>
|
|
(sink1, sink2, sink3) =>
|
|
import FlowGraph.Implicits._
|
|
|
|
val bcast = b.add(Broadcast[Int](3))
|
|
myElements ~> bcast
|
|
|
|
bcast.buffer(10, OverflowStrategy.dropHead) ~> sink1
|
|
bcast.buffer(10, OverflowStrategy.dropHead) ~> sink2
|
|
bcast.buffer(10, OverflowStrategy.dropHead) ~> sink3
|
|
}
|
|
//#droppy-bcast
|
|
|
|
Await.result(graph.run()._3, 3.seconds).sum should be(5050)
|
|
|
|
sub1.expectSubscription().request(10)
|
|
sub2.expectSubscription().request(10)
|
|
|
|
for (i <- 91 to 100) {
|
|
sub1.expectNext(i)
|
|
sub2.expectNext(i)
|
|
}
|
|
|
|
sub1.expectComplete()
|
|
sub2.expectComplete()
|
|
|
|
}
|
|
}
|
|
|
|
}
|