crb 0.0.38

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

use anyhow::Result;
use async_trait::async_trait;
use crb::agent::{Address, Agent, Context, DoAsync, ManagedContext, Next, OnEvent, Standalone};
use crb::superagent::StreamSession;
use futures::stream::{StreamExt, once};

struct TestStreams {
    counter: usize,
}

impl TestStreams {
    pub fn new() -> Self {
        Self { counter: 0 }
    }
}

impl Standalone for TestStreams {}

impl Agent for TestStreams {
    type Context = StreamSession<Self>;
    type Link = Address<Self>;

    fn begin(&mut self) -> Next<Self> {
        Next::do_async(Consume)
    }
}

struct Consume;

#[async_trait]
impl DoAsync<Consume> for TestStreams {
    async fn handle(&mut self, _: Consume, ctx: &mut Context<Self>) -> Result<Next<Self>> {
        let stream = once(async { () }).boxed();
        ctx.consume(stream);
        Ok(Next::events())
    }
}

#[async_trait]
impl OnEvent<()> for TestStreams {
    async fn handle(&mut self, _: (), ctx: &mut Context<Self>) -> Result<()> {
        self.counter += 1;
        if self.counter < 5 {
            ctx.do_next(Next::do_async(Consume));
        } else {
            ctx.shutdown();
        }
        Ok(())
    }
}

#[tokio::test]
async fn test_streams() -> Result<()> {
    let mut addr = TestStreams::new().spawn();
    addr.join().await?;
    Ok(())
}