async_pub_sub 0.2.1

A library aiming at making async pub-sub easier in Rust
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

use async_pub_sub::{PublisherImpl, PublisherWrapper, Result, SubscriberImpl};

// #[derive(DerivePublisher)]
struct MultiPub {
    // #[publisher(i32)]
    publisher_a: PublisherImpl<i32>,
    // #[publisher(String)]
    publisher_b: PublisherImpl<String>,
}

impl PublisherWrapper<i32> for MultiPub {
    fn get_publisher(&self) -> &dyn async_pub_sub::Publisher<Message = i32> {
        &self.publisher_a
    }

    fn get_publisher_mut(&mut self) -> &mut dyn async_pub_sub::Publisher<Message = i32> {
        &mut self.publisher_a
    }
}

impl PublisherWrapper<String> for MultiPub {
    fn get_publisher(&self) -> &dyn async_pub_sub::Publisher<Message = String> {
        &self.publisher_b
    }

    fn get_publisher_mut(&mut self) -> &mut dyn async_pub_sub::Publisher<Message = String> {
        &mut self.publisher_b
    }
}

impl MultiPub {
    fn new() -> Self {
        Self {
            publisher_a: PublisherImpl::new("publisher", 1),
            publisher_b: PublisherImpl::new("publisher", 1),
        }
    }
}

#[tokio::test]
async fn test_multi_pub() -> Result<()> {
    let mut subscriber1 = SubscriberImpl::<i32>::new("subscriber1");
    let mut subscriber2 = SubscriberImpl::<String>::new("subscriber2");

    let mut publisher = MultiPub::new();
    subscriber1.subscribe_to(PublisherWrapper::<i32>::get_publisher_mut(&mut publisher))?;
    subscriber2.subscribe_to(PublisherWrapper::<String>::get_publisher_mut(
        &mut publisher,
    ))?;

    publisher.publish(42).await?;
    let message = subscriber1.receive().await;
    assert_eq!(message, 42);

    publisher.publish("toto".to_string()).await?;
    let message = subscriber2.receive().await;
    assert_eq!(message, "toto");

    Ok(())
}