refactor(planner): port the plan-selection crate split to flat DAGs and Operator<NodeId> payloads

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01W7qG9aFyPij5uWsyAJCxDW

Author: zzylol

crates/devtools/src/bin/stage_pipeline.rs

@@ -25,7 +25,9 @@
 
 use std::rc::Rc;
 
-use asap_logical_optimizer::pass1::logical_candidates::{choice_index, enumerate_local_logical_candidates, LocalLogicalCandidates};
+use asap_logical_optimizer::pass1::logical_candidates::{
+    choice_index, enumerate_local_logical_candidates, LocalLogicalCandidates,
+};
 use asap_logical_optimizer::Realization;
 use asap_plan_selection::PlanningModels;
 use asap_plan_selection::{select_exhaustive, Selection, MAX_ENUMERATED_CANDIDATES};
@@ -34,7 +36,12 @@ use asap_types::ir::schema::SketchAlgorithm;
 use asap_types::ir::schema_support::with_promql_series_identity;
 use asap_types::ir::{OperatorNode, QueryRoot};
 use asap_types::types::AccuracyTarget;
-use asap_types::workload::{AccuracyRequirement, BatchEntry, DataArrival, DataDistribution, DataWorkload, DurationMs, Evidence, EvidenceSource, LatencyRequirement, PlanningWorkload, Predictability, Query, QueryLanguage, QueryRecurrence, QueryRequirements, QueryTimeScope, QueryWorkload, Rate, RepeatedDemand, RepeatingEntry, RepetitionInterval, TimeSelection};
+use asap_types::workload::{
+    AccuracyRequirement, BatchEntry, DataArrival, DataDistribution, DataWorkload, DurationMs,
+    Evidence, EvidenceSource, LatencyRequirement, PlanningWorkload, Predictability, Query,
+    QueryLanguage, QueryRecurrence, QueryRequirements, QueryTimeScope, QueryWorkload, Rate,
+    RepeatedDemand, RepeatingEntry, RepetitionInterval, TimeSelection,
+};
 use serde_json::{json, Value};
 
 const USAGE: &str = "usage: stage_pipeline (--example planner-layering-1 | --promql <query>... \

crates/integration-tests/tests/planner_layering_example1.rs

@@ -15,12 +15,17 @@
 use std::collections::{BTreeMap, BTreeSet, HashSet};
 
 use asap_plan_selection::PlanningModels;
-use asap_types::ir::schema::state_type::{GroupingStrategy, HydraKind, SketchAlgorithm};
 use asap_types::ir::flat::{flatten, FlatDag, NodeId};
+use asap_types::ir::schema::state_type::{GroupingStrategy, HydraKind, SketchAlgorithm};
 use asap_types::ir::schema::FieldDataType;
 use asap_types::ir::{ASAPOp, Operator};
 use asap_types::types::AccuracyTarget;
-use asap_types::workload::{AccuracyRequirement, DataArrival, DataDistribution, DataWorkload, DurationMs, Evidence, EvidenceSource, LatencyRequirement, PlanningWorkload, Predictability, Query, QueryLanguage, QueryRequirements, QueryTimeScope, QueryWorkload, Rate, RepeatedDemand, RepeatingEntry, RepetitionInterval, TimeSelection};
+use asap_types::workload::{
+    AccuracyRequirement, DataArrival, DataDistribution, DataWorkload, DurationMs, Evidence,
+    EvidenceSource, LatencyRequirement, PlanningWorkload, Predictability, Query, QueryLanguage,
+    QueryRequirements, QueryTimeScope, QueryWorkload, Rate, RepeatedDemand, RepeatingEntry,
+    RepetitionInterval, TimeSelection,
+};
 
 type Payload = Operator<NodeId>;
 
@@ -31,7 +36,9 @@ mod stages {
     use std::rc::Rc;
 
     use super::*;
-    use asap_logical_optimizer::pass1::logical_candidates::{compose_logical_candidate, enumerate_local_logical_candidates};
+    use asap_logical_optimizer::pass1::logical_candidates::{
+        compose_logical_candidate, enumerate_local_logical_candidates,
+    };
     use asap_types::ir::{OperatorNode, QueryRoot};
 
     /// One whole-workload candidate. `query_roots` holds one root per

crates/integration-tests/tests/promql_to_post_asap.rs

@@ -9,18 +9,31 @@
 use std::rc::Rc;
 
 use asap_integration_tests::fixtures::lower_promql;
-use asap_integration_tests::post_asap::{maintained, maintained_post_asap_dag, post_asap_dag, timed};
-use asap_logical_optimizer::accuracy::{AccuracyEvidenceProvider, DefaultAccuracyModel, EqualSplitAllocator, PropagationStats, QuantileInputDomain};
-use asap_logical_optimizer::pass1::replacement::{is_logical_rewrite, retain_exact, RealizationError};
-use asap_logical_optimizer::{search_workload, search_workload_with_targets, ASAPStrategies, AccuracyModel, Replacement, ReplacementStrategy, ReplacementSubDAG, TargetSubDAG};
+use asap_integration_tests::post_asap::{
+    maintained, maintained_post_asap_dag, post_asap_dag, timed,
+};
+use asap_logical_optimizer::accuracy::{
+    AccuracyEvidenceProvider, DefaultAccuracyModel, EqualSplitAllocator, PropagationStats,
+    QuantileInputDomain,
+};
+use asap_logical_optimizer::pass1::replacement::{
+    is_logical_rewrite, retain_exact, RealizationError,
+};
+use asap_logical_optimizer::{
+    search_workload, search_workload_with_targets, ASAPStrategies, AccuracyModel, Replacement,
+    ReplacementStrategy, ReplacementSubDAG, TargetSubDAG,
+};
 use asap_plan_selection::candidate_selection::global_selection;
 use asap_plan_selection::cost::cost_model::DefaultCostModel;
-use asap_types::ir::physical_export::{PhysicalASAPOperatorPayload};
 use asap_types::ir::operator::operator_properties::Reduction;
+use asap_types::ir::physical_export::PhysicalASAPOperatorPayload;
 use asap_types::ir::properties::CompositionOperator;
 use asap_types::ir::scalar::ColumnRef;
 use asap_types::ir::schema::DataType;
-use asap_types::ir::schema::{EntityIdentity, ExactKind, ExactParams, FieldDataType, GroupingStrategy, Schema, SketchAlgorithm, SketchKind, SketchParams, SketchStatistic, SummaryInputExpr, SummaryUpdate};
+use asap_types::ir::schema::{
+    EntityIdentity, ExactKind, ExactParams, FieldDataType, GroupingStrategy, Schema,
+    SketchAlgorithm, SketchKind, SketchParams, SketchStatistic, SummaryInputExpr, SummaryUpdate,
+};
 use asap_types::ir::{ASAPOp, NonASAPOp, Operator, OperatorNode, ScalarExpr};
 use asap_types::types::AccuracyTarget;
 

crates/integration-tests/tests/sql_to_post_asap.rs

@@ -22,14 +22,20 @@ use std::rc::Rc;
 use asap_frontend_sql::{lower_sql, lower_sql_dialect, SqlCatalog};
 use asap_integration_tests::post_asap::post_asap_dag;
 use asap_logical_optimizer::pass1::replacement::{retain_exact, RealizationError};
-use asap_logical_optimizer::{search_workload, ASAPStrategies, Replacement, ReplacementStrategy, ReplacementSubDAG, TargetSubDAG};
+use asap_logical_optimizer::{
+    search_workload, ASAPStrategies, Replacement, ReplacementStrategy, ReplacementSubDAG,
+    TargetSubDAG,
+};
 use asap_plan_selection::candidate_selection::global_selection;
 use asap_plan_selection::DefaultCostModel;
-use asap_types::ir::physical_export::{EdgeRole, PhysicalASAPNodeId, PhysicalASAPOperatorPayload};
 use asap_types::ir::operator::operator_properties::Reduction;
+use asap_types::ir::physical_export::{EdgeRole, PhysicalASAPNodeId, PhysicalASAPOperatorPayload};
 use asap_types::ir::scalar::ColumnRef;
 use asap_types::ir::schema::{DataType, Field, Schema};
-use asap_types::ir::schema::{ExactKind, ExactParams, FieldDataType, GroupingStrategy, SketchAlgorithm, SketchKind, SketchParams, SketchStatistic, SummaryUpdate};
+use asap_types::ir::schema::{
+    ExactKind, ExactParams, FieldDataType, GroupingStrategy, SketchAlgorithm, SketchKind,
+    SketchParams, SketchStatistic, SummaryUpdate,
+};
 use asap_types::ir::{ASAPOp, NonASAPOp, Operator, OperatorNode, Predicate, ScalarExpr};
 use asap_types::types::AccuracyTarget;
 use asap_types::workload::SqlDialect;