KeshavAnandCode/de-socket-espace-room-server
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

added error handling, random input etc

Browse Source
This commit is contained in:
Keshav Anand
2026-05-14 19:07:53 -05:00
parent 59477c3ea7
commit 7232c56e4e
3 changed files with 29 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
src/main.rs
View File

@@ -33,7 +33,6 @@ async fn handle_connection(mut socket: TcpStream, db: Db) -> std::io::Result<()>
let mut buf = [0; 1024];
let n = socket.read(&mut buf).await?;
// Connection closed
if n == 0 {
return Ok(());
}
@@ -46,25 +45,29 @@ async fn handle_connection(mut socket: TcpStream, db: Db) -> std::io::Result<()>
}
match parts[0].to_uppercase().as_str() {
// Teams submit their solution: PUT [team_id] [value]
"PUT" => {
if parts.len() == 3 {
let team_id = parts[1].to_string();
let val = parts[2].to_string();
let val_str = parts[2];
let mut data = db.write().await; // Async lock
data.insert(team_id.clone(), val.clone());
socket.write_all(b"ACK\n").await?;
println!("[LOG] Team {} submitted solution: {}", team_id, val);
// Validate if the value is an integer
if val_str.parse::<i32>().is_ok() {
let mut data = db.write().await;
data.insert(team_id.clone(), val_str.to_string());
socket.write_all(b"ACK\n").await?;
println!("[LOG] Team {} submitted solution: {}", team_id, val_str);
} else {
// Send error but don't stop the server
socket.write_all(b"ERR_NOT_AN_INT\n").await?;
println!("[WARN] Team {} tried to send non-int: {}", team_id, val_str);
}
}
}
// Teams request previous solution: GET [prev_team_id]
"GET" => {
if parts.len() == 2 {
let target_id = parts[1];
let data = db.read().await; // Async lock
let data = db.read().await;
let response = data
.get(target_id)
@@ -75,17 +78,20 @@ async fn handle_connection(mut socket: TcpStream, db: Db) -> std::io::Result<()>
}
}
// Admin override: ADMIN SET [team_id] [value]
"ADMIN" => {
if parts.len() == 4 && parts[1].to_uppercase() == "SET" {
let team_id = parts[2].to_string();
let val = parts[3].to_string();
let val_str = parts[3];
let mut data = db.write().await;
data.insert(team_id.clone(), val.clone());
socket.write_all(b"ADMIN_OK\n").await?;
println!("[ADMIN] Manual set: Team {} -> {}", team_id, val);
if val_str.parse::<i32>().is_ok() {
let mut data = db.write().await;
data.insert(team_id.clone(), val_str.to_string());
socket.write_all(b"ADMIN_OK\n").await?;
println!("[ADMIN] Manual set: Team {} -> {}", team_id, val_str);
} else {
socket.write_all(b"ADMIN_ERR_NOT_AN_INT\n").await?;
println!("[ADMIN] Manual set failed: {} is not an int", val_str);
}
}
}