1use crate::{
7 db::{
8 cursor::ContinuationSignature,
9 direction::Direction,
10 predicate::{MissingRowPolicy, Predicate},
11 query::{
12 builder::scalar_projection::render_scalar_projection_expr_plan_label,
13 plan::{
14 expr::{Expr, FieldId, normalize_bool_expr},
15 order_contract::DeterministicSecondaryOrderContract,
16 semantics::LogicalPushdownEligibility,
17 },
18 },
19 },
20 model::field::FieldKind,
21};
22
23#[derive(Clone, Copy, Debug, Eq, PartialEq)]
32pub enum QueryMode {
33 Load(LoadSpec),
34 Delete(DeleteSpec),
35}
36
37#[derive(Clone, Copy, Debug, Default, Eq, PartialEq)]
44pub struct LoadSpec {
45 pub(crate) limit: Option<u32>,
46 pub(crate) offset: u32,
47}
48
49impl LoadSpec {
50 #[must_use]
52 pub const fn limit(&self) -> Option<u32> {
53 self.limit
54 }
55
56 #[must_use]
58 pub const fn offset(&self) -> u32 {
59 self.offset
60 }
61}
62
63#[derive(Clone, Copy, Debug, Default, Eq, PartialEq)]
71pub struct DeleteSpec {
72 pub(crate) limit: Option<u32>,
73 pub(crate) offset: u32,
74}
75
76impl DeleteSpec {
77 #[must_use]
79 pub const fn limit(&self) -> Option<u32> {
80 self.limit
81 }
82
83 #[must_use]
85 pub const fn offset(&self) -> u32 {
86 self.offset
87 }
88}
89
90#[derive(Clone, Copy, Debug, Eq, PartialEq)]
95pub enum OrderDirection {
96 Asc,
97 Desc,
98}
99
100#[derive(Clone, Eq, PartialEq)]
110pub(crate) struct OrderTerm {
111 pub(crate) expr: Expr,
112 pub(crate) direction: OrderDirection,
113}
114
115impl OrderTerm {
116 #[must_use]
118 pub(crate) const fn new(expr: Expr, direction: OrderDirection) -> Self {
119 Self { expr, direction }
120 }
121
122 #[must_use]
124 pub(crate) fn field(field: impl Into<String>, direction: OrderDirection) -> Self {
125 Self::new(Expr::Field(FieldId::new(field.into())), direction)
126 }
127
128 #[must_use]
130 pub(crate) const fn expr(&self) -> &Expr {
131 &self.expr
132 }
133
134 #[must_use]
136 pub(crate) const fn direct_field(&self) -> Option<&str> {
137 let Expr::Field(field) = &self.expr else {
138 return None;
139 };
140
141 Some(field.as_str())
142 }
143
144 #[must_use]
146 pub(crate) fn rendered_label(&self) -> String {
147 render_scalar_projection_expr_plan_label(&self.expr)
148 }
149
150 #[must_use]
152 pub(crate) const fn direction(&self) -> OrderDirection {
153 self.direction
154 }
155}
156
157impl std::fmt::Debug for OrderTerm {
158 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
159 f.debug_struct("OrderTerm")
160 .field("label", &self.rendered_label())
161 .field("expr", &self.expr)
162 .field("direction", &self.direction)
163 .finish()
164 }
165}
166
167impl PartialEq<(String, OrderDirection)> for OrderTerm {
168 fn eq(&self, other: &(String, OrderDirection)) -> bool {
169 self.rendered_label() == other.0 && self.direction == other.1
170 }
171}
172
173impl PartialEq<OrderTerm> for (String, OrderDirection) {
174 fn eq(&self, other: &OrderTerm) -> bool {
175 self.0 == other.rendered_label() && self.1 == other.direction
176 }
177}
178
179#[must_use]
182pub(in crate::db) fn render_scalar_filter_expr_plan_label(expr: &Expr) -> String {
183 render_scalar_projection_expr_plan_label(&normalize_bool_expr(expr.clone()))
184}
185
186#[derive(Clone, Debug, Eq, PartialEq)]
193pub(crate) struct OrderSpec {
194 pub(crate) fields: Vec<OrderTerm>,
195}
196
197#[derive(Clone, Copy, Debug, Eq, PartialEq)]
203pub(crate) struct DeleteLimitSpec {
204 pub(crate) limit: Option<u32>,
205 pub(crate) offset: u32,
206}
207
208#[derive(Clone, Copy, Debug, Eq, PartialEq)]
217pub(crate) enum DistinctExecutionStrategy {
218 None,
219 PreOrdered,
220 HashMaterialize,
221}
222
223impl DistinctExecutionStrategy {
224 #[must_use]
226 pub(crate) const fn is_enabled(self) -> bool {
227 !matches!(self, Self::None)
228 }
229}
230
231#[derive(Clone, Debug, Eq, PartialEq)]
240pub(in crate::db) struct PlannerRouteProfile {
241 continuation_policy: ContinuationPolicy,
242 logical_pushdown_eligibility: LogicalPushdownEligibility,
243 secondary_order_contract: Option<DeterministicSecondaryOrderContract>,
244}
245
246impl PlannerRouteProfile {
247 #[must_use]
249 pub(in crate::db) const fn new(
250 continuation_policy: ContinuationPolicy,
251 logical_pushdown_eligibility: LogicalPushdownEligibility,
252 secondary_order_contract: Option<DeterministicSecondaryOrderContract>,
253 ) -> Self {
254 Self {
255 continuation_policy,
256 logical_pushdown_eligibility,
257 secondary_order_contract,
258 }
259 }
260
261 #[must_use]
264 pub(in crate::db) const fn seeded_unfinalized(is_grouped: bool) -> Self {
265 Self {
266 continuation_policy: ContinuationPolicy::new(true, true, !is_grouped),
267 logical_pushdown_eligibility: LogicalPushdownEligibility::new(false, is_grouped, false),
268 secondary_order_contract: None,
269 }
270 }
271
272 #[must_use]
274 pub(in crate::db) const fn continuation_policy(&self) -> &ContinuationPolicy {
275 &self.continuation_policy
276 }
277
278 #[must_use]
280 pub(in crate::db) const fn logical_pushdown_eligibility(&self) -> LogicalPushdownEligibility {
281 self.logical_pushdown_eligibility
282 }
283
284 #[must_use]
286 pub(in crate::db) const fn secondary_order_contract(
287 &self,
288 ) -> Option<&DeterministicSecondaryOrderContract> {
289 self.secondary_order_contract.as_ref()
290 }
291}
292
293#[derive(Clone, Copy, Debug, Eq, PartialEq)]
302pub(in crate::db) struct ContinuationPolicy {
303 requires_anchor: bool,
304 requires_strict_advance: bool,
305 is_grouped_safe: bool,
306}
307
308impl ContinuationPolicy {
309 #[must_use]
311 pub(in crate::db) const fn new(
312 requires_anchor: bool,
313 requires_strict_advance: bool,
314 is_grouped_safe: bool,
315 ) -> Self {
316 Self {
317 requires_anchor,
318 requires_strict_advance,
319 is_grouped_safe,
320 }
321 }
322
323 #[must_use]
325 pub(in crate::db) const fn requires_anchor(self) -> bool {
326 self.requires_anchor
327 }
328
329 #[must_use]
331 pub(in crate::db) const fn requires_strict_advance(self) -> bool {
332 self.requires_strict_advance
333 }
334
335 #[must_use]
337 pub(in crate::db) const fn is_grouped_safe(self) -> bool {
338 self.is_grouped_safe
339 }
340}
341
342#[derive(Clone, Copy, Debug, Eq, PartialEq)]
351pub(in crate::db) struct ExecutionShapeSignature {
352 continuation_signature: ContinuationSignature,
353}
354
355impl ExecutionShapeSignature {
356 #[must_use]
358 pub(in crate::db) const fn new(continuation_signature: ContinuationSignature) -> Self {
359 Self {
360 continuation_signature,
361 }
362 }
363
364 #[must_use]
366 pub(in crate::db) const fn continuation_signature(self) -> ContinuationSignature {
367 self.continuation_signature
368 }
369}
370
371#[derive(Clone, Debug, Eq, PartialEq)]
377pub(crate) struct PageSpec {
378 pub(crate) limit: Option<u32>,
379 pub(crate) offset: u32,
380}
381
382#[derive(Clone, Copy, Debug, Eq, PartialEq)]
392pub enum AggregateKind {
393 Count,
394 Sum,
395 Avg,
396 Exists,
397 Min,
398 Max,
399 First,
400 Last,
401}
402
403#[derive(Clone, Copy, Debug, Eq, PartialEq)]
413pub(crate) enum GlobalDistinctAggregateKind {
414 Count,
415 Sum,
416 Avg,
417}
418
419impl GlobalDistinctAggregateKind {}
420
421#[derive(Clone, Copy, Debug, Eq, PartialEq)]
432pub(crate) enum GroupedPlanAggregateFamily {
433 CountRowsOnly,
434 FieldTargetRows,
435 GenericRows,
436}
437
438impl GroupedPlanAggregateFamily {
439 #[must_use]
441 pub(crate) const fn code(self) -> &'static str {
442 match self {
443 Self::CountRowsOnly => "count_rows_only",
444 Self::FieldTargetRows => "field_target_rows",
445 Self::GenericRows => "generic_rows",
446 }
447 }
448}
449
450impl AggregateKind {
451 #[must_use]
453 pub(in crate::db) const fn canonical_label(self) -> &'static str {
454 match self {
455 Self::Count => "COUNT",
456 Self::Sum => "SUM",
457 Self::Avg => "AVG",
458 Self::Exists => "EXISTS",
459 Self::First => "FIRST",
460 Self::Last => "LAST",
461 Self::Min => "MIN",
462 Self::Max => "MAX",
463 }
464 }
465
466 #[must_use]
468 pub(crate) const fn is_count(self) -> bool {
469 matches!(self, Self::Count)
470 }
471
472 #[must_use]
474 pub(in crate::db) const fn is_sum(self) -> bool {
475 matches!(self, Self::Sum | Self::Avg)
476 }
477
478 #[must_use]
480 pub(in crate::db) const fn is_extrema(self) -> bool {
481 matches!(self, Self::Min | Self::Max)
482 }
483
484 #[must_use]
486 pub(in crate::db) const fn supports_field_target_v1(self) -> bool {
487 matches!(
488 self,
489 Self::Count | Self::Sum | Self::Avg | Self::Min | Self::Max
490 )
491 }
492
493 #[must_use]
495 pub(in crate::db) const fn requires_decoded_id(self) -> bool {
496 !matches!(self, Self::Count | Self::Sum | Self::Avg | Self::Exists)
497 }
498
499 #[must_use]
501 pub(in crate::db) const fn supports_grouped_distinct_v1(self) -> bool {
502 matches!(self, Self::Count | Self::Sum | Self::Avg)
503 }
504
505 #[must_use]
508 pub(in crate::db) const fn uses_grouped_distinct_value_dedup_v1(self) -> bool {
509 matches!(self, Self::Count | Self::Sum | Self::Avg)
510 }
511
512 #[must_use]
515 pub(in crate::db::query) const fn fingerprint_tag(self) -> u8 {
516 match self {
517 Self::Count => 0x01,
518 Self::Sum => 0x02,
519 Self::Exists => 0x03,
520 Self::Min => 0x04,
521 Self::Max => 0x05,
522 Self::First => 0x06,
523 Self::Last => 0x07,
524 Self::Avg => 0x08,
525 }
526 }
527
528 #[must_use]
530 pub(in crate::db) const fn global_distinct_kind(self) -> Option<GlobalDistinctAggregateKind> {
531 match self {
532 Self::Count => Some(GlobalDistinctAggregateKind::Count),
533 Self::Sum => Some(GlobalDistinctAggregateKind::Sum),
534 Self::Avg => Some(GlobalDistinctAggregateKind::Avg),
535 Self::Exists | Self::Min | Self::Max | Self::First | Self::Last => None,
536 }
537 }
538
539 #[must_use]
541 pub(in crate::db) const fn supports_global_distinct_without_group_keys(self) -> bool {
542 self.global_distinct_kind().is_some()
543 }
544
545 #[must_use]
547 pub(in crate::db) const fn grouped_plan_family(
548 self,
549 has_target_field: bool,
550 ) -> GroupedPlanAggregateFamily {
551 if has_target_field && self.supports_field_target_v1() {
552 GroupedPlanAggregateFamily::FieldTargetRows
553 } else {
554 GroupedPlanAggregateFamily::GenericRows
555 }
556 }
557
558 #[must_use]
560 pub(in crate::db) const fn supports_grouped_streaming_v1(
561 self,
562 has_target_field: bool,
563 distinct: bool,
564 ) -> bool {
565 if self.supports_field_target_v1() {
566 return !distinct && (self.is_count() || has_target_field);
567 }
568
569 !has_target_field && (!distinct || self.supports_grouped_distinct_v1())
570 }
571
572 #[must_use]
574 pub(crate) const fn extrema_direction(self) -> Option<Direction> {
575 match self {
576 Self::Min => Some(Direction::Asc),
577 Self::Max => Some(Direction::Desc),
578 Self::Count | Self::Sum | Self::Avg | Self::Exists | Self::First | Self::Last => None,
579 }
580 }
581
582 #[must_use]
584 pub(crate) const fn materialized_fold_direction(self) -> Direction {
585 match self {
586 Self::Min => Direction::Desc,
587 Self::Count
588 | Self::Sum
589 | Self::Avg
590 | Self::Exists
591 | Self::Max
592 | Self::First
593 | Self::Last => Direction::Asc,
594 }
595 }
596
597 #[must_use]
599 pub(crate) const fn supports_bounded_probe_hint(self) -> bool {
600 !self.is_count() && !self.is_sum()
601 }
602
603 #[must_use]
605 pub(crate) fn bounded_probe_fetch_hint(
606 self,
607 direction: Direction,
608 offset: usize,
609 page_limit: Option<usize>,
610 ) -> Option<usize> {
611 match self {
612 Self::Exists | Self::First => Some(offset.saturating_add(1)),
613 Self::Min if direction == Direction::Asc => Some(offset.saturating_add(1)),
614 Self::Max if direction == Direction::Desc => Some(offset.saturating_add(1)),
615 Self::Last => page_limit.map(|limit| offset.saturating_add(limit)),
616 Self::Count | Self::Sum | Self::Avg | Self::Min | Self::Max => None,
617 }
618 }
619
620 #[must_use]
622 pub(in crate::db) const fn explain_projection_mode_label(
623 self,
624 has_projected_field: bool,
625 covering_projection: bool,
626 ) -> &'static str {
627 if has_projected_field {
628 if covering_projection {
629 "field_idx"
630 } else {
631 "field_mat"
632 }
633 } else if matches!(self, Self::Min | Self::Max | Self::First | Self::Last) {
634 "entity_term"
635 } else {
636 "scalar_agg"
637 }
638 }
639
640 #[must_use]
642 pub(in crate::db) const fn supports_covering_existing_rows_terminal(self) -> bool {
643 matches!(self, Self::Count | Self::Exists)
644 }
645}
646
647#[derive(Clone, Debug)]
658pub(crate) struct GroupAggregateSpec {
659 pub(crate) kind: AggregateKind,
660 #[cfg(test)]
661 #[cfg(test)]
662 pub(crate) target_field: Option<String>,
663 pub(crate) input_expr: Option<Box<Expr>>,
664 pub(crate) filter_expr: Option<Box<Expr>>,
665 pub(crate) distinct: bool,
666}
667
668impl PartialEq for GroupAggregateSpec {
669 fn eq(&self, other: &Self) -> bool {
670 self.identity() == other.identity() && self.filter_expr == other.filter_expr
671 }
672}
673
674impl Eq for GroupAggregateSpec {}
675
676impl GroupedPlanAggregateFamily {
677 #[must_use]
679 pub(in crate::db) fn from_grouped_aggregates(aggregates: &[GroupAggregateSpec]) -> Self {
680 if matches!(aggregates, [aggregate] if aggregate.identity().is_count_rows_only()) {
681 return Self::CountRowsOnly;
682 }
683
684 if aggregates.iter().all(|aggregate| {
685 aggregate
686 .kind()
687 .grouped_plan_family(aggregate.target_field().is_some())
688 == Self::FieldTargetRows
689 }) {
690 return Self::FieldTargetRows;
691 }
692
693 Self::GenericRows
694 }
695}
696
697#[derive(Clone, Debug)]
708pub(crate) struct FieldSlot {
709 pub(crate) index: usize,
710 pub(crate) field: String,
711 pub(crate) kind: Option<FieldKind>,
712}
713
714impl PartialEq for FieldSlot {
715 fn eq(&self, other: &Self) -> bool {
716 self.index == other.index && self.field == other.field
717 }
718}
719
720impl Eq for FieldSlot {}
721
722#[derive(Clone, Copy, Debug, Eq, PartialEq)]
731pub(crate) struct GroupedExecutionConfig {
732 pub(crate) max_groups: u64,
733 pub(crate) max_group_bytes: u64,
734}
735
736#[derive(Clone, Debug, Eq, PartialEq)]
745pub(crate) struct GroupSpec {
746 pub(crate) group_fields: Vec<FieldSlot>,
747 pub(crate) aggregates: Vec<GroupAggregateSpec>,
748 pub(crate) execution: GroupedExecutionConfig,
749}
750
751#[derive(Clone, Debug, Eq, PartialEq)]
772pub(crate) struct ScalarPlan {
773 pub(crate) mode: QueryMode,
775
776 pub(crate) filter_expr: Option<Expr>,
778
779 pub(crate) predicate_covers_filter_expr: bool,
781
782 pub(crate) predicate: Option<Predicate>,
784
785 pub(crate) order: Option<OrderSpec>,
787
788 pub(crate) distinct: bool,
790
791 pub(crate) delete_limit: Option<DeleteLimitSpec>,
793
794 pub(crate) page: Option<PageSpec>,
796
797 pub(crate) consistency: MissingRowPolicy,
799}
800
801#[derive(Clone, Debug, Eq, PartialEq)]
809pub(crate) struct GroupPlan {
810 pub(crate) scalar: ScalarPlan,
811 pub(crate) group: GroupSpec,
812 pub(crate) having_expr: Option<Expr>,
813}
814
815#[derive(Clone, Debug, Eq, PartialEq)]
825pub(crate) enum LogicalPlan {
826 Scalar(ScalarPlan),
827 Grouped(GroupPlan),
828}