ezsockets 0.7.1

WebSockets server & client made easy
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

mod chat;
mod client;

use chat::ChatClient;
use chat::ChatServer;

use axum::extract::Extension;
use axum::response::IntoResponse;
use axum::routing::get;
use axum::Router;
use ezsockets::axum::Upgrade;
use ezsockets::Server;
use ezsockets::ServerExt;
use std::net::SocketAddr;
use tokio::net::TcpListener;

async fn websocket_handler<E>(
    Extension(server): Extension<Server<E>>,
    ezsocket: Upgrade,
) -> impl IntoResponse
where
    E: ServerExt + 'static,
{
    ezsocket.on_upgrade(server)
}

async fn run<E>(create_fn: impl FnOnce(Server<E>) -> E) -> (Server<E>, SocketAddr)
where
    E: ServerExt + 'static,
{
    let (server, _) = Server::create(create_fn);
    let app = Router::new()
        .route("/websocket", get(websocket_handler::<E>))
        .layer(Extension(server.clone()));

    let address = SocketAddr::from(([127, 0, 0, 1], 0));

    tracing::debug!("listening on {}", address);
    let listener = TcpListener::bind(address).await.unwrap();
    let future = axum::serve(
        listener,
        app.into_make_service_with_connect_info::<SocketAddr>(),
    );
    let address = future.local_addr().unwrap();
    tokio::spawn(async move {
        future.await.unwrap();
    });
    (server, address)
}

#[tokio::test]
async fn test_axum_chat() {
    tracing_subscriber::fmt::init();
    let (_, address) = run(ChatServer::new).await;
    let alice = client::connect(ChatClient::new, address).await;
    let bob = client::connect(ChatClient::new, address).await;
    chat::test(alice, bob).await;
}