2018-04-13 09:18:42 +08:00
|
|
|
//! Simple echo websocket server.
|
|
|
|
//! Open `http://localhost:8080/ws/index.html` in browser
|
2018-05-20 18:28:45 -07:00
|
|
|
//! or [python console client](https://github.com/actix/examples/blob/master/websocket/websocket-client.py)
|
2018-04-13 09:18:42 +08:00
|
|
|
//! could be used for testing.
|
|
|
|
|
2019-03-09 18:03:09 -08:00
|
|
|
use std::time::{Duration, Instant};
|
2018-09-27 22:37:19 +03:00
|
|
|
|
2018-04-13 09:18:42 +08:00
|
|
|
use actix::prelude::*;
|
2019-03-18 05:31:32 -07:00
|
|
|
use actix_files as fs;
|
2018-05-20 21:03:29 -07:00
|
|
|
use actix_web::{
|
2019-03-18 05:31:32 -07:00
|
|
|
error, middleware, web, App, Error, HttpRequest, HttpResponse, HttpServer,
|
2018-05-20 21:03:29 -07:00
|
|
|
};
|
2019-03-18 05:31:32 -07:00
|
|
|
use actix_web_actors::ws;
|
|
|
|
use bytes::Bytes;
|
|
|
|
use futures::Stream;
|
2018-04-13 09:18:42 +08:00
|
|
|
|
2018-09-27 22:37:19 +03:00
|
|
|
/// How often heartbeat pings are sent
|
|
|
|
const HEARTBEAT_INTERVAL: Duration = Duration::from_secs(5);
|
|
|
|
/// How long before lack of client response causes a timeout
|
|
|
|
const CLIENT_TIMEOUT: Duration = Duration::from_secs(10);
|
|
|
|
|
2018-04-13 09:18:42 +08:00
|
|
|
/// do websocket handshake and start `MyWebSocket` actor
|
2019-03-18 05:31:32 -07:00
|
|
|
fn ws_index<S>(r: HttpRequest, stream: web::Payload<S>) -> Result<HttpResponse, Error>
|
|
|
|
where
|
|
|
|
S: Stream<Item = Bytes, Error = error::PayloadError> + 'static,
|
|
|
|
{
|
|
|
|
println!("{:?}", r);
|
|
|
|
let res = ws::start(MyWebSocket::new(), &r, stream);
|
|
|
|
println!("{:?}", res.as_ref().unwrap());
|
|
|
|
res
|
2018-04-13 09:18:42 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
/// websocket connection is long running connection, it easier
|
|
|
|
/// to handle with an actor
|
2018-09-27 22:37:19 +03:00
|
|
|
struct MyWebSocket {
|
|
|
|
/// Client must send ping at least once per 10 seconds (CLIENT_TIMEOUT),
|
|
|
|
/// otherwise we drop connection.
|
|
|
|
hb: Instant,
|
|
|
|
}
|
2018-04-13 09:18:42 +08:00
|
|
|
|
|
|
|
impl Actor for MyWebSocket {
|
|
|
|
type Context = ws::WebsocketContext<Self>;
|
2018-09-27 22:37:19 +03:00
|
|
|
|
|
|
|
/// Method is called on actor start. We start the heartbeat process here.
|
|
|
|
fn started(&mut self, ctx: &mut Self::Context) {
|
|
|
|
self.hb(ctx);
|
|
|
|
}
|
2018-04-13 09:18:42 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
/// Handler for `ws::Message`
|
|
|
|
impl StreamHandler<ws::Message, ws::ProtocolError> for MyWebSocket {
|
|
|
|
fn handle(&mut self, msg: ws::Message, ctx: &mut Self::Context) {
|
|
|
|
// process websocket messages
|
|
|
|
println!("WS: {:?}", msg);
|
|
|
|
match msg {
|
2018-09-27 22:37:19 +03:00
|
|
|
ws::Message::Ping(msg) => {
|
|
|
|
self.hb = Instant::now();
|
|
|
|
ctx.pong(&msg);
|
|
|
|
}
|
2018-09-30 14:09:00 -06:00
|
|
|
ws::Message::Pong(_) => {
|
|
|
|
self.hb = Instant::now();
|
|
|
|
}
|
2018-04-13 09:18:42 +08:00
|
|
|
ws::Message::Text(text) => ctx.text(text),
|
|
|
|
ws::Message::Binary(bin) => ctx.binary(bin),
|
|
|
|
ws::Message::Close(_) => {
|
|
|
|
ctx.stop();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-09-27 22:37:19 +03:00
|
|
|
impl MyWebSocket {
|
|
|
|
fn new() -> Self {
|
|
|
|
Self { hb: Instant::now() }
|
|
|
|
}
|
|
|
|
|
|
|
|
/// helper method that sends ping to client every second.
|
|
|
|
///
|
|
|
|
/// also this method checks heartbeats from client
|
|
|
|
fn hb(&self, ctx: &mut <Self as Actor>::Context) {
|
|
|
|
ctx.run_interval(HEARTBEAT_INTERVAL, |act, ctx| {
|
|
|
|
// check client heartbeats
|
|
|
|
if Instant::now().duration_since(act.hb) > CLIENT_TIMEOUT {
|
|
|
|
// heartbeat timed out
|
|
|
|
println!("Websocket Client heartbeat failed, disconnecting!");
|
|
|
|
|
|
|
|
// stop actor
|
|
|
|
ctx.stop();
|
|
|
|
|
|
|
|
// don't try to send a ping
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
ctx.ping("");
|
|
|
|
});
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-03-18 05:31:32 -07:00
|
|
|
fn main() -> std::io::Result<()> {
|
|
|
|
std::env::set_var("RUST_LOG", "actix_server=info,actix_web=info");
|
2018-04-13 09:18:42 +08:00
|
|
|
env_logger::init();
|
|
|
|
|
2019-03-18 05:31:32 -07:00
|
|
|
HttpServer::new(|| {
|
2019-03-09 18:03:09 -08:00
|
|
|
App::new()
|
2018-04-13 09:18:42 +08:00
|
|
|
// enable logger
|
|
|
|
.middleware(middleware::Logger::default())
|
|
|
|
// websocket route
|
2019-03-18 05:31:32 -07:00
|
|
|
.service(web::resource("/ws/").route(web::get().to(ws_index)))
|
2018-04-13 09:18:42 +08:00
|
|
|
// static files
|
2019-03-18 05:31:32 -07:00
|
|
|
.service(fs::Files::new("/", "static/").index_file("index.html"))
|
2019-03-09 18:03:09 -08:00
|
|
|
})
|
|
|
|
// start http server on 127.0.0.1:8080
|
2019-03-18 05:31:32 -07:00
|
|
|
.bind("127.0.0.1:8080")?
|
|
|
|
.run()
|
2018-04-13 09:18:42 +08:00
|
|
|
}
|