add stateful example

README.md

@@ -26,12 +26,13 @@ To use `actix-web`, add this to your `Cargo.toml`:
 
 ```toml
 [dependencies]
-actix-web = { git = "https://github.com/actix/actix-web.git" }
+actix-web = "0.1"
 ```
 
 ## Example
 
 * [Basic](https://github.com/actix/actix-web/tree/master/examples/basic.rs)
+* [Stateful](https://github.com/actix/actix-web/tree/master/examples/state.rs)
 * [Mulitpart streams](https://github.com/actix/actix-web/tree/master/examples/multipart)
 * [Simple websocket session](https://github.com/actix/actix-web/tree/master/examples/websocket.rs)
 * [Tcp/Websocket chat](https://github.com/actix/actix-web/tree/master/examples/websocket-chat)

examples/state.rs

@@ -0,0 +1,86 @@
+//! 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<usize>,
+}
+
+/// 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<Self>;
+}
+
+impl Route for MyWebSocket {
+    /// Shared application state
+    type State = AppState;
+
+    fn request(req: &mut HttpRequest,
+               payload: Payload, ctx: &mut HttpContext<Self>) -> RouteResult<Self>
+    {
+        let resp = ws::handshake(req)?;
+        ctx.start(resp);
+        ctx.add_stream(ws::WsStream::new(payload));
+        Reply::async(MyWebSocket{counter: 0})
+    }
+}
+impl StreamHandler<ws::Message> for MyWebSocket {}
+impl Handler<ws::Message> for MyWebSocket {
+    fn handle(&mut self, msg: ws::Message, ctx: &mut HttpContext<Self>)
+              -> Response<Self, ws::Message>
+    {
+        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::<MyWebSocket>())
+            // 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();
+}