aeronet_websocket 0.18.0

WebSocket IO layer implementation for `aeronet`
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80

use {
    super::{ClientConfig, ClientError, ConnectTarget},
    crate::{client::ToConnected, session::SessionError},
    aeronet_io::connection::DisconnectReason,
    futures::{channel::oneshot, never::Never},
    tracing::debug,
};

pub async fn start(
    config: ClientConfig,
    target: ConnectTarget,
    tx_connected: oneshot::Sender<ToConnected>,
) -> Result<Never, DisconnectReason> {
    let (connected, backend) = {
        #[cfg(target_family = "wasm")]
        {
            // suppress `unused_variables`
            _ = config;

            debug!("Spawning backend task to connect to {target:?}");

            let socket = web_sys::WebSocket::new(&target)
                .map_err(crate::JsError::from)
                .map_err(ClientError::CreateSocket)?;
            debug!("Created socket");

            let (frontend, backend) = crate::session::backend::wasm::split(socket);
            (ToConnected { frontend }, backend)
        }

        #[cfg(not(target_family = "wasm"))]
        {
            use tokio_tungstenite::MaybeTlsStream;

            let target = target.map_err(ClientError::CreateTarget)?;

            debug!("Spawning backend task to connect to {:?}", target.uri());

            let (stream, _) = {
                let socket_config = Some(config.socket);
                let disable_nagle = !config.nagle;

                tokio_tungstenite::connect_async_tls_with_config(
                    target,
                    socket_config,
                    disable_nagle,
                    Some(config.connector),
                )
            }
            .await
            .map_err(ClientError::Connect)?;

            let socket = match stream.get_ref() {
                MaybeTlsStream::Plain(socket) => socket,
                MaybeTlsStream::Rustls(stream) => stream.get_ref().0,
                _ => panic!("should not be using this kind of stream - {stream:?}"),
            };
            let local_addr = socket.local_addr().map_err(SessionError::GetLocalAddr)?;
            let peer_addr = socket.peer_addr().map_err(SessionError::GetPeerAddr)?;
            debug!("Created stream");

            let (frontend, backend) = crate::session::backend::native::split(stream);
            (
                ToConnected {
                    local_addr,
                    peer_addr,
                    frontend,
                },
                backend,
            )
        }
    };

    tx_connected
        .send(connected)
        .map_err(|_| SessionError::FrontendClosed)?;

    debug!("Starting session loop");
    backend.start().await
}