use crate::{
CollectionNode, GraphError, GraphResult, MapDiff, PlanContext, PlanError, ResourcePlan,
SetDiff, Transaction, resource::ResourcePlanner,
};
impl<C: 'static, O> Transaction<'_, C, O>
where
O: Clone + PartialEq + 'static,
{
pub fn map_resource_planner<K, V>(
&mut self,
collection: CollectionNode<K, V>,
scope: crate::ScopeId,
planner: impl for<'ctx> Fn(
&PlanContext<'ctx, MapDiff<K, V>>,
) -> Result<ResourcePlan<C>, PlanError>
+ 'static,
) -> GraphResult<()>
where
K: Clone + Ord + 'static,
V: Clone + PartialEq + 'static,
{
self.ensure_open()?;
self.working.require_scope_open(scope)?;
self.working
.validate_map_collection_read::<K, V>(collection.id())?;
let resource_planner = ResourcePlanner::new(collection.id(), scope, move |graph| {
let Some(diff) = graph.map_diff(collection)? else {
return Ok(ResourcePlan::new());
};
let ctx = PlanContext::new(scope, diff);
planner(&ctx).map_err(|error| GraphError::PlanFailed(scope, error))
});
self.staged_resource_planner_collections
.push(collection.id());
self.working.resource_planners.push(resource_planner);
self.graph_mutated = true;
Ok(())
}
pub fn set_resource_planner<K>(
&mut self,
collection: CollectionNode<K, ()>,
scope: crate::ScopeId,
planner: impl for<'ctx> Fn(&PlanContext<'ctx, SetDiff<K>>) -> Result<ResourcePlan<C>, PlanError>
+ 'static,
) -> GraphResult<()>
where
K: Clone + Ord + 'static,
{
self.ensure_open()?;
self.working.require_scope_open(scope)?;
self.working
.validate_set_collection_read::<K>(collection.id())?;
let resource_planner = ResourcePlanner::new(collection.id(), scope, move |graph| {
let Some(diff) = graph.set_diff(collection)? else {
return Ok(ResourcePlan::new());
};
let ctx = PlanContext::new(scope, diff);
planner(&ctx).map_err(|error| GraphError::PlanFailed(scope, error))
});
self.staged_resource_planner_collections
.push(collection.id());
self.working.resource_planners.push(resource_planner);
self.graph_mutated = true;
Ok(())
}
}