use crate::db::{
access::validate_access_runtime_invariants_with_schema,
query::plan::{
AccessPlannedQuery, LogicalPlan, OrderSpec, ScalarPlan,
validate::{
GroupPlanError, PlanError,
grouped::{
validate_group_cursor_constraints, validate_group_policy, validate_group_structure,
validate_projection_expr_types,
},
order::{
validate_no_duplicate_non_pk_order_fields, validate_order,
validate_primary_key_tie_break,
},
validate_plan_shape,
},
},
query::predicate::validate_predicate,
schema::SchemaInfo,
};
fn validate_accepted_access_structure_for_plan(
schema: &SchemaInfo,
plan: &AccessPlannedQuery,
) -> Result<(), PlanError> {
validate_access_runtime_invariants_with_schema(schema, &plan.access).map_err(PlanError::from)
}
pub(in crate::db::query) fn validate_query_semantics_with_schema(
schema: &SchemaInfo,
plan: &AccessPlannedQuery,
) -> Result<(), PlanError> {
let logical = plan.scalar_plan();
let projection = plan.projection_spec_with_schema(schema);
validate_scalar_plan_semantic_gates(
schema,
logical,
plan,
validate_order,
validate_accepted_access_structure_for_plan,
true,
)?;
validate_projection_expr_types(schema, &projection)?;
Ok(())
}
pub(in crate::db::query) fn validate_group_query_semantics_with_schema(
schema: &SchemaInfo,
plan: &AccessPlannedQuery,
) -> Result<(), PlanError> {
let (logical, group, having_expr) = match &plan.logical {
LogicalPlan::Grouped(grouped) => (
&grouped.scalar,
&grouped.group,
grouped.having_expr.as_ref(),
),
LogicalPlan::Scalar(_) => {
return Err(PlanError::from(
GroupPlanError::grouped_logical_plan_required(),
));
}
};
let projection = plan.projection_spec_with_schema(schema);
validate_scalar_plan_semantic_gates(
schema,
logical,
plan,
validate_order,
validate_accepted_access_structure_for_plan,
false,
)?;
validate_group_structure(schema, group, &projection, having_expr)?;
validate_group_policy(schema, logical, group, having_expr)?;
validate_group_cursor_constraints(logical, group)?;
validate_projection_expr_types(schema, &projection)?;
Ok(())
}
fn validate_scalar_plan_semantic_gates<FOrder, FAccess>(
schema: &SchemaInfo,
logical: &ScalarPlan,
plan: &AccessPlannedQuery,
validate_order_fn: FOrder,
validate_access_fn: FAccess,
require_primary_key_tie_break: bool,
) -> Result<(), PlanError>
where
FOrder: Fn(&SchemaInfo, &OrderSpec) -> Result<(), PlanError>,
FAccess: Fn(&SchemaInfo, &AccessPlannedQuery) -> Result<(), PlanError>,
{
if let Some(predicate) = &logical.predicate {
validate_predicate(schema, predicate)?;
}
if let Some(order) = &logical.order {
validate_order_fn(schema, order)?;
validate_no_duplicate_non_pk_order_fields(schema, order)?;
if require_primary_key_tie_break {
validate_primary_key_tie_break(schema, order)?;
}
}
validate_access_fn(schema, plan)?;
validate_plan_shape(&plan.logical)?;
Ok(())
}