datadius/rust-news-trader

rust-news-trader

Example of websocket in rust.

Thanks tesioai.

snapview/tokio-tungstenite#137 (comment)

use tokio_tungstenite::tungstenite::protocol::Message;
use tokio_tungstenite::connect_async;
use futures::{ SinkExt, StreamExt };

#[tokio::main]
async fn main() {
    let url = "ws://localhost:3000/socket";

    // connect to socket
    if let Ok((mut socket, _)) = connect_async(url).await {
        println!("Successfully connected to the WebSocket");
        
        // create message
        let message = Message::from("message");

        // send message
        if let Err(e) = socket.send(message).await {
            eprintln!("Error sending message: {:?}", e);
        }

        // recieve response
        if let Some(Ok(response)) = socket.next().await {
            println!("{response}");
        }
    } else {
        eprintln!("Failed to connect to the WebSocket");
    }
}

Example provided by tokio-tungstenite was vital in figuring out how to maintain the connection open while doing ping pong properly.

let (mut ws_stream, _) = connect_async(case_url).await?;
    while let Some(msg) = ws_stream.next().await {
        let msg = msg?;
        if msg.is_text() || msg.is_binary() {
            ws_stream.send(msg).await?;
        }
    }

the next() method handles the ping pong automatically.

Having it in the while loop makes sure to keep the ping pong going.

There is a risk of the connection timing out if the message handler doesn't finish in time and sends a ping back.