在Rust中使用axum處理WebSocket連接，你需要使用tokio-tungstenite庫。首先，確保在你的Cargo.toml文件中添加了以下依賴：
[dependencies]
axum = "0.6"
tokio = { version = "1", features = ["full"] }
tokio-tungstenite = "0.15"
接下來，你可以創建一個簡單的WebSocket服務器，如下所示：
use axum::{service::{make_service_fn, service_fn}, Server, Body};
use tokio_tungstenite::{accept_async, WebSocketStream};
use std::convert::Infallible;
async fn handle_connection(ws_stream: WebSocketStream<Body>) {
let (mut user_ws_sender, mut user_ws_receiver) = ws_stream.split();
while let Some(msg) = user_ws_receiver.next().await {
let msg = msg.expect("error caught in in_loop");
if msg.is_text() || msg.is_binary() {
user_ws_sender.send(msg).await.expect("error sending message back");
} else {
println!("unsupported message type");
}
}
}
#[tokio::main]
async fn main() {
let make_svc = make_service_fn(|_conn| {
async { Ok::<_, Infallible>(service_fn(handle_connection)) }
});
let addr = ([127, 0, 0, 1], 3000).into();
let server = Server::bind(&addr).serve(make_svc);
if let Err(e) = server.await {
eprintln!("server error: {}", e);
}
}
這個示例創建了一個簡單的WebSocket服務器，監聽本地地址127.0.0.1:3000。當客戶端連接到服務器時，handle_connection函數將被調用。這個函數接收一個WebSocketStream，然后在一個循環中處理傳入的消息。如果消息是文本或二進制類型，它將被發送回客戶端。
要測試這個服務器，你可以使用任何支持WebSocket的客戶端庫，例如瀏覽器中的JavaScript WebSocket API。
