gruberb/protohackers
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Added logs, optimize

Browse source
This commit is contained in:
Bastian Gruber 2023-04-21 07:57:11 +02:00
parent 100e020585
commit f99ee4c64d
No known key found for this signature in database
GPG key ID: BE9F8C772B188CBF
1 changed files with 10 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
problem_01/src/main.rs
View file

@ -54,19 +54,19 @@ async fn handle_request(mut socket: TcpStream) {
match validate_request(buf.clone()) {
Ok(m) => {
log::info!("Valid request");
let _ = write.write(&m.as_bytes()).await;
let _ = write.write(&[b'\n']).await;
let _ = buf.clear();
buf.clear();
}
Err(_) => {
log::error!("Not valid request");
let _ = write.write(&MAL_FORMAT.as_bytes()).await;
}
}
let _ = write.write(&[b'\n']).await;
let _ = write.flush().await;
buf.clear();
}
}
}
}
fn validate_request(message: Vec<u8>) -> Result<String, std::io::Error> {
@ -75,6 +75,7 @@ fn validate_request(message: Vec<u8>) -> Result<String, std::io::Error> {
let possible_prime = match m.number.to_string().parse::<u64>() {
Ok(n) => n,
Err(_) => {
log::error!("Not a valid number for a prime candidate: {}", m.number);
return Ok(serde_json::to_string(&Response {
method: IS_PRIME.to_owned(),
prime: false,
@ -84,12 +85,14 @@ fn validate_request(message: Vec<u8>) -> Result<String, std::io::Error> {
};
if m.method == IS_PRIME.to_owned() {
log::info!("Method isPrime and possible prime number");
return Ok(serde_json::to_string(&Response {
method: IS_PRIME.to_owned(),
prime: is_prime(possible_prime),
})
.unwrap());
} else {
log::error!("Method is not isPrime");
return Err(std::io::Error::new(
ErrorKind::InvalidInput,
"Method is not isPrime",
@ -97,6 +100,7 @@ fn validate_request(message: Vec<u8>) -> Result<String, std::io::Error> {
}
}
Err(_) => {
log::error!("Message is not a valid JSON or Request type");
return Err(std::io::Error::new(
ErrorKind::InvalidData,
"Message is not a Request",