adding FloatType in ColumnarLiteral

Signed-off-by: Yuan Zhou <[email protected]>
oap-project · May 25, 2022 · 4cda7d5 · 4cda7d5
1 parent 007717e
commit 4cda7d5
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/native-sql-engine/core/src/main/scala/com/intel/oap/expression/ColumnarLiterals.scala b/native-sql-engine/core/src/main/scala/com/intel/oap/expression/ColumnarLiterals.scala
@@ -42,7 +42,7 @@ class ColumnarLiteral(lit: Literal)
 
   def buildCheck(): ArrowType = {
     val supportedTypes =
-      List(StringType, IntegerType, LongType, DoubleType, DateType,
+      List(StringType, IntegerType, LongType, DoubleType, FloatType, DateType,
            BooleanType, CalendarIntervalType, BinaryType, TimestampType)
     if (supportedTypes.indexOf(dataType) == -1 && !dataType.isInstanceOf[DecimalType]) {
       // Decimal is supported in ColumnarLiteral
@@ -111,6 +111,13 @@ class ColumnarLiteral(lit: Literal)
           case _ =>
             (TreeBuilder.makeLiteral(value.asInstanceOf[java.lang.Double]), resultType)
         }
+      case t: FloatType =>
+        value match {
+          case null =>
+            (TreeBuilder.makeNull(resultType), resultType)
+          case _ =>
+            (TreeBuilder.makeLiteral(value.asInstanceOf[java.lang.Float]), resultType)
+        }
       case d: DecimalType =>
         value match {
           case null =>