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

Full workaround

Browse source
This commit is contained in:
Bastian Gruber 2023-05-01 10:38:14 +02:00
parent a86d38f913
commit 8c4c4d1d36
No known key found for this signature in database
GPG key ID: BE9F8C772B188CBF
1 changed files with 11 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
problem_04/bin/server.rs
View file

@ -13,7 +13,7 @@ async fn main() -> io::Result<()> {
tokio::spawn(async move { tokio::spawn(async move {
while let Some((bytes, addr)) = rx.recv().await { while let Some((bytes, addr)) = rx.recv().await {
let len = s.send_to(&bytes, &addr).await.unwrap(); let _ = s.send_to(&bytes, &addr).await.unwrap();
} }
}); });
@ -22,17 +22,22 @@ async fn main() -> io::Result<()> {
let (len, addr) = r.recv_from(&mut buf).await?; let (len, addr) = r.recv_from(&mut buf).await?;
let message = str::from_utf8(&buf[..len]).unwrap().trim_matches('\n'); let message = str::from_utf8(&buf[..len]).unwrap().trim_matches('\n');
if message.contains("=") { if message.contains("version") {
let (key, value) = message.split_once('=').unwrap(); let message = format!("version=gruberb 1.0");
tx.send((message.as_bytes().to_vec(), addr)).await.unwrap();
} else if message.contains("=") {
let (mut key, value) = message.split_once('=').unwrap();
if key.is_empty() {
key = " ";
}
storage storage
.lock() .lock()
.unwrap() .unwrap()
.insert(key.to_string(), value.to_string()); .insert(key.to_string(), value.to_string());
// continue;
} else { } else {
let value = storage.lock().unwrap().get(message).unwrap().clone(); let value = storage.lock().unwrap().get(message).unwrap().clone();
tx.send((value.as_bytes().to_vec(), addr)).await.unwrap(); let message = format!("{message}={value}");
tx.send((message.as_bytes().to_vec(), addr)).await.unwrap();
} }
buf.fill(0); buf.fill(0);