actix/examples
1
0
Fork 0
mirror of https://github.com/actix/examples synced 2025-02-25 10:32:49 +01:00
Code Issues Releases Wiki Activity

update state example

Browse Source
This commit is contained in:
Nikolay Kim 2018-05-02 06:20:43 -07:00
parent 95358b8864
commit 0b9d7ab855
1 changed files with 15 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
state/src/main.rs
View File

@ -1,7 +1,15 @@
#![cfg_attr(feature="cargo-clippy", allow(needless_pass_by_value))] #![cfg_attr(feature = "cargo-clippy", allow(needless_pass_by_value))]
//! There are two level of statefulness in actix-web. Application has state //! There are two level of statefulness in actix-web. Application has state
//! that is shared across all handlers within same Application. //! that is shared across all handlers within same Application.
//! And individual handler can have state. //! And individual handler can have state.
//!
//! > **Note**: http server accepts an application factory rather than an application
//! > instance. Http server constructs an application instance for each thread,
//! > thus application state
//! > must be constructed multiple times. If you want to share state between different
//! > threads, a shared object should be used, e.g. `Arc`.
//!
//! Check [user guide](https://actix.rs/book/actix-web/sec-2-application.html) for more info.
extern crate actix; extern crate actix;
extern crate actix_web; extern crate actix_web;
@ -10,8 +18,7 @@ extern crate env_logger;
use std::cell::Cell; use std::cell::Cell;
use actix::prelude::*; use actix::prelude::*;
use actix_web::{ use actix_web::{http, middleware, server, ws, App, HttpRequest, HttpResponse};
http, server, ws, middleware, App, HttpRequest, HttpResponse};
/// Application state /// Application state
struct AppState { struct AppState {
@ -26,50 +33,19 @@ fn index(req: HttpRequest<AppState>) -> HttpResponse {
HttpResponse::Ok().body(format!("Num of requests: {}", req.state().counter.get())) HttpResponse::Ok().body(format!("Num of requests: {}", req.state().counter.get()))
} }
/// `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 = ws::WebsocketContext<Self, AppState>;
}
impl StreamHandler<ws::Message, ws::ProtocolError> for MyWebSocket {
fn handle(&mut self, msg: ws::Message, ctx: &mut Self::Context) {
self.counter += 1;
println!("WS({}): {:?}", self.counter, msg);
match msg {
ws::Message::Ping(msg) => ctx.pong(&msg),
ws::Message::Text(text) => ctx.text(text),
ws::Message::Binary(bin) => ctx.binary(bin),
ws::Message::Close(_) => {
ctx.stop();
}
_ => (),
}
}
}
fn main() { fn main() {
::std::env::set_var("RUST_LOG", "actix_web=info"); ::std::env::set_var("RUST_LOG", "actix_web=info");
env_logger::init(); env_logger::init();
let sys = actix::System::new("ws-example"); let sys = actix::System::new("ws-example");
server::new( server::new(|| {
|| App::with_state(AppState{counter: Cell::new(0)}) App::with_state(AppState{counter: Cell::new(0)}) // <- create app with state
// enable logger // enable logger
.middleware(middleware::Logger::default()) .middleware(middleware::Logger::default())
// websocket route
.resource(
"/ws/", |r|
r.method(http::Method::GET).f(
|req| ws::start(req, MyWebSocket{counter: 0})))
// register simple handler, handle all methods // register simple handler, handle all methods
.resource("/", |r| r.f(index))) .resource("/", |r| r.f(index))
.bind("127.0.0.1:8080").unwrap() }).bind("127.0.0.1:8080")
.unwrap()
.start(); .start();
println!("Started http server: 127.0.0.1:8080"); println!("Started http server: 127.0.0.1:8080");