use crate::db::{
cursor::CursorPlanError,
predicate::{
CompareOp,
grouped_having_compare_op_supported as predicate_grouped_having_compare_op_supported,
},
query::plan::{GroupHavingSpec, GroupPlan},
};
#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub(crate) enum GroupedCursorPolicyViolation {
ContinuationRequiresLimit,
GlobalDistinctContinuationUnsupported,
}
impl GroupedCursorPolicyViolation {
#[must_use]
pub(in crate::db) const fn invariant_message(self) -> &'static str {
match self {
Self::ContinuationRequiresLimit => {
"grouped continuation cursors require an explicit LIMIT"
}
Self::GlobalDistinctContinuationUnsupported => {
"global DISTINCT grouped aggregates do not support continuation cursors"
}
}
}
#[must_use]
pub(in crate::db) fn into_cursor_plan_error(self) -> CursorPlanError {
CursorPlanError::continuation_cursor_invariant(self.invariant_message())
}
}
#[must_use]
pub(crate) const fn grouped_having_compare_op_supported(op: CompareOp) -> bool {
predicate_grouped_having_compare_op_supported(op)
}
#[must_use]
pub(in crate::db) fn grouped_cursor_policy_violation(
grouped: &GroupPlan,
cursor_present: bool,
) -> Option<GroupedCursorPolicyViolation> {
if !cursor_present {
return None;
}
if grouped
.scalar
.page
.as_ref()
.and_then(|page| page.limit)
.is_none()
{
return Some(GroupedCursorPolicyViolation::ContinuationRequiresLimit);
}
if grouped.is_global_distinct_aggregate_without_group_keys() {
return Some(GroupedCursorPolicyViolation::GlobalDistinctContinuationUnsupported);
}
None
}
pub(in crate::db::query::plan::semantics) fn grouped_having_streaming_compatible(
having: Option<&GroupHavingSpec>,
) -> bool {
having.is_none_or(|having| {
having
.clauses()
.iter()
.all(|clause| grouped_having_compare_op_supported(clause.op()))
})
}