use std::time::Duration;

async fn main() -> Result<(), bonsaidb::local::Error> {

    let db = Database::open::<()>(StorageConfiguration::new("pubsub.bonsaidb")).await?;

    let subscriber = db.create_subscriber().await?;

    subscriber.subscribe_to("pong").await?;

    tokio::spawn(pinger(db.clone()));

    // Launch a task that receives "ping" messages and sends "pong" responses.

    tokio::spawn(ponger(db.clone()));

        let message = subscriber.receiver().recv_async().await?;

        let pings_remaining = message.payload::<usize>()?;

        println!(

            "<-- Received {}, pings remaining: {}",

            message.topic, pings_remaining

        );

        if pings_remaining == 0 {

            break;

        }

    println!("Received all pongs.");

    Ok(())

}

async fn pinger<P: PubSub>(pubsub: P) -> Result<(), bonsaidb::local::Error> {

    let mut ping_count = 0u32;

        ping_count += 1;

        println!("-> Sending ping {}", ping_count);

        pubsub.publish("ping", &ping_count).await?;

        sleep(Duration::from_millis(250)).await;

async fn ponger<P: PubSub>(pubsub: P) -> Result<(), bonsaidb::local::Error> {

    let subscriber = pubsub.create_subscriber().await?;

    subscriber.subscribe_to("ping").await?;

    let mut pings_remaining = NUMBER_OF_PONGS;

    println!(

        "Ponger started, waiting to respond to {} pings",

        pings_remaining

    );

    while pings_remaining > 0 {

        let message = subscriber.receiver().recv_async().await?;

        println!(

            "<- Received {}, id {}",

            message.topic,

            message.payload::<u32>()?

        pings_remaining -= 1;

        pubsub.publish("pong", &pings_remaining).await?;

    println!("Ponger finished.");

    Ok(())

}

#[test]

fn runs() {

    main().unwrap()

}