fix(interactive): Fix Bugs of Count Estimation When Feasible Plan Is …

…Unavailable in Equivalent Set (#4181)
alibaba · Sep 3, 2024 · 73c1f6c · 73c1f6c
1 parent 2d3268e
commit 73c1f6c
Showing 1 changed file with 19 additions and 2 deletions.
diff --git a/...va/com/alibaba/graphscope/common/ir/meta/glogue/calcite/handler/GraphRowCountHandler.java b/...va/com/alibaba/graphscope/common/ir/meta/glogue/calcite/handler/GraphRowCountHandler.java
@@ -109,8 +109,19 @@ public Double getRowCount(RelNode node, RelMetadataQuery mq) {
                     }
                 }
             }
-            throw new UnsupportedOperationException(
-                    "estimate count for pattern " + pattern + " is unsupported yet");
+            double totalRowCount = 1.0d;
+            for (PatternEdge edge : pattern.getEdgeSet()) {
+                totalRowCount *= countEstimator.estimate(edge);
+            }
+            for (PatternVertex vertex : pattern.getVertexSet()) {
+                int degree = pattern.getEdgesOf(vertex).size();
+                if (degree > 0) {
+                    double vertexCount = countEstimator.estimate(vertex);
+                    vertexCount = Double.compare(vertexCount, 0.0d) == 0 ? 1.0d : vertexCount;
+                    totalRowCount /= Math.pow(vertexCount, degree - 1);
+                }
+            }
+            return totalRowCount;
         } else if (node instanceof TableScan) {
             return getRowCount((TableScan) node, mq);
         } else if (node instanceof Filter) {
@@ -131,6 +142,12 @@ public Double getRowCount(RelNode node, RelMetadataQuery mq) {
                     return mq.getRowCount(subset);
                 }
             }
+            Pattern original =
+                    (node instanceof GraphExtendIntersect)
+                            ? ((GraphExtendIntersect) node).getGlogueEdge().getDstPattern()
+                            : ((GraphJoinDecomposition) node).getParentPatten();
+            return mq.getRowCount(
+                    new GraphPattern(node.getCluster(), node.getTraitSet(), original));
         } else if (node instanceof Join) {
             return mdRowCount.getRowCount((Join) node, mq);
         } else if (node instanceof Union) {