task-supervisor 0.4.1

Supervisor for long-lived tokio tasks
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

mod common;

use common::HealthyTask;
use std::{sync::Arc, time::Duration};
use task_supervisor::SupervisorBuilder;
use tokio::time::pause;

#[tokio::test]
async fn test_consecutive_waits() {
    pause();

    let handle = SupervisorBuilder::new().build().run();

    let run_flag = Arc::new(std::sync::atomic::AtomicBool::new(true));
    let task = HealthyTask {
        run_flag: run_flag.clone(),
    };
    handle.add_task("task", task).unwrap();

    let handle_clone = handle.clone();
    tokio::spawn(async move {
        tokio::time::sleep(Duration::from_millis(5)).await;
        handle_clone.shutdown().unwrap();
    });

    let result1 = handle.wait().await;
    let result2 = handle.wait().await;
    let result3 = handle.wait().await;

    assert!(result1.is_ok());
    assert!(result2.is_ok());
    assert!(result3.is_ok());
}

#[tokio::test]
async fn test_parallel_waits() {
    pause();

    let handle = SupervisorBuilder::new().build().run();

    let run_flag = Arc::new(std::sync::atomic::AtomicBool::new(true));
    let task = HealthyTask {
        run_flag: run_flag.clone(),
    };
    handle.add_task("task", task).unwrap();

    let handle_clone = handle.clone();
    tokio::spawn(async move {
        tokio::time::sleep(Duration::from_millis(5)).await;
        handle_clone.shutdown().unwrap();
    });

    let handle1 = handle.clone();
    let handle2 = handle.clone();
    let handle3 = handle.clone();

    let (result1, result2, result3) = tokio::join!(handle1.wait(), handle2.wait(), handle3.wait());

    assert!(result1.is_ok());
    assert!(result2.is_ok());
    assert!(result3.is_ok());
}