add all casts

Author: fupelaqu

sql/src/main/scala/app/softnetwork/elastic/sql/SQLParser.scala

@@ -413,11 +413,18 @@ trait SQLParser extends RegexParsers with PackratParsers { _: SQLWhereParser =>
     "minute",
     "second",
     "quarter",
+    "char",
     "string",
+    "byte",
+    "tinyint",
+    "short",
+    "smallint",
     "int",
     "integer",
     "long",
     "bigint",
+    "real",
+    "float",
     "double",
     "boolean",
     "time",
@@ -460,7 +467,11 @@ trait SQLParser extends RegexParsers with PackratParsers { _: SQLWhereParser =>
       )
     }
 
-  def string_type: PackratParser[SQLTypes.Varchar.type] = "(?i)string".r ^^ (_ => SQLTypes.Varchar)
+  def char_type: PackratParser[SQLTypes.Char.type] =
+    "(?i)char".r ^^ (_ => SQLTypes.Char)
+
+  def string_type: PackratParser[SQLTypes.Varchar.type] =
+    "(?i)varchar|string".r ^^ (_ => SQLTypes.Varchar)
 
   def date_type: PackratParser[SQLTypes.Date.type] = "(?i)date".r ^^ (_ => SQLTypes.Date)
 
@@ -475,14 +486,22 @@ trait SQLParser extends RegexParsers with PackratParsers { _: SQLWhereParser =>
   def boolean_type: PackratParser[SQLTypes.Boolean.type] =
     "(?i)boolean".r ^^ (_ => SQLTypes.Boolean)
 
+  def byte_type: PackratParser[SQLTypes.TinyInt.type] =
+    "(?i)(byte|tinyint)".r ^^ (_ => SQLTypes.TinyInt)
+
+  def short_type: PackratParser[SQLTypes.SmallInt.type] =
+    "(?i)(short|smallint)".r ^^ (_ => SQLTypes.SmallInt)
+
+  def int_type: PackratParser[SQLTypes.Int.type] = "(?i)(int|integer)".r ^^ (_ => SQLTypes.Int)
+
   def long_type: PackratParser[SQLTypes.BigInt.type] = "(?i)long|bigint".r ^^ (_ => SQLTypes.BigInt)
 
   def double_type: PackratParser[SQLTypes.Double.type] = "(?i)double".r ^^ (_ => SQLTypes.Double)
 
-  def int_type: PackratParser[SQLTypes.Int.type] = "(?i)(int|integer)".r ^^ (_ => SQLTypes.Int)
+  def float_type: PackratParser[SQLTypes.Real.type] = "(?i)float|real".r ^^ (_ => SQLTypes.Real)
 
   def sql_type: PackratParser[SQLType] =
-    string_type | datetime_type | timestamp_type | date_type | time_type | boolean_type | long_type | double_type | int_type
+    char_type | string_type | datetime_type | timestamp_type | date_type | time_type | boolean_type | long_type | double_type | float_type | int_type | short_type | byte_type
 
   private[this] def castFunctionWithIdentifier: PackratParser[SQLIdentifier] =
     "(?i)cast".r ~ start ~ (identifierWithTransformation | identifierWithSystemFunction | identifierWithArithmeticFunction | identifierWithFunction | date_diff_identifier | extract_identifier | identifier) ~ Alias.regex.? ~ sql_type ~ end ~ arithmeticFunction.? ^^ {

sql/src/main/scala/app/softnetwork/elastic/sql/SQLTypes.scala

@@ -19,7 +19,7 @@ object SQLTypes {
   case object Int extends SQLInt { val typeId = "int" }
   case object BigInt extends SQLBigInt { val typeId = "bigint" }
   case object Double extends SQLDouble { val typeId = "double" }
-  case object Real extends SQLReal { val typeId = "float" }
+  case object Real extends SQLReal { val typeId = "real" }
 
   case object Literal extends SQLLiteral { val typeId = "literal" }
 

sql/src/test/scala/app/softnetwork/elastic/sql/SQLParserSpec.scala

@@ -143,6 +143,8 @@ object Queries {
     "select coalesce(nullif(createdAt, parse_date('2025-09-11', 'yyyy-MM-dd') - interval 2 day), current_date) as c, identifier from Table"
   val cast: String =
     "select cast(coalesce(nullif(createdAt, parse_date('2025-09-11', 'yyyy-MM-dd')), current_date - interval 2 hour) bigint) as c, identifier from Table"
+  val allCasts =
+    "select cast(identifier as int) as c1, cast(identifier as bigint) as c2, cast(identifier as double) as c3, cast(identifier as real) as c4, cast(identifier as boolean) as c5, cast(identifier as char) as c6, cast(identifier as varchar) as c7, cast(createdAt as date) as c8, cast(createdAt as time) as c9, cast(createdAt as datetime) as c10, cast(createdAt as timestamp) as c11, cast(identifier as smallint) as c12, cast(identifier as tinyint) as c13 from Table"
   val caseWhen: String =
     "select case when lastUpdated > now - interval 7 day then lastUpdated when isnotnull(lastSeen) then lastSeen + interval 2 day else createdAt end as c, identifier from Table"
   val caseWhenExpr: String =
@@ -573,6 +575,13 @@ class SQLParserSpec extends AnyFlatSpec with Matchers {
     )
   }
 
+  it should "parse all casts function" in {
+    val result = SQLParser(allCasts)
+    result.toOption.flatMap(_.left.toOption.map(_.sql)).getOrElse("") should ===(
+      allCasts
+    )
+  }
+
   it should "parse case when expression" in {
     val result = SQLParser(caseWhen)
     result.toOption.flatMap(_.left.toOption.map(_.sql)).getOrElse("") should ===(