use futures_util::{SinkExt, StreamExt, TryStreamExt};
use http::header;
use rquest::{Client, Message};
use std::time::Duration;
#[tokio::main]
async fn main() -> Result<(), rquest::Error> {
tracing_subscriber::fmt()
.with_max_level(tracing::Level::TRACE)
.init();
let client = Client::builder()
.connect_timeout(Duration::from_secs(10))
.cert_verification(false)
.build()?;
let websocket = client
.websocket("wss://127.0.0.1:3000/ws")
.header(header::USER_AGENT, env!("CARGO_PKG_NAME"))
.use_http2()
.send()
.await?;
assert_eq!(websocket.version(), http::Version::HTTP_2);
let (mut tx, mut rx) = websocket.into_websocket().await?.split();
tokio::spawn(async move {
for i in 1..11 {
if let Err(err) = tx.send(Message::text(format!("Hello, World! #{i}"))).await {
eprintln!("failed to send message: {err}");
}
}
});
while let Some(message) = rx.try_next().await? {
if let Message::Text(text) = message {
println!("received: {text}");
}
}
Ok(())
}