use std::collections::HashMap; use std::sync::{Arc, RwLock}; use actors::ws::WebsocketActor; use actors::{Actor, ActorHandle}; use warp::Filter; type Arbiter = Arc>>>; #[tokio::main] async fn main() { let arbiter = Arc::new(RwLock::new(HashMap::new())); let arbiter = warp::any().map(move || arbiter.clone()); // GET /chat -> websocket upgrade let chat = warp::path("chat") // The `ws()` filter will prepare Websocket handshake... .and(warp::ws()) .and(arbiter) .map(|ws: warp::ws::Ws, arbiter: Arbiter| { // This will call our function if the handshake succeeds. ws.on_upgrade(move |socket| { let actor = WebsocketActor::new(socket); let handle = actor.start(); arbiter.write().unwrap().insert(0, handle); futures::future::ready(()) }) }); // GET / -> index html let index = warp::path::end().map(|| warp::reply::html(INDEX_HTML)); let routes = index.or(chat); warp::serve(routes).run(([127, 0, 0, 1], 3030)).await; } static INDEX_HTML: &str = r#" Warp Chat

Warp chat

Connecting...

"#;