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

Simplyfy

Browse source
This commit is contained in:
Bastian Gruber 2023-05-01 11:52:22 +02:00
parent a25a66e008
commit bcc51d3f0e
No known key found for this signature in database
GPG key ID: BE9F8C772B188CBF
1 changed files with 3 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
problem_04/bin/server.rs
View file

@ -1,5 +1,4 @@
use std::collections::HashMap;
use std::sync::Mutex;
use std::{io, net::SocketAddr, str, sync::Arc};
use tokio::{net::UdpSocket, sync::mpsc};
use tracing::info;
@ -14,7 +13,7 @@ async fn main() -> io::Result<()> {
let r = Arc::new(sock);
let s = r.clone();
let (tx, mut rx) = mpsc::channel::<(Vec<u8>, SocketAddr)>(1_000);
let storage = Arc::new(Mutex::new(HashMap::<String, String>::new()));
let mut storage = HashMap::<String, String>::new();
tokio::spawn(async move {
info!("Send back!");
@ -33,16 +32,11 @@ async fn main() -> io::Result<()> {
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 = " ";
}
let (key, value) = message.split_once('=').unwrap();
storage
.lock()
.unwrap()
.insert(key.to_string(), value.to_string());
} else {
let value = storage.lock().unwrap().get(message).unwrap_or(&String::new()).clone();
let value = storage.get(message).unwrap_or(&String::new()).clone();
let message = format!("{message}={value}");
tx.send((message.as_bytes().to_vec(), addr)).await.unwrap();
}