add tinyint and smallint precisions

linkedin · Apr 11, 2024 · 071b434 · 071b434
1 parent 0218a7f
commit 071b434
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/coral-common/src/main/java/com/linkedin/coral/common/HiveTypeSystem.java b/coral-common/src/main/java/com/linkedin/coral/common/HiveTypeSystem.java
@@ -29,6 +29,8 @@ public class HiveTypeSystem extends RelDataTypeSystemImpl {
   private static final int DEFAULT_CHAR_PRECISION = 255;
   private static final int MAX_BINARY_PRECISION = Integer.MAX_VALUE;
   private static final int MAX_TIMESTAMP_PRECISION = 9;
+  private static final int DEFAULT_TINYINT_PRECISION  = 3;
+  private static final int DEFAULT_SMALLINT_PRECISION = 5;
   private static final int DEFAULT_INTEGER_PRECISION = 10;
   private static final int DEFAULT_BIGINT_PRECISION = 19;
 
@@ -86,6 +88,10 @@ public int getDefaultPrecision(SqlTypeName typeName) {
       case INTERVAL_MINUTE_SECOND:
       case INTERVAL_SECOND:
         return SqlTypeName.DEFAULT_INTERVAL_START_PRECISION;
+      case TINYINT:
+        return DEFAULT_TINYINT_PRECISION;
+      case SMALLINT:
+        return DEFAULT_SMALLINT_PRECISION;
       case INTEGER:
         return DEFAULT_INTEGER_PRECISION;
       case BIGINT: