WebSocket 入门 #

[dependencies]
actix-web = "4"
actix-ws = "0.6"
tokio = { version = "1", features = ["full"] }

use actix_ws::{Message, ProtocolError};
use actix_web::{web, App, Error, HttpRequest, HttpResponse, HttpServer};

async fn ws_index(req: HttpRequest, body: web::Payload) -> Result<HttpResponse, Error> {
    let (response, mut session, mut msg_stream) = actix_ws::handle(&req, body)?;
    
    actix_web::rt::spawn(async move {
        while let Some(msg_result) = msg_stream.next().await {
            match msg_result {
                Ok(msg) => {
                    match msg {
                        Message::Text(text) => {
                            session.text(text).await.unwrap();
                        }
                        Message::Binary(bin) => {
                            session.binary(bin).await.unwrap();
                        }
                        Message::Ping(msg) => {
                            session.pong(&msg).await.unwrap();
                        }
                        Message::Close(reason) => {
                            session.close(reason).await.unwrap();
                            return;
                        }
                        _ => {}
                    }
                }
                Err(e) => {
                    eprintln!("Error: {}", e);
                    return;
                }
            }
        }
    });
    
    Ok(response)
}

#[actix_web::main]
async fn main() -> std::io::Result<()> {
    HttpServer::new(|| {
        App::new()
            .route("/ws", web::get().to(ws_index))
    })
    .bind("127.0.0.1:8080")?
    .run()
    .await
}

const ws = new WebSocket('ws://localhost:8080/ws');

ws.onopen = () => {
    console.log('Connected');
    ws.send('Hello, Server!');
};

ws.onmessage = (event) => {
    console.log('Received:', event.data);
};

ws.onclose = () => {
    console.log('Disconnected');
};