yew-agent 0.5.0

Agents for Yew
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

use super::traits::Oneshot;
use crate::worker::{HandlerId, Worker, WorkerDestroyHandle, WorkerScope};

pub(crate) enum Message<T>
where
    T: Oneshot,
{
    Finished {
        handler_id: HandlerId,
        output: T::Output,
    },
}

pub(crate) struct OneshotWorker<T>
where
    T: 'static + Oneshot,
{
    running_tasks: usize,
    destruct_handle: Option<WorkerDestroyHandle<Self>>,
}

impl<T> Worker for OneshotWorker<T>
where
    T: 'static + Oneshot,
{
    type Input = T::Input;
    type Message = Message<T>;
    type Output = T::Output;

    fn create(_scope: &WorkerScope<Self>) -> Self {
        Self {
            running_tasks: 0,
            destruct_handle: None,
        }
    }

    fn update(&mut self, scope: &WorkerScope<Self>, msg: Self::Message) {
        let Message::Finished { handler_id, output } = msg;

        self.running_tasks -= 1;

        scope.respond(handler_id, output);

        if self.running_tasks == 0 {
            self.destruct_handle = None;
        }
    }

    fn received(&mut self, scope: &WorkerScope<Self>, input: Self::Input, handler_id: HandlerId) {
        self.running_tasks += 1;

        scope.send_future(async move {
            let output = T::create(input).await;

            Message::Finished { handler_id, output }
        });
    }

    fn destroy(&mut self, _scope: &WorkerScope<Self>, destruct: WorkerDestroyHandle<Self>) {
        if self.running_tasks > 0 {
            self.destruct_handle = Some(destruct);
        }
    }
}