im.actor.server.persist.FilePartRepo.scala Maven / Gradle / Ivy

Go to download
package im.actor.server.persist

import slick.dbio.Effect.{ Read, Write }
import slick.driver.PostgresDriver.api._
import slick.profile.{ FixedSqlStreamingAction, FixedSqlAction }

import im.actor.server.model

final class FilePartTable(tag: Tag) extends Table[model.FilePart](tag, "file_parts") {
  def fileId = column[Long]("file_id", O.PrimaryKey)

  def number = column[Int]("number", O.PrimaryKey)

  def size = column[Int]("size")

  def uploadKey = column[String]("upload_key")

  def * = (fileId, number, size, uploadKey) <> (model.FilePart.tupled, model.FilePart.unapply)
}

object FilePartRepo {
  val parts = TableQuery[FilePartTable]

  def createOrUpdate(fileId: Long, number: Int, size: Int, uploadKey: String): FixedSqlAction[Int, NoStream, Write] =
    parts.insertOrUpdate(model.FilePart(fileId, number, size, uploadKey))

  def findByFileId(fileId: Long): FixedSqlStreamingAction[Seq[model.FilePart], model.FilePart, Read] =
    parts.filter(_.fileId === fileId).sortBy(_.number).result
}