com.algorithmia.handler.ResponseHandler.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of algorithmia-scala_2.13 Show documentation

algorithmia-scala

The newest version!

package com.algorithmia.handler

import java.io.{FileOutputStream, PrintStream}

import play.api.libs.json.{Json, Writes}

import scala.util.Try

case class ResponseHandler[O]() {
  val FIFOPATH = "/tmp/algoout"

  private def write(data: String): Try[Unit] = {
    Try(new PrintStream(new FileOutputStream(this.FIFOPATH, true)))
      .map({ s =>
        s.println(data)
        s.flush()
        s.close()
      })
  }

  def writeErrorToPipe(e: Throwable): Try[Unit] = {
    val serializable = SerializableException.fromException(e)
    val serialized = Json.toJson(serializable)
    write(serialized.toString())
  }

  def writeResponseToPipe(output: O)(implicit writer: Writes[O]): Try[Unit] = {
    val metadata = output match {
      case _: String => "text"
      case _ => "json"
    }
    val response = Response(Metadata(metadata), output)
    val serialized = Json.toJson(response)
    write(serialized.toString())
  }
}