spray-chunked-test.scala

package ru.metahouse.metabus

import akka.actor._
import akka.io.IO
import spray.can.Http
import spray.routing.HttpServiceActor
import spray.http._
import spray.http.MediaTypes._
import scala.concurrent.ExecutionContext.Implicits.global
import spray.http.ChunkedResponseStart
import spray.http.HttpResponse
import scala.concurrent.duration._

object Program {
  def main(args: Array[String]): Unit = {
    implicit val system = ActorSystem("test")
    val frontend = system.actorOf(Props[HttpService], name = "frontend")
    IO(Http) ! Http.Bind(frontend, interface = "127.0.0.1", port = 8282)
  }
}

class HttpService extends HttpServiceActor with ActorLogging {
  def receive = runRoute(
    path("test") {
      get {
        request =>
          context.actorOf(Props(classOf[TestStream], request.responder))
      }
    }
  )
}

class TestStream(responder: ActorRef) extends Actor with ActorLogging{
  private case class Timer(n: Int)
  private val header = (1 to 1024).map(_ => "\uFEFF").mkString("")

  self ! Timer(10)
  responder ! ChunkedResponseStart(HttpResponse(entity = HttpEntity(ContentType(`text/plain`, HttpCharsets.`UTF-8`), header)))

  def receive = {
    case Timer(0) =>
      responder ! ChunkedMessageEnd()
      context.stop(self)

    case Timer(n) =>
      responder !  MessageChunk(n.toString + "\n")
      context.system.scheduler.scheduleOnce(1 second, self, Timer(n - 1))

    case ev: Http.ConnectionClosed =>
      log.debug("connection closed")
      context.stop(self)
  }
}