fix(interactive): Fix Bugs of Type Inference in Collect(labels(n))

interactive_engine/compiler/src/main/java/com/alibaba/graphscope/common/ir/runtime/proto/Utils.java

@@ -296,15 +296,14 @@ public static final Common.DataType protoBasicDataType(RelDataType basicType) {
             case MULTISET:
             case ARRAY:
                 RelDataType elementType = basicType.getComponentType();
-                switch (elementType.getSqlTypeName()) {
-                    case INTEGER:
+                Common.DataType dataType = protoBasicDataType(elementType);
+                switch (dataType) {
+                    case INT32:
                         return Common.DataType.INT32_ARRAY;
-                    case BIGINT:
+                    case INT64:
                         return Common.DataType.INT64_ARRAY;
-                    case CHAR:
+                    case STRING:
                         return Common.DataType.STRING_ARRAY;
-                    case DECIMAL:
-                    case FLOAT:
                     case DOUBLE:
                         return Common.DataType.DOUBLE_ARRAY;
                     default:

interactive_engine/compiler/src/test/java/com/alibaba/graphscope/common/ir/runtime/GraphRelToProtoTest.java

@@ -27,12 +27,7 @@
 import com.alibaba.graphscope.common.ir.tools.GraphBuilder;
 import com.alibaba.graphscope.common.ir.tools.GraphStdOperatorTable;
 import com.alibaba.graphscope.common.ir.tools.LogicalPlan;
-import com.alibaba.graphscope.common.ir.tools.config.ExpandConfig;
-import com.alibaba.graphscope.common.ir.tools.config.GetVConfig;
-import com.alibaba.graphscope.common.ir.tools.config.GraphOpt;
-import com.alibaba.graphscope.common.ir.tools.config.LabelConfig;
-import com.alibaba.graphscope.common.ir.tools.config.PathExpandConfig;
-import com.alibaba.graphscope.common.ir.tools.config.SourceConfig;
+import com.alibaba.graphscope.common.ir.tools.config.*;
 import com.alibaba.graphscope.common.utils.FileUtils;
 import com.google.common.collect.ImmutableList;
 import com.google.common.collect.ImmutableMap;
@@ -1237,6 +1232,24 @@ public void intersect_test_02() throws Exception {
         }
     }
 
+    @Test
+    public void collect_labels_test() {
+        Configs configs = getMockCBOConfig();
+        GraphRelOptimizer optimizer = new GraphRelOptimizer(configs);
+        IrMeta irMeta = getMockCBOMeta(optimizer);
+        GraphBuilder builder = Utils.mockGraphBuilder(optimizer, irMeta);
+        RelNode before =
+                com.alibaba.graphscope.cypher.antlr4.Utils.eval(
+                                "Match (n:PERSON) Return collect(labels(n));", builder)
+                        .build();
+        RelNode after = optimizer.optimize(before, new GraphIOProcessor(builder, irMeta));
+        PhysicalBuilder protoBuilder =
+                new GraphRelProtoPhysicalBuilder(configs, irMeta, new LogicalPlan(after));
+        Assert.assertEquals(
+                FileUtils.readJsonFromResource("proto/collect_labels.json"),
+                protoBuilder.build().explain().trim());
+    }
+
     private Configs getMockCBOConfig() {
         return new Configs(
                 ImmutableMap.of(

interactive_engine/compiler/src/test/resources/proto/collect_labels.json

@@ -0,0 +1,106 @@
+{
+  "plan": [{
+    "opr": {
+      "scan": {
+        "alias": 0,
+        "params": {
+          "tables": [{
+            "id": 1
+          }],
+          "sampleRatio": 1.0
+        }
+      }
+    },
+    "metaData": [{
+      "type": {
+        "graphType": {
+          "graphDataType": [{
+            "label": {
+              "label": 1
+            },
+            "props": [{
+              "propId": {
+                "name": "id"
+              },
+              "type": "INT64"
+            }, {
+              "propId": {
+                "name": "firstName"
+              },
+              "type": "STRING"
+            }, {
+              "propId": {
+                "name": "lastName"
+              },
+              "type": "STRING"
+            }, {
+              "propId": {
+                "name": "gender"
+              },
+              "type": "STRING"
+            }, {
+              "propId": {
+                "name": "birthday"
+              },
+              "type": "INT64"
+            }, {
+              "propId": {
+                "name": "creationDate"
+              },
+              "type": "DATE32"
+            }, {
+              "propId": {
+                "name": "locationIP"
+              },
+              "type": "STRING"
+            }, {
+              "propId": {
+                "name": "browserUsed"
+              },
+              "type": "STRING"
+            }]
+          }]
+        }
+      }
+    }]
+  }, {
+    "opr": {
+      "groupBy": {
+        "functions": [{
+          "vars": [{
+            "tag": {
+              "id": 0
+            },
+            "property": {
+              "label": {
+              }
+            },
+            "nodeType": {
+              "dataType": "INT32"
+            }
+          }],
+          "aggregate": "TO_LIST",
+          "alias": 1
+        }]
+      }
+    },
+    "metaData": [{
+      "type": {
+        "dataType": "INT32_ARRAY"
+      },
+      "alias": 1
+    }]
+  }, {
+    "opr": {
+      "sink": {
+        "tags": [{
+          "tag": 1
+        }],
+        "sinkTarget": {
+          "sinkDefault": {
+          }
+        }
+      }
+    }
+  }]
+}