Chat

use actix_web::{middleware::Logger, web, App, HttpRequest, HttpResponse, HttpServer}; use actix_ws::{Message, Session}; use futures::stream::{FuturesUnordered, StreamExt}; use log::info; use std::{ sync::Arc, time::{Duration, Instant}, }; use tokio::sync::Mutex; #[derive(Clone)] struct Chat { inner: Arc>, } struct ChatInner { sessions: Vec, } impl Chat { fn new() -> Self { Chat { inner: Arc::new(Mutex::new(ChatInner { sessions: Vec::new(), })), } } async fn insert(&self, session: Session) { self.inner.lock().await.sessions.push(session); } async fn send(&self, msg: String) { let mut inner = self.inner.lock().await; let mut unordered = FuturesUnordered::new(); for mut session in inner.sessions.drain(..) { let msg = msg.clone(); unordered.push(async move { let res = session.text(msg).await; res.map(|_| session).map_err(|_| info!("Dropping session")) }); } while let Some(res) = unordered.next().await { if let Ok(session) = res { inner.sessions.push(session); } } } } async fn ws( req: HttpRequest, body: web::Payload, chat: web::Data, ) -> Result { let (response, mut session, mut stream) = actix_ws::handle(&req, body)?; chat.insert(session.clone()).await; info!("Inserted session"); let alive = Arc::new(Mutex::new(Instant::now())); let mut session2 = session.clone(); let alive2 = alive.clone(); actix_rt::spawn(async move { let mut interval = actix_rt::time::interval(Duration::from_secs(5)); loop { interval.tick().await; if session2.ping(b"").await.is_err() { break; } if Instant::now().duration_since(alive2.lock().await.clone()) > Duration::from_secs(10) { let _ = session2.close(None).await; break; } } }); actix_rt::spawn(async move { while let Some(Ok(msg)) = stream.next().await { match msg { Message::Ping(bytes) => { if session.pong(&bytes).await.is_err() { return; } } Message::Text(s) => { info!("Relaying text, {}", s); chat.send(s).await; } Message::Close(reason) => { let _ = session.close(reason).await; info!("Got close, bailing"); return; } Message::Continuation(_) => { let _ = session.close(None).await; info!("Got continuation, bailing"); return; } Message::Pong(_) => { *alive.lock().await = Instant::now(); } _ => (), }; } let _ = session.close(None).await; }); info!("Spawned"); Ok(response) } async fn index() -> HttpResponse { let s = r#" Chat

"#; HttpResponse::Ok().content_type("text/html").body(s) } #[actix_rt::main] async fn main() -> Result<(), anyhow::Error> { std::env::set_var("RUST_LOG", "info"); pretty_env_logger::init(); let chat = Chat::new(); HttpServer::new(move || { App::new() .wrap(Logger::default()) .data(chat.clone()) .route("/", web::get().to(index)) .route("/ws", web::get().to(ws)) }) .bind("127.0.0.1:8080")? .run() .await?; Ok(()) }