Rust websocket 客户端实现[转]

最新推荐文章于 2025-04-18 06:19:09 发布

zeloas

最新推荐文章于 2025-04-18 06:19:09 发布

阅读量1k

点赞数

CC 4.0 BY-SA版权

文章标签： rust websocket 开发语言

本文链接：https://blog.csdn.net/yao_zhuang/article/details/130468361

文章讲述了在Rust中利用Websockets库和Tokio运行时建立WebSocket连接，并通过线程和异步操作定期发送ping来维持长连接的过程。遇到的问题包括闭包在循环中的使用以及如何有效管理WebSocket的读写操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

目前rust websocket文档较少，最近为了实现部分工作需要使用rust去做websocket链接网上找了不少，很多没有太多参考价值，websocket 在rust中要保持长连接，期间需要不停的去ping，不然会中断，但是使用线程在常规情况下闭包又无法在循环数据的时候持续的ping，所以引入了一下第三方包。

cargo.toml

[dependencies]
tokio = { version = "1.19.2", features = ["rt", "rt-multi-thread", "macros"] }
websockets= "0.3.0"
futures = "0.3.21"

main.rs

use std::thread;
use std::time::Duration;
use websockets::{WebSocket, WebSocketWriteHalf};

#[tokio::main]
async fn main() {
	// 建立websocket连接
    let mut ws = WebSocket::connect("wss://xxxxxxx").await.unwrap();
    
    let (mut r, mut w) = ws.split();
	let send_str = "你想提交的内容";
    w.send_text(send_str.to_string()).await.unwrap();
	
	// 定期推送ping，如果想改这里自己建立信息通讯mpsc::channel(0) 发送不同的数据;
    thread::spawn(move || {
        loop {
        	// 这里因为是async自己包裹一下不然跑步起来
            tokio::runtime::Builder::new_multi_thread()
                .enable_all()
                .build()
                .unwrap()
                .block_on(async {
                	// 睡眠定期推送数据
                    thread::sleep(Duration::new(20, 0));
                    w.send_text("ping".parse().unwrap()).await.unwrap();
                });
        }
    });
    
    // 循环打印数据
    loop {
        let s = r.receive().await.unwrap();
        let (ss, b, snake_case) = s.as_text().unwrap();
        println!("Got:{}", ss);
    }
}