bevy/bevy_app/src/schedule_runner.rs

use super::{App, AppBuilder, AppPlugin};
use std::{thread, time::Duration};

#[derive(Copy, Clone, Debug)]
pub enum RunMode {
    Loop { wait: Option<Duration> },
    Once,
}

impl Default for RunMode {
    fn default() -> Self {
        RunMode::Loop { wait: None }
    }
}

#[derive(Default)]
pub struct ScheduleRunnerPlugin {
    pub run_mode: RunMode,
}

impl AppPlugin for ScheduleRunnerPlugin {
    fn build(&self, app: &mut AppBuilder) {
        let run_mode = self.run_mode;
        app.set_runner(move |mut app: App| match run_mode {
            RunMode::Once => {
                if let Some(ref mut schedule) = app.schedule {
                    schedule.execute(&mut app.world, &mut app.resources);
                }
            }
            RunMode::Loop { wait } => loop {
                if let Some(ref mut schedule) = app.schedule {
                    schedule.execute(&mut app.world, &mut app.resources);
                }
                if let Some(wait) = wait {
                    thread::sleep(wait);
                }
            },
        });
    }
}