com.algorithmia.algo.Algorithm.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.algo

import com.algorithmia.{AlgorithmiaConf, AlgorithmiaClient}
import play.api.libs.json._
import scala.concurrent.duration.{Duration, FiniteDuration, SECONDS}

class Algorithm(client: AlgorithmiaClient, algoUrl: String, val options: Map[String, String] = Map.empty) {
  private val trimmedPath: String = algoUrl.replaceAll("^algo://|^/", "")
  val url: String = AlgorithmiaConf.apiAddress + "/v1/algo/" + trimmedPath

  def pipe[T](input: T)(implicit reads: Writes[T]): AlgoResponse = {
    val inputJson: JsValue = Json.toJson(input)
    val httpResponse = client.http.post(url, inputJson.toString)
    val rawOutput = httpResponse.body
    httpResponse.code match {
      case 200 => AlgoResponse(rawOutput, outputType)
      case _ => AlgoFailure(rawOutput, Metadata(0, ContentTypeVoid, None), rawOutput)
    }
  }

  /**
    * Set algorithm options, to be passed into algorithmia as query parameters.
    * We use "with" instead of "set" because the Algorithm object is immutable.
    */
  def withOptions(opts: (String, String)*): Algorithm = {
    new Algorithm(client, algoUrl, options ++ opts)
  }

  def timeout: Option[FiniteDuration] = {
    options.get("timeout").map(sec => Duration(sec.toLong, SECONDS))
  }
  def withTimeout(timeout: FiniteDuration): Algorithm = {
    withOptions("timeout" -> timeout.toSeconds.toString)
  }

  def outputType: AlgorithmOutputType = {
    options.get("output").map(AlgorithmOutputType.fromString).getOrElse(OutputDefault)
  }
  def withOutputType(outputType: AlgorithmOutputType): Algorithm = {
    withOptions("output" -> outputType.parameter)
  }

}