Skip to content

Commit

Permalink
Merge pull request #77 from kazumatsudo/feature/add_data_types_by_jan…
Browse files Browse the repository at this point in the history 
…us_graph

add data types by janus graph
  • Loading branch information
@kazumatsudo
kazumatsudo committed Jan 16, 2024
2 parents 249fc62 + c28b69f commit 3dccb3f
Show file tree
Hide file tree
Showing 4 changed files with 610 additions and 44 deletions.
220 changes: 200 additions & 20 deletions src/main/scala/domain/table/ddl/column/ColumnType.scala
View file
Original file line number Diff line number Diff line change
@@ -1,14 +1,27 @@
package domain.table.ddl.column

import java.util.UUID
import scala.annotation.tailrec

/** referenced by https://docs.janusgraph.org/schema/#property-key-data-type
*/
sealed trait ColumnType {
def toSqlSentence: String
}

case object ColumnTypeBoolean extends ColumnType {
override def toSqlSentence: String = "BOOLEAN"
}

case class ColumnTypeByte(private val length: ColumnLength) extends ColumnType {
override def toSqlSentence: String = s"TINYINT(${length.toSqlSentence})"
}

case class ColumnTypeShort(private val length: ColumnLength)
extends ColumnType {
override def toSqlSentence: String = s"SMALLINT(${length.toSqlSentence})"
}

case class ColumnTypeInt(private val length: ColumnLength) extends ColumnType {
override def toSqlSentence: String = s"INT(${length.toSqlSentence})"
}
Expand All @@ -17,11 +30,26 @@ case class ColumnTypeLong(private val length: ColumnLength) extends ColumnType {
override def toSqlSentence: String = s"INT(${length.toSqlSentence})"
}

case class ColumnTypeFloat(private val length: ColumnLength)
extends ColumnType {
override def toSqlSentence: String = s"FLOAT"
}

case class ColumnTypeDouble(private val length: ColumnLength)
extends ColumnType {
override def toSqlSentence: String = s"DOUBLE"
}

case object ColumnTypeUUID extends ColumnType {
val length: ColumnLength = ColumnLength(36)
override def toSqlSentence: String = s"CHAR(${length.toSqlSentence})"
}

case class ColumnTypeCharacter(private val length: ColumnLength)
extends ColumnType {
override def toSqlSentence: String = s"CHAR(${length.toSqlSentence})"
}

case class ColumnTypeString(private val length: ColumnLength)
extends ColumnType {
override def toSqlSentence: String = s"VARCHAR(${length.toSqlSentence})"
Expand All @@ -34,17 +62,27 @@ case object ColumnTypeUnknown extends ColumnType {
object ColumnType {

def apply(value: Any): ColumnType = value match {
case valueString: String =>
ColumnTypeString(ColumnLength(valueString.length))
case _: Boolean => ColumnTypeBoolean
case valueByte: Byte =>
ColumnTypeByte(ColumnLength(valueByte.toString.length))
case valueShort: Short =>
ColumnTypeShort(ColumnLength(valueShort.toString.length))
case valueInt: Int => ColumnTypeInt(ColumnLength(valueInt.toString.length))
case valueLong: Long =>
ColumnTypeLong(ColumnLength(valueLong.toString.length))
case valueFloat: Float =>
ColumnTypeFloat(
ColumnLength(valueFloat.toString.replaceAll("0*$", "").length)
)
case valueDouble: Double =>
ColumnTypeDouble(
ColumnLength(valueDouble.toString.replaceAll("0*$", "").length)
)
case _: Boolean => ColumnTypeBoolean
case _ => ColumnTypeUnknown // TODO: classify the type in detail
case _: UUID => ColumnTypeUUID
case _: Char => ColumnTypeCharacter(ColumnLength(1))
case valueString: String =>
ColumnTypeString(ColumnLength(valueString.length))
case _ => ColumnTypeUnknown // TODO: classify the type in detail
}

/** merges the attributes (type, length...) in two columns into one
Expand All @@ -61,65 +99,207 @@ object ColumnType {
(a, b) match {
// ColumnTypeBoolean
case (ColumnTypeBoolean, ColumnTypeBoolean) => ColumnTypeBoolean
case (ColumnTypeBoolean, ColumnTypeByte(length)) =>
ColumnTypeByte(length.max(5)) // 5 = false.toString
case (ColumnTypeBoolean, ColumnTypeShort(length)) =>
ColumnTypeShort(length.max(5)) // 5 = false.toString
case (ColumnTypeBoolean, ColumnTypeInt(length)) =>
ColumnTypeInt(length.max(5)) // 5 = false.toString
case (ColumnTypeBoolean, ColumnTypeLong(length)) =>
ColumnTypeLong(length.max(5)) // 5 = false.toString
case (ColumnTypeBoolean, ColumnTypeFloat(length)) =>
ColumnTypeFloat(length.max(5)) // 5 = false.toString
case (ColumnTypeBoolean, ColumnTypeDouble(length)) =>
ColumnTypeDouble(length.max(5)) // 5 = false.toString
case (ColumnTypeBoolean, ColumnTypeUUID) =>
// change UUID -> Character
ColumnTypeCharacter(ColumnTypeUUID.length)
case (ColumnTypeBoolean, ColumnTypeCharacter(length)) =>
ColumnTypeCharacter(length.max(5)) // 5 = false.toString
case (ColumnTypeBoolean, ColumnTypeString(length)) =>
ColumnTypeString(length.max(5)) // 5 = false.toString
case (ColumnTypeBoolean, ColumnTypeUnknown) => ColumnTypeUnknown

// ColumnTypeByte
case (ColumnTypeByte(_), ColumnTypeBoolean) => merge(b, a)
case (ColumnTypeByte(alength), ColumnTypeByte(blength)) =>
ColumnTypeByte(alength.max(blength))
case (ColumnTypeByte(alength), ColumnTypeShort(blength)) =>
ColumnTypeShort(alength.max(blength))
case (ColumnTypeByte(alength), ColumnTypeInt(blength)) =>
ColumnTypeInt(alength.max(blength))
case (ColumnTypeByte(alength), ColumnTypeLong(blength)) =>
ColumnTypeLong(alength.max(blength))
case (ColumnTypeByte(alength), ColumnTypeFloat(blength)) =>
ColumnTypeFloat(alength.max(blength))
case (ColumnTypeByte(alength), ColumnTypeDouble(blength)) =>
ColumnTypeDouble(alength.max(blength))
case (ColumnTypeByte(alength), ColumnTypeUUID) =>
// change UUID -> Character
ColumnTypeCharacter(alength.max(ColumnTypeUUID.length))
case (ColumnTypeByte(alength), ColumnTypeCharacter(blength)) =>
ColumnTypeCharacter(alength.max(blength))
case (ColumnTypeByte(alength), ColumnTypeString(blength)) =>
ColumnTypeString(alength.max(blength))
case (ColumnTypeByte(_), ColumnTypeUnknown) => ColumnTypeUnknown

// ColumnTypeShort
case (ColumnTypeShort(_), ColumnTypeBoolean) => merge(b, a)
case (ColumnTypeShort(_), ColumnTypeByte(_)) => merge(b, a)
case (ColumnTypeShort(alength), ColumnTypeShort(blength)) =>
ColumnTypeShort(alength.max(blength))
case (ColumnTypeShort(alength), ColumnTypeInt(blength)) =>
ColumnTypeInt(alength.max(blength))
case (ColumnTypeShort(alength), ColumnTypeLong(blength)) =>
ColumnTypeLong(alength.max(blength))
case (ColumnTypeShort(alength), ColumnTypeFloat(blength)) =>
ColumnTypeFloat(alength.max(blength))
case (ColumnTypeShort(alength), ColumnTypeDouble(blength)) =>
ColumnTypeDouble(alength.max(blength))
case (ColumnTypeShort(alength), ColumnTypeUUID) =>
// change UUID -> Character
ColumnTypeCharacter(alength.max(ColumnTypeUUID.length))
case (ColumnTypeShort(alength), ColumnTypeCharacter(blength)) =>
ColumnTypeCharacter(alength.max(blength))
case (ColumnTypeShort(alength), ColumnTypeString(blength)) =>
ColumnTypeString(alength.max(blength))
case (ColumnTypeShort(_), ColumnTypeUnknown) => ColumnTypeUnknown

// ColumnTypeInt
case (ColumnTypeInt(_), ColumnTypeBoolean) => merge(b, a)
case (ColumnTypeInt(_), ColumnTypeBoolean) => merge(b, a)
case (ColumnTypeInt(_), ColumnTypeByte(_)) => merge(b, a)
case (ColumnTypeInt(_), ColumnTypeShort(_)) => merge(b, a)
case (ColumnTypeInt(alength), ColumnTypeInt(blength)) =>
ColumnTypeInt(alength.max(blength))
case (ColumnTypeInt(alength), ColumnTypeLong(blength)) =>
ColumnTypeLong(alength.max(blength))
case (ColumnTypeInt(alength), ColumnTypeFloat(blength)) =>
ColumnTypeFloat(alength.max(blength))
case (ColumnTypeInt(alength), ColumnTypeDouble(blength)) =>
ColumnTypeDouble(alength.max(blength))
case (ColumnTypeInt(alength), ColumnTypeUUID) =>
// change UUID -> Character
ColumnTypeCharacter(alength.max(ColumnTypeUUID.length))
case (ColumnTypeInt(alength), ColumnTypeCharacter(blength)) =>
ColumnTypeCharacter(alength.max(blength))
case (ColumnTypeInt(alength), ColumnTypeString(blength)) =>
ColumnTypeString(alength.max(blength))
case (ColumnTypeInt(_), ColumnTypeUnknown) => ColumnTypeUnknown

// ColumnTypeLong
case (ColumnTypeLong(_), ColumnTypeBoolean) => merge(b, a)
case (ColumnTypeLong(_), ColumnTypeInt(_)) => merge(b, a)
case (ColumnTypeLong(_), ColumnTypeBoolean) => merge(b, a)
case (ColumnTypeLong(_), ColumnTypeByte(_)) => merge(b, a)
case (ColumnTypeLong(_), ColumnTypeShort(_)) => merge(b, a)
case (ColumnTypeLong(_), ColumnTypeInt(_)) => merge(b, a)
case (ColumnTypeLong(alength), ColumnTypeLong(blength)) =>
ColumnTypeLong(alength.max(blength))
case (ColumnTypeLong(alength), ColumnTypeFloat(blength)) =>
ColumnTypeFloat(alength.max(blength))
case (ColumnTypeLong(alength), ColumnTypeDouble(blength)) =>
ColumnTypeDouble(alength.max(blength))
case (ColumnTypeLong(alength), ColumnTypeUUID) =>
// change UUID -> Character
ColumnTypeCharacter(alength.max(ColumnTypeUUID.length))
case (ColumnTypeLong(alength), ColumnTypeCharacter(blength)) =>
ColumnTypeCharacter(alength.max(blength))
case (ColumnTypeLong(alength), ColumnTypeString(blength)) =>
ColumnTypeString(alength.max(blength))
case (ColumnTypeLong(_), ColumnTypeUnknown) => ColumnTypeUnknown

// ColumnTypeFloat
case (ColumnTypeFloat(_), ColumnTypeBoolean) => merge(b, a)
case (ColumnTypeFloat(_), ColumnTypeByte(_)) => merge(b, a)
case (ColumnTypeFloat(_), ColumnTypeShort(_)) => merge(b, a)
case (ColumnTypeFloat(_), ColumnTypeInt(_)) => merge(b, a)
case (ColumnTypeFloat(_), ColumnTypeLong(_)) => merge(b, a)
case (ColumnTypeFloat(alength), ColumnTypeFloat(blength)) =>
ColumnTypeFloat(alength.max(blength))
case (ColumnTypeFloat(alength), ColumnTypeDouble(blength)) =>
ColumnTypeDouble(alength.max(blength))
case (ColumnTypeFloat(alength), ColumnTypeUUID) =>
// change UUID -> Character
ColumnTypeCharacter(alength.max(ColumnTypeUUID.length))
case (ColumnTypeFloat(alength), ColumnTypeCharacter(blength)) =>
ColumnTypeCharacter(alength.max(blength))
case (ColumnTypeFloat(alength), ColumnTypeString(blength)) =>
ColumnTypeString(alength.max(blength))
case (ColumnTypeFloat(_), ColumnTypeUnknown) => ColumnTypeUnknown

// ColumnTypeDouble
case (ColumnTypeDouble(_), ColumnTypeBoolean) => merge(b, a)
case (ColumnTypeDouble(_), ColumnTypeInt(_)) => merge(b, a)
case (ColumnTypeDouble(_), ColumnTypeLong(_)) => merge(b, a)
case (ColumnTypeDouble(_), ColumnTypeBoolean) => merge(b, a)
case (ColumnTypeDouble(_), ColumnTypeByte(_)) => merge(b, a)
case (ColumnTypeDouble(_), ColumnTypeShort(_)) => merge(b, a)
case (ColumnTypeDouble(_), ColumnTypeInt(_)) => merge(b, a)
case (ColumnTypeDouble(_), ColumnTypeLong(_)) => merge(b, a)
case (ColumnTypeDouble(_), ColumnTypeFloat(_)) => merge(b, a)
case (ColumnTypeDouble(alength), ColumnTypeDouble(blength)) =>
ColumnTypeDouble(alength.max(blength))
case (ColumnTypeDouble(alength), ColumnTypeUUID) =>
// change UUID -> Character
ColumnTypeCharacter(alength.max(ColumnTypeUUID.length))
case (ColumnTypeDouble(alength), ColumnTypeCharacter(blength)) =>
ColumnTypeCharacter(alength.max(blength))
case (ColumnTypeDouble(alength), ColumnTypeString(blength)) =>
ColumnTypeString(alength.max(blength))
case (ColumnTypeDouble(_), ColumnTypeUnknown) => ColumnTypeUnknown

// ColumnTypeUUID
case (ColumnTypeUUID, ColumnTypeBoolean) => merge(b, a)
case (ColumnTypeUUID, ColumnTypeByte(_)) => merge(b, a)
case (ColumnTypeUUID, ColumnTypeShort(_)) => merge(b, a)
case (ColumnTypeUUID, ColumnTypeInt(_)) => merge(b, a)
case (ColumnTypeUUID, ColumnTypeLong(_)) => merge(b, a)
case (ColumnTypeUUID, ColumnTypeFloat(_)) => merge(b, a)
case (ColumnTypeUUID, ColumnTypeDouble(_)) => merge(b, a)
case (ColumnTypeUUID, ColumnTypeUUID) => ColumnTypeUUID
case (ColumnTypeUUID, ColumnTypeCharacter(blength)) =>
ColumnTypeCharacter(ColumnTypeUUID.length.max(blength))
case (ColumnTypeUUID, ColumnTypeString(blength)) =>
ColumnTypeString(ColumnTypeUUID.length.max(blength))
case (ColumnTypeUUID, ColumnTypeUnknown) => ColumnTypeUnknown

// ColumnTypeCharacter
case (ColumnTypeCharacter(_), ColumnTypeBoolean) => merge(b, a)
case (ColumnTypeCharacter(_), ColumnTypeByte(_)) => merge(b, a)
case (ColumnTypeCharacter(_), ColumnTypeShort(_)) => merge(b, a)
case (ColumnTypeCharacter(_), ColumnTypeInt(_)) => merge(b, a)
case (ColumnTypeCharacter(_), ColumnTypeLong(_)) => merge(b, a)
case (ColumnTypeCharacter(_), ColumnTypeFloat(_)) => merge(b, a)
case (ColumnTypeCharacter(_), ColumnTypeDouble(_)) => merge(b, a)
case (ColumnTypeCharacter(alength), ColumnTypeUUID) =>
ColumnTypeCharacter(alength.max(ColumnTypeUUID.length))
case (ColumnTypeCharacter(alength), ColumnTypeCharacter(blength)) =>
ColumnTypeCharacter(alength.max(blength))
case (ColumnTypeCharacter(alength), ColumnTypeString(blength)) =>
ColumnTypeString(alength.max(blength))
case (ColumnTypeCharacter(_), ColumnTypeUnknown) => ColumnTypeUnknown

// ColumnTypeString
case (ColumnTypeString(_), ColumnTypeBoolean) => merge(b, a)
case (ColumnTypeString(_), ColumnTypeInt(_)) => merge(b, a)
case (ColumnTypeString(_), ColumnTypeLong(_)) => merge(b, a)
case (ColumnTypeString(_), ColumnTypeDouble(_)) => merge(b, a)
case (ColumnTypeString(_), ColumnTypeBoolean) => merge(b, a)
case (ColumnTypeString(_), ColumnTypeByte(_)) => merge(b, a)
case (ColumnTypeString(_), ColumnTypeShort(_)) => merge(b, a)
case (ColumnTypeString(_), ColumnTypeInt(_)) => merge(b, a)
case (ColumnTypeString(_), ColumnTypeLong(_)) => merge(b, a)
case (ColumnTypeString(_), ColumnTypeFloat(_)) => merge(b, a)
case (ColumnTypeString(_), ColumnTypeDouble(_)) => merge(b, a)
case (ColumnTypeString(_), ColumnTypeUUID) => merge(b, a)
case (ColumnTypeString(_), ColumnTypeCharacter(_)) => merge(b, a)
case (ColumnTypeString(alength), ColumnTypeString(blength)) =>
ColumnTypeString(alength.max(blength))
case (ColumnTypeString(_), ColumnTypeUnknown) => ColumnTypeUnknown

// ColumnTypeUnknown
case (ColumnTypeUnknown, ColumnTypeBoolean) => merge(b, a)
case (ColumnTypeUnknown, ColumnTypeInt(_)) => merge(b, a)
case (ColumnTypeUnknown, ColumnTypeLong(_)) => merge(b, a)
case (ColumnTypeUnknown, ColumnTypeDouble(_)) => merge(b, a)
case (ColumnTypeUnknown, ColumnTypeString(_)) => merge(b, a)
case (ColumnTypeUnknown, ColumnTypeUnknown) => ColumnTypeUnknown
case (ColumnTypeUnknown, ColumnTypeBoolean) => merge(b, a)
case (ColumnTypeUnknown, ColumnTypeByte(_)) => merge(b, a)
case (ColumnTypeUnknown, ColumnTypeShort(_)) => merge(b, a)
case (ColumnTypeUnknown, ColumnTypeInt(_)) => merge(b, a)
case (ColumnTypeUnknown, ColumnTypeLong(_)) => merge(b, a)
case (ColumnTypeUnknown, ColumnTypeFloat(_)) => merge(b, a)
case (ColumnTypeUnknown, ColumnTypeDouble(_)) => merge(b, a)
case (ColumnTypeUnknown, ColumnTypeUUID) => merge(b, a)
case (ColumnTypeUnknown, ColumnTypeCharacter(_)) => merge(b, a)
case (ColumnTypeUnknown, ColumnTypeString(_)) => merge(b, a)
case (ColumnTypeUnknown, ColumnTypeUnknown) => ColumnTypeUnknown
}
}
}
22 changes: 16 additions & 6 deletions src/main/scala/domain/table/dml/RecordValue.scala
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3,10 +3,15 @@ package domain.table.dml
import domain.table.ddl.column.{
ColumnType,
ColumnTypeBoolean,
ColumnTypeByte,
ColumnTypeCharacter,
ColumnTypeDouble,
ColumnTypeFloat,
ColumnTypeInt,
ColumnTypeLong,
ColumnTypeShort,
ColumnTypeString,
ColumnTypeUUID,
ColumnTypeUnknown
}

Expand All @@ -18,12 +23,17 @@ final case class RecordValue(private val value: Map[String, Any])

val valuesForSql = values.map { value =>
ColumnType.apply(value) match {
case ColumnTypeBoolean => value
case ColumnTypeInt(_) => value
case ColumnTypeLong(_) => value
case ColumnTypeDouble(_) => value
case ColumnTypeString(_) => s"\"$value\""
case ColumnTypeUnknown => s"\"$value\""
case ColumnTypeBoolean => value
case ColumnTypeByte(_) => value
case ColumnTypeShort(_) => value
case ColumnTypeInt(_) => value
case ColumnTypeLong(_) => value
case ColumnTypeFloat(_) => value
case ColumnTypeDouble(_) => value
case ColumnTypeUUID => s"\"$value\""
case ColumnTypeCharacter(_) => s"\"$value\""
case ColumnTypeString(_) => s"\"$value\""
case ColumnTypeUnknown => s"\"$value\""
}
}

Expand Down

0 comments on commit 3dccb3f

Please sign in to comment.