waaa 0.2.1

WebAssembly, Abstracted Away
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

use super::WsMessage;
use futures_util::{SinkExt, StreamExt};
use gloo_net::websocket::futures::WebSocket as GlooWebSocket;
use gloo_net::websocket::Message as GlooMessage;

pub struct WebSocket {
    implem: GlooWebSocket,
}

impl WebSocket {
    pub async fn connect(url: &str) -> anyhow::Result<WebSocket> {
        Ok(WebSocket {
            implem: GlooWebSocket::open(url)?,
        })
    }

    pub async fn send(&mut self, msg: WsMessage) -> anyhow::Result<()> {
        Ok(self.implem.send(match msg {
            WsMessage::Text(text) => GlooMessage::Text(text),
            WsMessage::Binary(data) => GlooMessage::Bytes(data),
        }).await?)
    }

    pub async fn recv(&mut self) -> anyhow::Result<Option<WsMessage>> {
        let Some(msg) = self.implem.next().await else {
            return Ok(None);
        };
        Ok(Some(match msg? {
            GlooMessage::Text(text) => WsMessage::Text(text),
            GlooMessage::Bytes(data) => WsMessage::Binary(data),
        }))
    }
}