ignore partition row count if table row count is registerred

englefly · Jan 7, 2025 · 035e37a · 035e37a
1 parent a539e22
commit 035e37a
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/fe/fe-core/src/main/java/org/apache/doris/nereids/stats/StatsCalculator.java b/fe/fe-core/src/main/java/org/apache/doris/nereids/stats/StatsCalculator.java
@@ -449,6 +449,12 @@ private double getTableRowCount(CatalogRelation scan) {
         }
     }
 
+    private boolean isRegisteredRowCount(OlapScan olapScan) {
+        AnalysisManager analysisManager = Env.getCurrentEnv().getAnalysisManager();
+        TableStatsMeta tableMeta = analysisManager.findTableStatsStatus(olapScan.getTable().getId());
+        return tableMeta != null && tableMeta.userInjected;
+    }
+
     /**
      * if the table is not analyzed and BE does not report row count, return -1
      */
@@ -554,7 +560,8 @@ private Statistics computeOlapScan(OlapScan olapScan) {
             }
         }
 
-        if (olapScan.getSelectedPartitionIds().size() < olapScan.getTable().getPartitionNum()) {
+        if (!isRegisteredRowCount(olapScan)
+                && olapScan.getSelectedPartitionIds().size() < olapScan.getTable().getPartitionNum()) {
             // partition pruned
             // try to use selected partition stats, if failed, fall back to table stats
             double selectedPartitionsRowCount = getSelectedPartitionRowCount(olapScan, tableRowCount);