Websockets chapter done-ish.

examples/websockets/src/main.rs

@@ -1,29 +1,44 @@
 // <websockets>
-// use actix::*;
-// use actix_web::*;
+use actix::{Actor, StreamHandler};
+use actix_web::{web, App, Error, HttpRequest, HttpResponse, HttpServer};
+use actix_web_actors::ws;
 
-// /// Define http actor
-// struct Ws;
+/// Define http actor
+struct MyWs;
 
-// impl Actor for Ws {
-//     type Context = ws::WebsocketContext<Self>;
-// }
+impl Actor for MyWs {
+    type Context = ws::WebsocketContext<Self>;
+}
 
-// /// Handler for ws::Message message
-// impl StreamHandler<ws::Message, ws::ProtocolError> for Ws {
-//     fn handle(&mut self, msg: ws::Message, ctx: &mut Self::Context) {
-//         match msg {
-//             ws::Message::Ping(msg) => ctx.pong(&msg),
-//             ws::Message::Text(text) => ctx.text(text),
-//             ws::Message::Binary(bin) => ctx.binary(bin),
-//             _ => (),
-//         }
-//     }
-// }
+/// Handler for ws::Message message
+impl StreamHandler<ws::Message, ws::ProtocolError> for MyWs {
+    fn handle(&mut self, msg: ws::Message, ctx: &mut Self::Context) {
+        match msg {
+            ws::Message::Ping(msg) => ctx.pong(&msg),
+            ws::Message::Text(text) => ctx.text(text),
+            ws::Message::Binary(bin) => ctx.binary(bin),
+            _ => (),
+        }
+    }
+}
 
-// fn main() {
-//     App::new()
-//         .resource("/ws/", |r| r.f(|req| ws::start(req, Ws)))
-//         .finish();
-// }
+fn index(req: HttpRequest, stream: web::Payload) -> Result<HttpResponse, Error> {
+    let resp = ws::start(MyWs {}, &req, stream);
+    println!("{:?}", resp);
+    resp
+}
+
+fn main() {
+    HttpServer::new(|| App::new().route("/ws/", web::get().to(index)))
+        .bind("127.0.0.1:8088")
+        .unwrap()
+        .run()
+        .unwrap();
+}
 // </websockets>
+
+// testing requires specific headers:
+// Upgrade: websocket
+// Connection: Upgrade
+// Sec-WebSocket-Key: SOME_KEY
+// Sec-WebSocket-Version: 13