serviceless 0.6.1

An simple actor model in rust, like actix
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

use std::future::Future;

use crate::runtime::{Spawner, TaskHandle};
use async_trait::async_trait;
use tokio::task::{AbortHandle, JoinSet};

use super::Error;

pub struct TokioSpawner<T> {
    pub(crate) tasks: JoinSet<T>,
}

#[async_trait]
impl<T> Spawner<T> for TokioSpawner<T>
where
    T: Send + 'static,
{
    type Error = Error;

    type TaskHandle = TokioTaskHandle;

    fn len(&self) -> usize {
        self.tasks.len()
    }

    fn is_empty(&self) -> bool {
        self.tasks.is_empty()
    }

    fn spawn<F>(&mut self, task: F) -> Self::TaskHandle
    where
        F: Future<Output = T> + Send + 'static,
        T: Send,
    {
        let handle = self.tasks.spawn(task);
        TokioTaskHandle { handle }
    }

    fn spawn_blocking<F>(&mut self, f: F) -> Self::TaskHandle
    where
        F: FnOnce() -> T + Send + 'static,
        T: Send,
    {
        let handle = self.tasks.spawn_blocking(f);
        TokioTaskHandle { handle }
    }

    async fn join_next(&mut self) -> Option<Result<T, Self::Error>> {
        if let Some(result) = self.tasks.join_next().await {
            let res = result.map_err(Error::JoinError);

            Some(res)
        } else {
            None
        }
    }

    fn try_join_next(&mut self) -> Option<Result<T, Self::Error>> {
        self.tasks
            .try_join_next()
            .map(|result| result.map_err(Error::JoinError))
    }
}

pub struct TokioTaskHandle {
    pub(crate) handle: AbortHandle,
}

impl TaskHandle for TokioTaskHandle {
    fn abort(&self) {
        self.handle.abort();
    }

    fn is_finished(&self) -> bool {
        self.handle.is_finished()
    }
}