use crate::{
db::{
access::validate_access_structure_model as validate_access_structure_model_shared,
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,
},
},
schema::{SchemaInfo, validate},
},
model::entity::EntityModel,
};
pub(crate) fn validate_query_semantics(
schema: &SchemaInfo,
model: &EntityModel,
plan: &AccessPlannedQuery,
) -> Result<(), PlanError> {
let logical = plan.scalar_plan();
let projection = plan.projection_spec(model);
validate_plan_core(
schema,
model,
logical,
plan,
validate_order,
|schema, model, plan| {
validate_access_structure_model_shared(schema, model, &plan.access)
.map_err(PlanError::from)
},
)?;
validate_projection_expr_types(schema, &projection)?;
Ok(())
}
pub(crate) fn validate_group_query_semantics(
schema: &SchemaInfo,
model: &EntityModel,
plan: &AccessPlannedQuery,
) -> Result<(), PlanError> {
let (logical, group, having) = match &plan.logical {
LogicalPlan::Grouped(grouped) => (&grouped.scalar, &grouped.group, grouped.having.as_ref()),
LogicalPlan::Scalar(_) => {
return Err(PlanError::from(
GroupPlanError::grouped_logical_plan_required(),
));
}
};
let projection = plan.projection_spec(model);
validate_plan_core(
schema,
model,
logical,
plan,
validate_order,
|schema, model, plan| {
validate_access_structure_model_shared(schema, model, &plan.access)
.map_err(PlanError::from)
},
)?;
validate_group_structure(schema, model, group, &projection, having)?;
validate_group_policy(schema, logical, group, having)?;
validate_group_cursor_constraints(logical, group)?;
validate_projection_expr_types(schema, &projection)?;
Ok(())
}
fn validate_plan_core<FOrder, FAccess>(
schema: &SchemaInfo,
model: &EntityModel,
logical: &ScalarPlan,
plan: &AccessPlannedQuery,
validate_order_fn: FOrder,
validate_access_fn: FAccess,
) -> Result<(), PlanError>
where
FOrder: Fn(&SchemaInfo, &OrderSpec) -> Result<(), PlanError>,
FAccess: Fn(&SchemaInfo, &EntityModel, &AccessPlannedQuery) -> Result<(), PlanError>,
{
if let Some(predicate) = &logical.predicate {
validate(schema, predicate)?;
}
if let Some(order) = &logical.order {
validate_order_fn(schema, order)?;
validate_no_duplicate_non_pk_order_fields(model, order)?;
validate_primary_key_tie_break(model, order)?;
}
validate_access_fn(schema, model, plan)?;
validate_plan_shape(&plan.logical)?;
Ok(())
}