• Home
• com.fiftyonezero.eel
• eel-core_2.10
• 0.11.0
• source code
• domain.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.fiftyonezero.eel.domain.scala Maven / Gradle / Ivy

package com.fiftyonezero.eel

import scala.language.implicitConversions

object Field {
  implicit def toField(str: String): Field = Field(str)
}

case class Field(value: String)

case class Column(name: String, `type`: SchemaType, nullable: Boolean)

sealed trait SchemaType
object SchemaType {
  case object Int extends SchemaType
  case object Long extends SchemaType
  case object String extends SchemaType
  case object BigInt extends SchemaType
  case object Boolean extends SchemaType
  case object Double extends SchemaType
  case object Float extends SchemaType
  case object Decimal extends SchemaType
  case object Date extends SchemaType
}

object Column {
  implicit def apply(str: String): Column = Column(str, SchemaType.String, false)
}

object Row {
  def apply(map: Map[String, String]): Row = {
    Row(map.keys.map(Column.apply).toSeq, map.values.seq.map(Field.apply).toSeq)
  }
}

case class Row(columns: Seq[Column], fields: Seq[Field]) {

  require(columns.size == fields.size, "Columns and fields should have the same size")

  def apply(name: String): String = {
    val pos = columns.indexWhere(_.name == name)
    fields(pos).value
  }

  def join(other: Row): Row = Row(columns ++ other.columns, fields ++ other.fields)

  def size: Int = columns.size

  def addColumn(name: String, value: String): Row = {
    copy(columns = columns :+ Column(name), fields = fields :+ Field(value))
  }

  def removeColumn(name: String): Row = Row(toMap - name)

  def toMap: Map[String, String] = columns.map(_.name).zip(fields.map(_.value)).toMap
}