Method plan

datafusion/datasource/src/morsel/mocks.rs:522–609

(self: Box<Self>)

impl MorselPlanner for MockMorselPlanner {
    fn plan(self: Box<Self>) -> Result<Option<MorselPlan>> {
        let Self {
            observer,
            planner_name,
            mut steps,
        } = *self;

        observer.push(MorselEvent::PlannerCalled {
            planner_name: planner_name.clone(),
        });

        let Some(step) = steps.pop_front() else {
            return Ok(None);
        };

        match step {
            PlannerStep::Plan {
                morsels,
                ready_planners,
                pending_planner,
            } => {
                let mut ready_morsels = Vec::new();
                for MockMorselSpec {
                    morsel_id,
                    batch_ids,
                } in morsels
                {
                    observer.push(MorselEvent::MorselProduced {
                        planner_name: planner_name.clone(),
                        morsel_id,
                    });
                    ready_morsels.push(Box::new(MockMorsel::new(
                        observer.clone(),
                        morsel_id,
                        batch_ids,
                    )) as Box<dyn Morsel>);
                }

                let mut planners = child_planners(observer.clone(), ready_planners);
                if pending_planner.is_none() {
                    planners.extend(current_planner_continuation(
                        observer.clone(),
                        planner_name.clone(),
                        steps.clone(),
                    ));
                }

                let mut plan = MorselPlan::new()
                    .with_morsels(ready_morsels)
                    .with_planners(planners);

                if let Some(MockPendingPlanner {
                    io_future_id,
                    polls_to_resolve,
                    result,
                }) = pending_planner
                {
                    observer.push(MorselEvent::IoFutureCreated {