bft-crdt-experiment/side-node/src/websocket/mod.rs

use bft_crdt_derive::add_crdt_fields;
use bft_json_crdt::{
    json_crdt::{BaseCrdt, CrdtNode, IntoCrdtNode},
    keypair::make_keypair,
    list_crdt::ListCrdt,
    op::ROOT_ID,
};
use serde::{Deserialize, Serialize};
use tokio::time::{self};
use websockets::WebSocket;

/// Starts a websocket and periodically sends a BFT-CRDT message to the websocket server
pub(crate) async fn start() -> Result<(), websockets::WebSocketError> {
    println!("connecting to websocket at ws://localhost:8080/");
    let mut ws = WebSocket::connect("ws://localhost:8080/").await?;

    // generate a placeholder transaction
    let json = generate_transaction().unwrap();

    // set up a new BFT-CRDT
    let keys = make_keypair();
    let mut bft_crdt = BaseCrdt::<ListExample>::new(&keys);

    // next job is to keep adding to this guy
    let _a = bft_crdt.doc.list.insert(ROOT_ID, 'a').sign(&keys);

    let mut interval = every_two_seconds();
    loop {
        interval.tick().await;
        println!("Sending: {}", json);
        ws.send_text(json.clone()).await?;

        // TODO:  bft_crdt.apply() changes in here when we receive socket input from other nodes

        let msg = ws.receive().await?;
        println!("Received: {:?}", msg);
    }
}

fn every_two_seconds() -> time::Interval {
    time::interval(time::Duration::from_secs(2))
}

#[add_crdt_fields]
#[derive(Clone, CrdtNode)]
struct ListExample {
    list: ListCrdt<char>, // switch to Transaction as soon as char is working
}

/// A fake Transaction struct we can use as a simulated payload
#[derive(Serialize, Deserialize)]
struct Transaction {
    from: String,
    to: String,
    amount: u64,
}

fn generate_transaction() -> serde_json::Result<String> {
    let transaction = Transaction {
        from: "Alice".to_string(),
        to: "Bob".to_string(),
        amount: 100,
    };

    let json = serde_json::to_string(&transaction).unwrap();
    Ok(json)
}