use std::sync::Arc;
use axum::extract::ws::WebSocket;
use futures::StreamExt;
use super::WsState;
pub async fn handle_socket(_ws_state: Arc<WsState>, socket: WebSocket) {
let (_, mut receiver) = socket.split();
let mut send_task = tokio::spawn(async move {
loop {
tokio::time::sleep(std::time::Duration::from_secs(30)).await;
}
});
let mut recv_task = tokio::spawn(async move {
let mut cnt = 0;
while let Some(Ok(msg)) = receiver.next().await {
cnt += 1;
tracing::debug!("recv message: {:?}", msg);
}
cnt
});
tokio::select! {
rv_a = (&mut send_task) => {
match rv_a {
Ok(a) => tracing::debug!("{:?} messages sent", a),
Err(a) => tracing::error!("WS Error sending messages {:?}", a)
}
recv_task.abort();
},
rv_b = (&mut recv_task) => {
match rv_b {
Ok(b) => tracing::debug!("Received {:?} messages", b),
Err(b) => tracing::error!("Error receiving messages {:?}", b)
}
send_task.abort();
}
}
tracing::info!("WS over");
}