compio-actor 0.1.0

Actors for Compio
Documentation 
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
80
81
82
83
84
85
86
87
88
89
90
91

use std::{convert::Infallible, io};

use compio_actor::{
    Actor, ActorExit, Call, Cluster, Handler, Mailbox, supervisor::SupervisionEvent,
};

struct Worker;
struct Parent;

#[derive(Debug)]
struct Events;

impl Actor for Worker {
    type Arguments = ();
    type Error = Infallible;
    type State = ();

    async fn pre_start(
        &self,
        _myself: &Mailbox<Self>,
        (): Self::Arguments,
    ) -> Result<Self::State, Self::Error> {
        Ok(())
    }
}

impl Actor for Parent {
    type Arguments = ();
    type Error = Infallible;
    type State = Vec<&'static str>;

    async fn pre_start(
        &self,
        _myself: &Mailbox<Self>,
        (): Self::Arguments,
    ) -> Result<Self::State, Self::Error> {
        Ok(Vec::new())
    }
}

impl Handler<SupervisionEvent<Worker>> for Parent {
    async fn handle(
        &self,
        _myself: &Mailbox<Self>,
        event: SupervisionEvent<Worker>,
        state: &mut Self::State,
    ) -> Result<(), Self::Error> {
        match event {
            SupervisionEvent::ActorStarted(worker) => {
                state.push("started");
                worker.stop();
            }
            SupervisionEvent::ActorTerminated(_) => state.push("terminated"),
            SupervisionEvent::ActorFailed(_) => state.push("failed"),
        }
        Ok(())
    }
}

impl Handler<Call<Events, Vec<&'static str>>> for Parent {
    async fn handle(
        &self,
        _myself: &Mailbox<Self>,
        call: Call<Events, Vec<&'static str>>,
        state: &mut Self::State,
    ) -> Result<(), Self::Error> {
        call.reply(state.clone()).ok();
        Ok(())
    }
}

#[compio_macros::main]
async fn main() -> io::Result<()> {
    let cluster = Cluster::new()?;
    let (parent, parent_handle) = cluster.spawn(|| Parent, ()).await.unwrap();
    let (_worker, worker_handle) = cluster
        .spawn(|| Worker, ())
        .with_supervisor(&parent)
        .await
        .unwrap();

    assert_eq!(worker_handle.await.unwrap(), ActorExit::Stopped);
    assert_eq!(
        parent.call(Events).await.unwrap(),
        ["started", "terminated"]
    );

    parent.stop();
    assert_eq!(parent_handle.await.unwrap(), ActorExit::Stopped);
    cluster.join().await
}