com.algorithmia.data.DataFile.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.data

import com.algorithmia._
import java.io.{File, FileInputStream, InputStream}
import java.nio.file.{Files, StandardCopyOption}

class DataFile(client: AlgorithmiaClient, dataUrl: String) extends DataObject(client, dataUrl, DataFileType) {

  override def exists: Boolean = {
    client.http.head(url).code == 200
  }

  def getType: DataObjectType = DataFileType

  def getFile: File = {
    val response = client.http.getInputStream(url) { is =>
      // Create temp directory
      val dir = Files.createTempDirectory("algocache").toFile
      // Create file with same name as this object in temp dir
      val file = new File(dir, getName)
      // Copy inputstream to file
      Files.copy(is, file.toPath, StandardCopyOption.REPLACE_EXISTING)
      file
    }
    response.body
  }

  def getString: String = {
    client.http.get(url).body
  }

  def getBytes: Array[Byte] = {
    client.http.getBytes(url).body
  }

  def put(data: String): Unit = {
    client.http.put(url, data.getBytes)
  }

  def put(file: File): Unit = {
    val is = new FileInputStream(file)
    client.http.put(url, is, Some(file.length))
  }

  def put(is: InputStream): Unit = {
    client.http.put(url, is, None)
  }

  def delete(): Unit = {
    client.http.delete(url)
  }

}