use crate::prelude::*;
pub struct WsClient {
stream: WebSocketStream<MaybeTlsStream<TcpStream>>,
}
impl WsClient {
pub fn new(stream: WebSocketStream<MaybeTlsStream<TcpStream>>) -> Self {
Self { stream }
}
pub fn stream(&mut self) -> &mut WebSocketStream<MaybeTlsStream<TcpStream>> {
&mut self.stream
}
pub async fn disconnect(&mut self) -> Result<(), LimitlessError> {
self.stream
.close(None)
.await
.map_err(|e| LimitlessError::Base(format!("Error closing WebSocket: {}", e)))?;
Ok(())
}
}