aboutsummaryrefslogtreecommitdiff
path: root/server/src/LiveMessages.scala
blob: 307dd819972011930197b264066be0bc2e49cf6f (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
package triad

import akka.NotUsed
import akka.stream.scaladsl.{
  BroadcastHub,
  Keep,
  Source,
  SourceQueueWithComplete
}
import akka.stream.{Materializer, OverflowStrategy, QueueOfferResult}
import scala.concurrent.Future

/** A very basic streaming message router. */
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()

  /** Push a single message into the stream. */
  def push(message: Message): Future[QueueOfferResult] = in.offer(message)

  /** Obtain a stream of the message feed. */
  def feed: Source[Message, NotUsed] = out

}