use nebkor_maelstrom::{mk_payload, Body, Message, Node, Runner}; use std::{collections::HashSet, thread, time::Duration}; fn main() { let node = SingleNodeBroadcast::default(); let on_init = |runner: &Runner| { let backdoor = runner.get_backdoor(); let msg = Message::from_body(Body::from_type("tony_danza")); thread::spawn(move || { // send a `tony_danza` message to the broadcast node every 100ms thread::sleep(Duration::from_millis(100)); backdoor.send(msg.clone()).unwrap(); }); }; let on_init = Box::new(on_init); let runner = Runner::new(node); runner.run(Some(on_init)); } #[derive(Clone, Default)] struct SingleNodeBroadcast { pub store: HashSet, } impl Node for SingleNodeBroadcast { fn handle(&mut self, runner: &Runner, req: Message) { let typ = req.typ(); match typ { "broadcast" => { let val = req.body.payload.get("message").unwrap().as_i64().unwrap(); self.store.insert(val); // maybe before replying with "broadcast_ok" maybe consider telling your neighbors // about what you heard? maybe? runner.reply(&req, Body::from_type("broadcast_ok")); } "read" => { let vals = self.store.iter().cloned().collect::>(); let payload = mk_payload(&[("messages", vals.into())]); let body = Body::from_type("read_ok").with_payload(payload); runner.reply(&req, body); } "topology" => { let body = Body::from_type("topology_ok"); runner.reply(&req, body); } "tony_danza" => { eprintln!("Eh, oh, oh, eh!"); } _ => {} } } }