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

Add tracing

Browse source
This commit is contained in:
Bastian Gruber 2023-05-01 09:29:40 +00:00
parent 8c4c4d1d36
commit ddf8e1ceb1
1 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
problem_04/bin/server.rs
View file

@ -2,16 +2,22 @@ use std::collections::HashMap;
use std::sync::Mutex; use std::sync::Mutex;
use std::{io, net::SocketAddr, str, sync::Arc}; use std::{io, net::SocketAddr, str, sync::Arc};
use tokio::{net::UdpSocket, sync::mpsc}; use tokio::{net::UdpSocket, sync::mpsc};
use tracing::info;
#[tokio::main] #[tokio::main]
async fn main() -> io::Result<()> { async fn main() -> io::Result<()> {
let sock = UdpSocket::bind("0.0.0.0:8080".parse::<SocketAddr>().unwrap()).await?; tracing_subscriber::fmt::try_init().unwrap();
let sock = UdpSocket::bind("0.0.0.0:1222".parse::<SocketAddr>().unwrap()).await?;
info!("listening to new connections");
let r = Arc::new(sock); let r = Arc::new(sock);
let s = r.clone(); let s = r.clone();
let (tx, mut rx) = mpsc::channel::<(Vec<u8>, SocketAddr)>(1_000); let (tx, mut rx) = mpsc::channel::<(Vec<u8>, SocketAddr)>(1_000);
let storage = Arc::new(Mutex::new(HashMap::<String, String>::new())); let storage = Arc::new(Mutex::new(HashMap::<String, String>::new()));
tokio::spawn(async move { tokio::spawn(async move {
info!("Send back!");
while let Some((bytes, addr)) = rx.recv().await { while let Some((bytes, addr)) = rx.recv().await {
let _ = s.send_to(&bytes, &addr).await.unwrap(); let _ = s.send_to(&bytes, &addr).await.unwrap();
} }
@ -19,6 +25,7 @@ async fn main() -> io::Result<()> {
let mut buf = [0; 1024]; let mut buf = [0; 1024];
loop { loop {
info!("Recv loop");
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');