mirror of
https://github.com/fafhrd91/actix-web
synced 2024-11-24 00:21:08 +01:00
87 lines
2.5 KiB
Rust
87 lines
2.5 KiB
Rust
//! 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(
|
|
format!("Num of requests: {}", 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 = 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();
|
|
}
|