speare 0.4.3

actor-like thin abstraction over tokio::task and flume channels
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

use speare::{Actor, Ctx, ExitReason, Node};
use tokio::task;

mod sync_vec;
use sync_vec::SyncVec;

struct Worker;

impl Actor for Worker {
    type Props = SyncVec<String>;
    type Msg = ();
    type Err = ();

    async fn init(ctx: &mut Ctx<Self>) -> Result<Self, Self::Err> {
        ctx.props().push("init".into()).await;
        Ok(Worker)
    }

    async fn exit(_: Option<Self>, _: ExitReason<Self>, ctx: &mut Ctx<Self>) {
        ctx.props().push("exit".into()).await;
    }
}

#[tokio::test]
async fn shutdown_stops_all_children() {
    // Arrange
    let mut node = Node::default();
    let h1 = node.actor::<Worker>(Default::default()).spawn();
    let h2 = node.actor::<Worker>(Default::default()).spawn();
    task::yield_now().await;
    assert!(h1.is_alive());
    assert!(h2.is_alive());

    // Act
    node.shutdown().await;

    // Assert
    assert!(!h1.is_alive());
    assert!(!h2.is_alive());
}

#[tokio::test]
async fn shutdown_calls_exit_on_all_children() {
    // Arrange
    let mut node = Node::default();
    let log: SyncVec<String> = Default::default();
    node.actor::<Worker>(log.clone()).spawn();
    node.actor::<Worker>(log.clone()).spawn();
    task::yield_now().await;

    // Act
    node.shutdown().await;

    // Assert
    let events = log.clone_vec().await;
    assert_eq!(events.iter().filter(|e| *e == "init").count(), 2);
    assert_eq!(events.iter().filter(|e| *e == "exit").count(), 2);
}

struct Parent;

impl Actor for Parent {
    type Props = SyncVec<String>;
    type Msg = ();
    type Err = ();

    async fn init(ctx: &mut Ctx<Self>) -> Result<Self, Self::Err> {
        ctx.actor::<Worker>(ctx.props().clone()).spawn();
        ctx.actor::<Worker>(ctx.props().clone()).spawn();
        Ok(Parent)
    }
}

#[tokio::test]
async fn shutdown_stops_nested_children() {
    // Arrange
    let mut node = Node::default();
    let log: SyncVec<String> = Default::default();
    node.actor::<Parent>(log.clone()).spawn();
    task::yield_now().await;

    // Act
    node.shutdown().await;

    // Assert - both grandchildren should have their exit called
    let events = log.clone_vec().await;
    assert_eq!(events.iter().filter(|e| *e == "init").count(), 2);
    assert_eq!(events.iter().filter(|e| *e == "exit").count(), 2);
}