server/src/main/scala/LiveMessages.scala


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

package triad

import akka.NotUsed
import akka.stream.scaladsl.{
  BroadcastHub,
  Keep,
  Source,
  SourceQueueWithComplete
}
import akka.stream.{Materializer, OverflowStrategy}

class LiveMessages(implicit materializer: Materializer) {

  private val (in: SourceQueueWithComplete[Message],
               out: Source[Message, NotUsed]) = Source
    .queue[Message](10, OverflowStrategy.dropTail)
    .toMat(BroadcastHub.sink[Message])(Keep.both)
    .run()

  def push(message: Message) = in.offer(message)
  def feed = out

}