refactor(interactive): Integrate Common Gremlin Steps to Calcite-Based IR Layer

interactive_engine/compiler/src/main/java/com/alibaba/graphscope/common/ir/rel/graph/GraphLogicalSource.java

@@ -22,12 +22,15 @@
 import org.apache.calcite.plan.GraphOptCluster;
 import org.apache.calcite.rel.RelWriter;
 import org.apache.calcite.rel.hint.RelHint;
+import org.apache.calcite.rex.RexNode;
 import org.checkerframework.checker.nullness.qual.Nullable;
 
 import java.util.List;
+import java.util.Objects;
 
 public class GraphLogicalSource extends AbstractBindableTableScan {
     private final GraphOpt.Source opt;
+    private @Nullable RexNode uniqueKeyFilters;
 
     protected GraphLogicalSource(
             GraphOptCluster cluster,
@@ -54,6 +57,16 @@ public GraphOpt.Source getOpt() {
 
     @Override
     public RelWriter explainTerms(RelWriter pw) {
-        return super.explainTerms(pw).item("opt", getOpt());
+        return super.explainTerms(pw)
+                .item("opt", getOpt())
+                .itemIf("uniqueKeyFilters", uniqueKeyFilters, uniqueKeyFilters != null);
+    }
+
+    public void setUniqueKeyFilters(RexNode uniqueKeyFilters) {
+        this.uniqueKeyFilters = Objects.requireNonNull(uniqueKeyFilters);
+    }
+
+    public @Nullable RexNode getUniqueKeyFilters() {
+        return uniqueKeyFilters;
     }
 }

interactive_engine/compiler/src/main/java/com/alibaba/graphscope/common/ir/rex/operator/SqlArrayValueConstructor.java

@@ -0,0 +1,62 @@
+/*
+ * Copyright 2020 Alibaba Group Holding Limited.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.alibaba.graphscope.common.ir.rex.operator;
+
+import org.apache.calcite.rel.type.RelDataType;
+import org.apache.calcite.rel.type.RelDataTypeFactory;
+import org.apache.calcite.sql.SqlCallBinding;
+import org.apache.calcite.sql.SqlKind;
+import org.apache.calcite.sql.SqlOperatorBinding;
+import org.apache.calcite.sql.fun.SqlMultisetValueConstructor;
+import org.apache.calcite.sql.type.SqlTypeName;
+import org.apache.calcite.sql.type.SqlTypeUtil;
+import org.checkerframework.checker.nullness.qual.Nullable;
+
+import java.util.List;
+
+public class SqlArrayValueConstructor extends SqlMultisetValueConstructor {
+    public SqlArrayValueConstructor() {
+        super("ARRAY", SqlKind.ARRAY_VALUE_CONSTRUCTOR);
+    }
+
+    @Override
+    public RelDataType inferReturnType(SqlOperatorBinding opBinding) {
+        RelDataTypeFactory typeFactory = opBinding.getTypeFactory();
+        List<RelDataType> argTypes = opBinding.collectOperandTypes();
+        RelDataType componentType = getComponentType(typeFactory, argTypes);
+        return SqlTypeUtil.createArrayType(typeFactory, componentType, false);
+    }
+
+    // operands of array value constructor can be any, even if empty, i.e []
+    @Override
+    public boolean checkOperandTypes(SqlCallBinding callBinding, boolean throwOnFailure) {
+        return true;
+    }
+
+    @Override
+    protected @Nullable RelDataType getComponentType(
+            RelDataTypeFactory typeFactory, List<RelDataType> argTypes) {
+        try {
+            RelDataType componentType = typeFactory.leastRestrictive(argTypes);
+            return (componentType == null)
+                    ? typeFactory.createSqlType(SqlTypeName.ANY)
+                    : componentType;
+        } catch (AssertionError e) {
+            return typeFactory.createSqlType(SqlTypeName.ANY);
+        }
+    }
+}

interactive_engine/compiler/src/main/java/com/alibaba/graphscope/common/ir/rex/operator/SqlMapValueConstructor.java

@@ -0,0 +1,70 @@
+/*
+ * Copyright 2020 Alibaba Group Holding Limited.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.alibaba.graphscope.common.ir.rex.operator;
+
+import org.apache.calcite.rel.type.RelDataType;
+import org.apache.calcite.rel.type.RelDataTypeFactory;
+import org.apache.calcite.sql.SqlCallBinding;
+import org.apache.calcite.sql.SqlKind;
+import org.apache.calcite.sql.SqlOperatorBinding;
+import org.apache.calcite.sql.fun.SqlMultisetValueConstructor;
+import org.apache.calcite.sql.type.SqlTypeName;
+import org.apache.calcite.sql.type.SqlTypeUtil;
+import org.apache.calcite.util.Pair;
+import org.apache.calcite.util.Static;
+import org.apache.calcite.util.Util;
+import org.checkerframework.checker.nullness.qual.Nullable;
+
+import java.util.List;
+
+public class SqlMapValueConstructor extends SqlMultisetValueConstructor {
+    public SqlMapValueConstructor() {
+        super("MAP", SqlKind.MAP_VALUE_CONSTRUCTOR);
+    }
+
+    public RelDataType inferReturnType(SqlOperatorBinding opBinding) {
+        RelDataTypeFactory typeFactory = opBinding.getTypeFactory();
+        List<RelDataType> argTypes = opBinding.collectOperandTypes();
+        Pair<RelDataType, RelDataType> type =
+                Pair.of(
+                        getComponentType(typeFactory, Util.quotientList(argTypes, 2, 0)),
+                        getComponentType(typeFactory, Util.quotientList(argTypes, 2, 1)));
+        return SqlTypeUtil.createMapType(opBinding.getTypeFactory(), type.left, type.right, false);
+    }
+
+    @Override
+    public boolean checkOperandTypes(SqlCallBinding callBinding, boolean throwOnFailure) {
+        List<RelDataType> argTypes = callBinding.collectOperandTypes();
+        if (argTypes.size() % 2 > 0) {
+            throw callBinding.newValidationError(Static.RESOURCE.mapRequiresEvenArgCount());
+        }
+        return true;
+    }
+
+    @Override
+    protected @Nullable RelDataType getComponentType(
+            RelDataTypeFactory typeFactory, List<RelDataType> argTypes) {
+        try {
+            RelDataType componentType = typeFactory.leastRestrictive(argTypes);
+            return (componentType == null)
+                    ? typeFactory.createSqlType(SqlTypeName.ANY)
+                    : componentType;
+        } catch (AssertionError e) {
+            return typeFactory.createSqlType(SqlTypeName.ANY);
+        }
+    }
+}

interactive_engine/compiler/src/main/java/com/alibaba/graphscope/common/ir/runtime/ffi/FfiPhysicalBuilder.java

@@ -68,7 +68,8 @@ public FfiPhysicalBuilder(
             Configs graphConfig, IrMeta irMeta, LogicalPlan logicalPlan, PlanPointer planPointer) {
         super(
                 logicalPlan,
-                new GraphRelShuttleWrapper(new RelToFfiConverter(irMeta.getSchema().isColumnId())));
+                new GraphRelShuttleWrapper(
+                        new RelToFfiConverter(irMeta.getSchema().isColumnId(), graphConfig)));
         this.graphConfig = graphConfig;
         this.irMeta = irMeta;
         this.planPointer = Objects.requireNonNull(planPointer);