crb-superagent 0.0.29

CRB | Composable Runtime Blocks | Agent Extensions
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

use super::{Supervisor, SupervisorContext};
use crb_agent::{Address, Agent, Context, RunAgent};
use crb_runtime::{InteractiveRuntime, Runtime};

pub struct Stacker<S: Supervisor> {
    scheduled: Vec<(Box<dyn Runtime>, S::GroupBy)>,
}

impl<S: Supervisor> Default for Stacker<S> {
    fn default() -> Self {
        Self::new()
    }
}

impl<S: Supervisor> Stacker<S> {
    pub fn new() -> Self {
        Self {
            scheduled: Vec::new(),
        }
    }

    pub fn schedule<A>(&mut self, agent: A, group: S::GroupBy) -> Address<A>
    where
        A: Agent,
        A::Context: Default,
    {
        let runtime = RunAgent::<A>::new(agent);
        let addr = runtime.address();
        self.scheduled.push((Box::new(runtime), group));
        addr
    }

    pub fn spawn_scheduled(&mut self, ctx: &mut Context<S>)
    where
        S::Context: SupervisorContext<S>,
    {
        let runtimes: Vec<_> = self.scheduled.drain(..).collect();
        for (runtime, group) in runtimes {
            ctx.session().spawn_trackable(runtime, group);
        }
    }
}