//! There are two level of statfulness in actix-web. Application has state //! that is shared across all handlers within same Application. //! And individual handler can have state. extern crate actix; extern crate actix_web; extern crate env_logger; use actix::*; use actix_web::*; use std::cell::Cell; struct AppState { counter: Cell, } /// somple handle fn index(req: &mut HttpRequest, _: Payload, state: &AppState) -> HttpResponse { println!("{:?}", req); state.counter.set(state.counter.get() + 1); httpcodes::HTTPOk.with_body( Body::Binary(format!("Num of requests: {}", state.counter.get()).into())) } /// `MyWebSocket` counts how many messages it receives from peer, /// websocket-client.py could be used for tests struct MyWebSocket { counter: usize, } impl Actor for MyWebSocket { type Context = HttpContext; } impl Route for MyWebSocket { /// Shared application state type State = AppState; fn request(req: &mut HttpRequest, payload: Payload, ctx: &mut HttpContext) -> RouteResult { let resp = ws::handshake(req)?; ctx.start(resp); ctx.add_stream(ws::WsStream::new(payload)); Reply::async(MyWebSocket{counter: 0}) } } impl StreamHandler for MyWebSocket {} impl Handler for MyWebSocket { fn handle(&mut self, msg: ws::Message, ctx: &mut HttpContext) -> Response { self.counter += 1; println!("WS({}): {:?}", self.counter, msg); match msg { ws::Message::Ping(msg) => ws::WsWriter::pong(ctx, msg), ws::Message::Text(text) => ws::WsWriter::text(ctx, &text), ws::Message::Binary(bin) => ws::WsWriter::binary(ctx, bin), ws::Message::Closed | ws::Message::Error => { ctx.stop(); } _ => (), } Self::empty() } } fn main() { ::std::env::set_var("RUST_LOG", "actix_web=info"); let _ = env_logger::init(); let sys = actix::System::new("ws-example"); HttpServer::new( Application::builder("/", AppState{counter: Cell::new(0)}) // enable logger .middleware(Logger::new(None)) // websocket route .resource("/ws/", |r| r.get::()) // register simple handler, handle all methods .handler("/", index)) .serve::<_, ()>("127.0.0.1:8080").unwrap(); println!("Started http server: 127.0.0.1:8080"); let _ = sys.run(); }