[feature][dingo-calcite] Support window function

Author: guojn1

dingo-calcite/src/main/java/io/dingodb/calcite/DingoParser.java

@@ -414,6 +414,11 @@ public RelNode optimize(RelNode relNode) {
         List<RelOptRule> rules = DingoRules.rules();
         ImmutableList.Builder<RelOptRule> builder = ImmutableList.builder();
         builder.addAll(rules);
+
+        builder.addAll(DingoRules.ABSTRACT_RELATIONAL_RULES);
+        builder.addAll(DingoRules.ABSTRACT_RULES);
+        builder.addAll(DingoRules.BASE_RULES);
+
         if (!context.getConfig().topDownOpt()) {
             // This is needed for `IterativeRuleDriver`.
             builder.add(AbstractConverter.ExpandConversionRule.INSTANCE);

dingo-calcite/src/main/java/io/dingodb/calcite/rel/DingoWindow.java

@@ -0,0 +1,76 @@
+/*
+ * Copyright 2021 DataCanvas
+ *
+ * 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 io.dingodb.calcite.rel;
+
+import com.sun.istack.internal.Nullable;
+import io.dingodb.calcite.visitor.DingoRelVisitor;
+import org.apache.calcite.adapter.enumerable.PhysType;
+import org.apache.calcite.adapter.enumerable.RexToLixTranslator;
+import org.apache.calcite.linq4j.tree.BlockBuilder;
+import org.apache.calcite.linq4j.tree.Expression;
+import org.apache.calcite.plan.RelOptCluster;
+import org.apache.calcite.plan.RelTraitSet;
+import org.apache.calcite.rel.RelNode;
+import org.apache.calcite.rel.core.Window;
+import org.apache.calcite.rel.hint.RelHint;
+import org.apache.calcite.rel.type.RelDataType;
+import org.apache.calcite.rex.RexLiteral;
+import org.checkerframework.checker.nullness.qual.NonNull;
+
+import java.lang.reflect.Type;
+import java.util.List;
+
+public class DingoWindow extends Window implements DingoRel {
+
+    public DingoWindow(
+        RelOptCluster cluster, RelTraitSet traitSet, List<RelHint> hints, RelNode input,
+        List<RexLiteral> constants, RelDataType rowType, List<Group> groups
+    ) {
+        super(cluster, traitSet, hints, input, constants, rowType, groups);
+    }
+
+    private static class WindowRelInputGetter
+        implements RexToLixTranslator.InputGetter {
+        private final Expression row;
+        private final PhysType rowPhysType;
+        private final int actualInputFieldCount;
+        private final List<Expression> constants;
+
+        private WindowRelInputGetter(Expression row,
+                                     PhysType rowPhysType, int actualInputFieldCount,
+                                     List<Expression> constants) {
+            this.row = row;
+            this.rowPhysType = rowPhysType;
+            this.actualInputFieldCount = actualInputFieldCount;
+            this.constants = constants;
+        }
+
+        @Override public Expression field(BlockBuilder list, int index, @Nullable Type storageType) {
+            if (index < actualInputFieldCount) {
+                Expression current = list.append("current", row);
+                return rowPhysType.fieldReference(current, index, storageType);
+            }
+            return constants.get(index - actualInputFieldCount);
+        }
+    }
+
+
+    @Override
+    public <T> T accept(@NonNull DingoRelVisitor<T> visitor) {
+        return visitor.visit(this);
+    }
+}

dingo-calcite/src/main/java/io/dingodb/calcite/rule/dingo/DingoWindowRule.java

@@ -0,0 +1,56 @@
+/*
+ * Copyright 2021 DataCanvas
+ *
+ * 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 io.dingodb.calcite.rule.dingo;
+
+import io.dingodb.calcite.rel.DingoWindow;
+import io.dingodb.calcite.rel.logical.LogicalScanWithRelOp;
+import io.dingodb.calcite.traits.DingoConvention;
+import org.apache.calcite.plan.Convention;
+import org.apache.calcite.plan.RelTraitSet;
+import org.apache.calcite.rel.RelNode;
+import org.apache.calcite.rel.convert.ConverterRule;
+import org.apache.calcite.rel.core.Window;
+import org.apache.calcite.rel.logical.LogicalWindow;
+import org.checkerframework.checker.nullness.qual.Nullable;
+
+public class DingoWindowRule extends ConverterRule {
+    public static final Config DEFAULT = Config.INSTANCE
+        .withConversion(
+            LogicalWindow.class,
+            Convention.NONE,
+            DingoConvention.INSTANCE,
+            "DingoWindowRule"
+        )
+        .withRuleFactory(DingoWindowRule::new);
+
+    protected DingoWindowRule(Config config) {
+        super(config);
+    }
+
+    @Override
+    public @Nullable RelNode convert(RelNode rel) {
+        final Window winAgg = (Window) rel;
+        final RelTraitSet traitSet =
+            winAgg.getTraitSet().replace(DingoConvention.INSTANCE);
+        final RelNode child = winAgg.getInput();
+        final RelNode convertedChild =
+            convert(child,
+                child.getTraitSet().replace(DingoConvention.INSTANCE));
+        return new DingoWindow(rel.getCluster(), traitSet, winAgg.getHints(), convertedChild,
+            winAgg.getConstants(), winAgg.getRowType(), winAgg.groups);
+    }
+}

dingo-calcite/src/main/java/io/dingodb/calcite/visitor/DingoJobVisitor.java

@@ -44,6 +44,7 @@
 import io.dingodb.calcite.rel.DingoUnion;
 import io.dingodb.calcite.rel.DingoValues;
 import io.dingodb.calcite.rel.DingoVector;
+import io.dingodb.calcite.rel.DingoWindow;
 import io.dingodb.calcite.rel.DocumentStreamConvertor;
 import io.dingodb.calcite.rel.VectorStreamConvertor;
 import io.dingodb.calcite.rel.dingo.DingoDocumentScanFilter;
@@ -97,6 +98,7 @@
 import io.dingodb.calcite.visitor.function.DingoValuesVisitFun;
 import io.dingodb.calcite.visitor.function.DingoVectorStreamingVisitFun;
 import io.dingodb.calcite.visitor.function.DingoVectorVisitFun;
+import io.dingodb.calcite.visitor.function.DingoWindowVisitFun;
 import io.dingodb.common.ExecuteVariables;
 import io.dingodb.common.Location;
 import io.dingodb.common.log.LogUtils;
@@ -419,4 +421,8 @@ public Collection<Vertex> visit(@NonNull DingoDocumentScanFilter dingoDocumentSc
         return DingoDocumentScanFilterVisitFun.visit(job, idGenerator, currentLocation, this, transaction, dingoDocumentScanFilter);
     }
 
+    public Collection<Vertex> visit(@NonNull DingoWindow dingoWindow) {
+        return DingoWindowVisitFun.visit(job, idGenerator, currentLocation, this, transaction, dingoWindow);
+    }
+
 }

dingo-calcite/src/main/java/io/dingodb/calcite/visitor/DingoRelVisitor.java

@@ -44,6 +44,7 @@
 import io.dingodb.calcite.rel.DingoUnion;
 import io.dingodb.calcite.rel.DingoValues;
 import io.dingodb.calcite.rel.DingoVector;
+import io.dingodb.calcite.rel.DingoWindow;
 import io.dingodb.calcite.rel.DocumentStreamConvertor;
 import io.dingodb.calcite.rel.VectorStreamConvertor;
 import io.dingodb.calcite.rel.dingo.DingoDocumentScanFilter;
@@ -142,4 +143,6 @@ public interface DingoRelVisitor<T> {
 
     T visit(DingoDocumentScanFilter documentIndexRangeScan);
 
+    T visit(DingoWindow dingoWindow);
+
 }

dingo-calcite/src/main/java/io/dingodb/calcite/visitor/function/DingoWindowVisitFun.java

@@ -0,0 +1,46 @@
+/*
+ * Copyright 2021 DataCanvas
+ *
+ * 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 io.dingodb.calcite.visitor.function;
+
+import io.dingodb.calcite.rel.DingoWindow;
+import io.dingodb.calcite.rel.dingo.DingoDocumentScanFilter;
+import io.dingodb.calcite.visitor.DingoJobVisitor;
+import io.dingodb.common.Location;
+import io.dingodb.exec.base.IdGenerator;
+import io.dingodb.exec.base.Job;
+import io.dingodb.exec.dag.Vertex;
+import io.dingodb.exec.transaction.base.ITransaction;
+import org.checkerframework.checker.nullness.qual.NonNull;
+
+import java.util.Collection;
+
+public class DingoWindowVisitFun {
+
+    private DingoWindowVisitFun() {
+    }
+
+    public static @NonNull Collection<Vertex> visit(
+        Job job,
+        @NonNull IdGenerator idGenerator,
+        Location currentLocation,
+        DingoJobVisitor visitor,
+        ITransaction transaction,
+        @NonNull DingoWindow rel
+    ) {
+        return null;
+    }
+}