gossip-glomers/02-unique-id/src/main.rs

use serde::{Deserialize, Serialize};
use std::io::{self, BufRead, Write};
use uuid7::Uuid;

#[derive(Serialize, Deserialize, Debug)]
struct Message {
    src: String,
    dest: String,
    body: Body,
}

#[derive(Serialize, Deserialize, Debug)]
#[serde(tag = "type")]
enum Body {
    #[serde(rename = "init")]
    Init {
        msg_id: u64,
        node_id: String,
        node_ids: Vec<String>,
    },
    #[serde(rename = "init_ok")]
    InitOk { in_reply_to: u64 },
    #[serde(rename = "generate")]
    Generate {
        msg_id: u64,
    },
    #[serde(rename = "generate_ok")]
    GenerateOk {
        msg_id: u64,
        in_reply_to: u64,
        id: Uuid,
    },
    #[serde(rename = "error")]
    Error {
        in_reply_to: u64,
        code: u64,
        text: String,
    },
}

fn main() {
    let stdin = io::stdin();
    let mut stdout = io::stdout();
    let mut node_id = String::new();

    for line in stdin.lock().lines() {
        let input: Message = serde_json::from_str(&line.unwrap()).unwrap();
        match input.body {
            Body::Init {
                msg_id,
                node_id: id,
                ..
            } => {
                node_id = id;
                let output = Message {
                    src: node_id.clone(),
                    dest: input.src,
                    body: Body::InitOk {
                        in_reply_to: msg_id,
                    },
                };
                let output_json = serde_json::to_string(&output).unwrap();
                writeln!(stdout, "{}", output_json).unwrap();
                stdout.flush().unwrap();
            }
            Body::Generate { msg_id, .. } => {
                let uuid = uuid7::uuid7();
                let output = Message {
                    src: node_id.clone(),
                    dest: input.src,
                    body: Body::GenerateOk {
                        msg_id,
                        in_reply_to: msg_id,
                        id: uuid,
                    },
                };
                let output_json = serde_json::to_string(&output).unwrap();
                writeln!(stdout, "{}", output_json).unwrap();
                stdout.flush().unwrap();
            }
            Body::Error {
                in_reply_to,
                code,
                text,
            } => {
                eprintln!(
                    "Error received (in_reply_to: {}, code: {}, text: {})",
                    in_reply_to, code, text
                );
            }
            _ => (),
        }
    }
}