rust/greptimedb
1
0
Fork 0
mirror of https://github.com/GreptimeTeam/greptimedb.git synced 2026-11-14 06:46:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(promql): read bounded scan metrics through cooperative wrapper

Browse Source 
Signed-off-by: discord9 <55937128+discord9@users.noreply.github.com>
This commit is contained in:
discord9
2026-10-25 14:46:45 +08:00
parent c3dd11bf48
commit 222b1916ec
1 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests-integration/src/tests/promql_test.rs
+11 -2
View File
@@ -24,8 +24,8 @@ use common_query::prelude::{GREPTIME_TEMPORALITY_DELTA, OTLP_AGGREGATION_TEMPORA
use common_query::prometheus::PROMETHEUS_STALE_NAN_BITS;
use common_query::{Output, OutputData};
use common_recordbatch::util::collect_batches;
use datafusion::physical_plan::ExecutionPlanProperties;
use datafusion::physical_plan::coop::CooperativeExec;
use datafusion::physical_plan::{ExecutionPlan, ExecutionPlanProperties};
use datatypes::arrow::array::{
Array, Float64Array, Int64Array, StringArray, TimestampMillisecondArray,
};
@@ -2276,7 +2276,16 @@ async fn native_promql_two_occurrences_route_one_scan_per_partition() {
let legacy_bounded_rows =
native_result_rows(native_tql_output(&instance, &bounded_sql, false).await).await;
assert_eq!(bounded_rows, legacy_bounded_rows);
let scan_frame_rows = bounded_plan.children()[0]
let child = bounded_plan.children()[0];
let child = child
.downcast_ref::<CooperativeExec>()
.map_or(child, |cooperative| cooperative.input());
let series_scan = child.downcast_ref::<SeriesScanExec>().unwrap_or_else(|| {
panic!(
"bounded fused node child is not a prepared series scan: child={child:?}; plan={bounded_plan:?}"
)
});
let scan_frame_rows = series_scan
.metrics()
.expect("series scan metrics")
.output_rows()