summaryrefslogtreecommitdiff
path: root/ecs/examples/error_handling.rs
blob: dc34c5f79490686bcd08a63eb2134dbfa7d9adf6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
use ecs::error::Error;
use ecs::event::component::{Changed, EventMatchExt};
use ecs::pair::Pair;
use ecs::phase::UPDATE;
use ecs::query::Query;
use ecs::system::observer::Observe;
use ecs::{Component, World, error};
use tracing::level_filters::LevelFilter;
use tracing_subscriber::EnvFilter;
use tracing_subscriber::fmt::time::ChronoLocal;
use tracing_subscriber::layer::SubscriberExt;
use tracing_subscriber::util::SubscriberInitExt;

#[derive(Component)]
struct State
{
    value: u32,
}

fn do_something_fallible(query: Query<(&mut State,)>) -> Result<(), Error>
{
    for (mut state,) in &query {
        state.value += 1;

        state.set_changed();

        if state.value > 3 {
            return Err(error!("Invalid state value {}", state.value));
        }
    }

    Ok(())
}

fn handle_state_changed(observe: Observe<Pair<Changed, State>>) -> Result<(), Error>
{
    for evt_match in &observe {
        let state = evt_match.get_ent_target_comp();

        if state.value > 3 {
            return Err(error!("Invalid state value {}", state.value));
        }

        tracing::info!("State has valid value {}", state.value);
    }

    Ok(())
}

fn main()
{
    tracing_subscriber::registry()
        .with(
            tracing_subscriber::fmt::layer()
                .with_timer(ChronoLocal::new("%T%.6f".to_string())),
        )
        .with(
            EnvFilter::builder()
                .with_default_directive(LevelFilter::DEBUG.into())
                .from_env()
                .unwrap(),
        )
        .init();

    let mut world = World::new();

    world.set_err_handler(ecs::error::err_handler_log_error);

    world.create_entity((State { value: 0 },));

    world.register_system(*UPDATE, do_something_fallible);

    world.register_observer(handle_state_changed);

    world.step();
    world.step();
    world.step();
    world.step();
}