ishan/distributed-systems-flyio
1
0
Fork 0
Code

fixed bugs in parsing inputs and responding to them

Browse Source
This commit is contained in:
Ishan Jain 2024-01-04 00:44:15 +05:30
parent e911ee72d4
commit b7728d43c5
Signed by: ishan
GPG Key ID: 0506DB2A1CC75C27
1 changed files with 37 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

69
src/lib.rs
View File

@ -43,6 +43,7 @@ impl MalestormIo {
fn write(&self, buf: &[u8]) -> Result<(), IoError> {
let mut writer = self.stdout.lock();
writer.write_all(buf)?;
writer.write_all(&[b'\n'])?;
writer.flush()
}
@ -59,40 +60,44 @@ impl<'a> Malestorm<'a> {
let mut buf = Vec::new();
let read = match program.io.read_line(&mut buf) {
Ok(v) => v,
Err(e) => panic!("{:?}", e),
};
loop {
let read = match program.io.read_line(&mut buf) {
Ok(v) => v,
Err(e) => panic!("{:?}", e),
};
if read == 0 {
return;
if read == 0 {
continue;
}
let message: Message = match serde_json::from_slice(&buf) {
Ok(v) => v,
Err(e) => {
eprintln!(
"error in parsing input: {:?} msg = {:?}",
e,
String::from_utf8_lossy(&buf)
);
continue;
}
};
buf.clear();
let mtype = message.body.message_type.clone();
let pc = program.clone();
let handler = match program.handlers.get(&mtype.as_str()) {
Some(v) => v,
None => {
eprintln!("no handler found for {}", message.body.message_type);
continue;
}
};
handler(message, pc);
}
let message: Message = match serde_json::from_slice(&buf) {
Ok(v) => v,
Err(e) => {
eprintln!(
"error in parsing input: {:?} msg = {}",
e,
String::from_utf8_lossy(&buf)
);
return;
}
};
let mtype = message.body.message_type.clone();
let pc = program.clone();
let handler = match program.handlers.get(&mtype.as_str()) {
Some(v) => v,
None => {
eprintln!("no handler found for {}", message.body.message_type);
return;
}
};
handler(message, pc);
}
pub fn set_node_id(&self, node_id: String) {