fix ut

Signed-off-by: yanz <[email protected]>
StarRocks · Oct 10, 2024 · 51701db · 51701db
1 parent a033954
commit 51701db
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/fe/fe-core/src/main/java/com/starrocks/planner/DeltaLakeScanNode.java b/fe/fe-core/src/main/java/com/starrocks/planner/DeltaLakeScanNode.java
@@ -166,11 +166,15 @@ protected String getNodeExplainString(String prefix, TExplainLevel detailLevel)
                     deltaLakeTable.getCatalogName(), deltaLakeTable.getDbName(), deltaLakeTable.getTableName());
 
             if (selectedPartitionCount == -1) {
-                // we have to consume all scan ranges to know how many partition been selected.
-                while (scanRangeSource.hasMoreOutput()) {
-                    scanRangeSource.getOutputs(1000);
+                if (scanRangeSource != null) {
+                    // we have to consume all scan ranges to know how many partition been selected.
+                    while (scanRangeSource.hasMoreOutput()) {
+                        scanRangeSource.getOutputs(1000);
+                    }
+                    selectedPartitionCount = scanRangeSource.selectedPartitionCount();
+                } else {
+                    selectedPartitionCount = 0;
                 }
-                selectedPartitionCount = scanRangeSource.selectedPartitionCount();
             }
 
             output.append(prefix).append(