sttp.tapir.server.pekkohttp.PekkoServerSentEvents.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tapir-pekko-http-server_2.12 Show documentation

tapir-pekko-http-server

The newest version!

package sttp.tapir.server.pekkohttp

import org.apache.pekko.NotUsed
import org.apache.pekko.stream.scaladsl.{Flow, Framing, Source}
import org.apache.pekko.util.ByteString
import sttp.capabilities.pekko.PekkoStreams
import sttp.model.sse.ServerSentEvent

object PekkoServerSentEvents {

  def serialiseSSEToBytes: Source[ServerSentEvent, Any] => PekkoStreams.BinaryStream = sseStream =>
    sseStream.map(sse => {
      ByteString(s"${sse.toString()}\n\n")
    })

  def parseBytesToSSE: PekkoStreams.BinaryStream => Source[ServerSentEvent, Any] = stream => stream.via(parse)

  private val parse: Flow[ByteString, ServerSentEvent, NotUsed] =
    Framing
      .delimiter(ByteString("\n\n"), maximumFrameLength = Int.MaxValue, allowTruncation = true)
      .map(_.utf8String)
      .map(_.split("\n").toList)
      .map(ServerSentEvent.parse)

}