axum rust如何处理WebSocket连接

在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。

声明：本文内容由网友自发贡献，本站不承担相应法律责任。对本内容有异议或投诉，请联系2913721942#qq.com核实处理，我们将尽快回复您，谢谢合作！