few bug fixes

This commit is contained in:
2025-03-14 03:33:48 +01:00
parent 663f395c6e
commit 2eaa843016
5 changed files with 46 additions and 30 deletions
+18 -13
View File
@@ -25,25 +25,30 @@ fn main() -> std::io::Result<()> {
let mut reader = BufReader::new(stream.try_clone().unwrap());
match methods::Session::new(&mut reader, database) {
Ok(session) => loop {
let mut request_string = String::new();
match reader.read_line(&mut request_string).unwrap() {
0 => break,
_ => {
let (method, params) = parser::parse(&request_string).unwrap();
let response = session.handle_request(&method, &params);
stream
.write_all(
Ok((session, res)) => {
writeln!(stream, "{res}").unwrap();
stream.flush().unwrap();
loop {
let mut request_string = String::new();
match reader.read_line(&mut request_string).unwrap() {
0 => break,
_ => {
let (method, params) = parser::parse(&request_string).unwrap();
let response = session.handle_request(&method, &params);
writeln!(
stream,
"{}",
match response {
Ok(response) => response.to_string(),
Err(e) => format!("!{e}"),
}
.as_bytes(),
)
.unwrap();
}
};
},
}
};
}
}
Err(e) => {
stream.write_all(format!("!{e}").as_bytes()).unwrap();
stream