org.scalarelational.mapper.MappedTable.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of scalarelational-mapper_2.12 Show documentation

scalarelational-mapper

The newest version!

package org.scalarelational.mapper

import org.scalarelational.column.property.PrimaryKey
import org.scalarelational.column.{Column, ColumnLike, ColumnValue, RefColumn}
import org.scalarelational.datatype._
import org.scalarelational.instruction.{InsertSingle, Query, Update}
import org.scalarelational.model.Datastore
import org.scalarelational.table.Table
import org.scalarelational.table.property.TableProperty
import org.scalarelational.{SelectExpression, Session}

abstract class MappedTable[MappedType](name: String, tableProperties: TableProperty*)
                                      (implicit val ds: Datastore)
  extends Table(name, tableProperties: _*)(ds) {
  def ref: ColumnLike[Ref[MappedType], Int] = RefColumn[MappedType](primaryKey.asInstanceOf[ColumnLike[MappedType, Int]])

  def query: Query[Vector[SelectExpression[_]], MappedType]

  def by[T, S](column: Column[T, S], value: T)
           (implicit manifest: Manifest[MappedType], session: Session): Option[MappedType] = {
    val q = query where column === value
    q.converted.headOption
  }

  private[scalarelational] def updateColumnValues(values: List[ColumnValue[Any, Any]]): Update[Ref[MappedType]] = {
    val primaryKey = values.find(_.column.has(PrimaryKey))
      .getOrElse(throw new RuntimeException("Update must have a PrimaryKey value specified to be able to update."))
    val primaryColumn = primaryKey.column
    val update = Update[Ref[MappedType]](this, values, None.orNull, Ref[MappedType])
    update where primaryColumn === primaryKey.value
  }

  private[scalarelational] def insertColumnValues(values: List[ColumnValue[Any, Any]]): InsertSingle[Ref[MappedType]] =
    InsertSingle[Ref[MappedType]](this, values, Ref[MappedType])
}