fix: partly fix ci error in databend (#784)

zhenglin-charlie-li · web-flow · commit 9e17ede91412 · 2023-05-20T21:43:51.000+08:00
* fix: fix ci error https://github.com/sqlancer/sqlancer/actions/runs/4822718555/jobs/8590290712 * style: run mvn format * fix: fix bool type size * fix: Nullable type * fix: add FLOAT64 * refactor: change the sequence correspond to https://databend.rs/doc/sql-reference/data-types/#general-data-types
diff --git a/src/sqlancer/databend/DatabendSchema.java b/src/sqlancer/databend/DatabendSchema.java
@@ -232,48 +232,65 @@ public DatabendTables getRandomTableNonEmptyAndViewTables() {
     }
 
     private static DatabendCompositeDataType getColumnType(String typeString) {
-        DatabendDataType primitiveType;
-        int size = -1;
         if (typeString.startsWith("DECIMAL")) { // Ugly hack
             return new DatabendCompositeDataType(DatabendDataType.FLOAT, 8);
         }
-        switch (typeString) {
-        case "INT":
+        if (typeString.startsWith("Nullable")) { // Ugly hack
+            String substring = typeString.substring(typeString.indexOf('(') + 1, typeString.indexOf(')'));
+            return getColumnTypeNormalCases(substring);
+        }
+        return getColumnTypeNormalCases(typeString);
+    }
+
+    private static DatabendCompositeDataType getColumnTypeNormalCases(String typeString) {
+        DatabendDataType primitiveType;
+        int size = -1;
+        switch (typeString.toUpperCase()) {
+        case "BOOLEAN":
+        case "BOOL":
+            primitiveType = DatabendDataType.BOOLEAN;
+            size = 1;
+            break;
+        case "TINYINT":
+        case "INT8":
             primitiveType = INT;
-            size = 4;
+            size = 1;
             break;
         case "SMALLINT":
+        case "INT16":
             primitiveType = INT;
             size = 2;
             break;
-        case "BIGINT":
+        case "INT":
+        case "INT32":
             primitiveType = INT;
-            size = 8;
+            size = 4;
             break;
-        case "TINYINT":
+        case "BIGINT":
+        case "INT64":
             primitiveType = INT;
-            size = 1;
-            break;
-        case "VARCHAR":
-            primitiveType = DatabendDataType.VARCHAR;
+            size = 8;
             break;
         case "FLOAT":
+        case "FLOAT32":
             primitiveType = DatabendDataType.FLOAT;
             size = 4;
             break;
         case "DOUBLE":
+        case "FLOAT64":
             primitiveType = DatabendDataType.FLOAT;
             size = 8;
             break;
-        case "BOOLEAN":
-            primitiveType = DatabendDataType.BOOLEAN;
-            break;
         // case "DATE":
         // primitiveType = DatabendDataType.DATE;
         // break;
         // case "TIMESTAMP":
         // primitiveType = DatabendDataType.TIMESTAMP;
         // break;
+        case "VARCHAR":
+        case "STRING":
+            primitiveType = DatabendDataType.VARCHAR;
+            break;
         case "NULL":
             primitiveType = DatabendDataType.NULL;
             break;
