summaryrefslogtreecommitdiffstats
path: root/ipfs-api/examples/pubsub.rs
blob: 2d24278963cb1740284b16bf8b6777fb5711b01b (plain)
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
78
79
80
// Copyright 2017 rust-ipfs-api Developers
//
// Licensed under the Apache License, Version 2.0, <LICENSE-APACHE or
// http://apache.org/licenses/LICENSE-2.0> or the MIT license <LICENSE-MIT or
// http://opensource.org/licenses/MIT>, at your option. This file may not be
// copied, modified, or distributed except according to those terms.
//

#[macro_use]
extern crate futures;

use futures::{future, FutureExt, StreamExt, TryStreamExt};
use ipfs_api::IpfsClient;
use std::time::Duration;
use tokio::time;

static TOPIC: &'static str = "test";

fn get_client() -> IpfsClient {
    eprintln!("connecting to localhost:5001...");

    IpfsClient::default()
}

// Creates an Ipfs client, and simultaneously publishes and reads from a pubsub
// topic.
//
#[cfg_attr(feature = "actix", actix_rt::main)]
#[cfg_attr(feature = "hyper", tokio::main)]
async fn main() {
    eprintln!("note: ipfs must be run with the --enable-pubsub-experiment flag");

    let publish_client = get_client();

    // This block will execute a repeating function that sends
    // a message to the "test" topic.
    //
    let mut publish = time::interval(Duration::from_secs(1))
        .then(|_| future::ok(())) // Coerce the stream into a TryStream
        .try_for_each(|_| {
            eprintln!();
            eprintln!("publishing message...");

            publish_client.pubsub_pub(TOPIC, "Hello World!").boxed_local()
        })
        .boxed_local()
        .fuse();

    // This block will execute a future that suscribes to a topic,
    // and reads any incoming messages.
    //
    let mut subscribe = {
        let client = get_client();

        client
            .pubsub_sub(TOPIC, false)
            .take(5)
            .try_for_each(|msg| {
                eprintln!();
                eprintln!("received ({:?})", msg);

                future::ok(())
            })
            .fuse()
    };

    eprintln!();
    eprintln!("publish messages to ({})...", TOPIC);
    eprintln!("waiting for messages from ({})...", TOPIC);

    select! {
        res = publish => if let Err(e) = res {
            eprintln!("error publishing messages: {}", e);
        },
        res = subscribe => match res {
            Ok(_) => eprintln!("done reading messages..."),
            Err(e) => eprintln!("error reading messages: {}", e)
        },
    }
}