yerpc 0.6.4

Ergonomic JSON-RPC library for async Rust with Axum support
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
81
82

use futures_util::StreamExt;
use yerpc::{rpc, RpcSession};

#[tokio::test]
async fn basic() -> anyhow::Result<()> {
    struct Api {}

    #[rpc(all_positional, ts_outdir = "typescript/generated")]
    impl Api {
        pub async fn constant(&self) -> String {
            "example".to_string()
        }

        pub async fn upper(&self, text: String) -> String {
            text.to_uppercase()
        }
    }

    let (session, mut out_rx) = RpcSession::create(Api {});

    let req = r#"{"jsonrpc":"2.0","method":"constant","id":3}"#;
    session.handle_incoming(req).await;
    let out = out_rx.next().await.unwrap();
    let out = serde_json::to_string(&out).unwrap();
    assert_eq!(out, r#"{"jsonrpc":"2.0","id":3,"result":"example"}"#);

    let req = r#"{"jsonrpc":"2.0","method":"upper","params":["foo"],"id":7}"#;
    session.handle_incoming(req).await;
    let out = out_rx.next().await.unwrap();
    let out = serde_json::to_string(&out).unwrap();
    assert_eq!(out, r#"{"jsonrpc":"2.0","id":7,"result":"FOO"}"#);

    let client = session.client().clone();
    tokio::spawn(async move {
        let out = out_rx.next().await.unwrap();
        let out = serde_json::to_string(&out).unwrap();
        assert_eq!(
            out,
            r#"{"jsonrpc":"2.0","method":"bar","params":["woo"],"id":1}"#
        );
        session
            .handle_incoming(r#"{"jsonrpc":"2.0","id":1,"result":"boo"}"#)
            .await;
    });
    let res = client.send_request("bar", Some(&["woo"])).await.unwrap();
    assert_eq!(res, "boo");
    Ok(())
}

#[tokio::test]
async fn basic_mixed_id_types() -> anyhow::Result<()> {
    struct Api {}

    #[rpc(all_positional, ts_outdir = "typescript/generated")]
    impl Api {
        pub async fn upper(&self, text: String) -> String {
            text.to_uppercase()
        }
    }

    let (session, mut out_rx) = RpcSession::create(Api {});

    let req = r#"{"jsonrpc":"2.0","method":"upper","params":["foo"],"id":"7"}"#;
    session.handle_incoming(req).await;
    let out = out_rx.next().await.unwrap();
    let out = serde_json::to_string(&out).unwrap();
    assert_eq!(out, r#"{"jsonrpc":"2.0","id":"7","result":"FOO"}"#);

    let req = r#"{"jsonrpc":"2.0","method":"upper","params":["foo"],"id":9}"#;
    session.handle_incoming(req).await;
    let out = out_rx.next().await.unwrap();
    let out = serde_json::to_string(&out).unwrap();
    assert_eq!(out, r#"{"jsonrpc":"2.0","id":9,"result":"FOO"}"#);

    let req = r#"{"jsonrpc":"2.0","method":"upper","params":["foo"],"id":"hi"}"#;
    session.handle_incoming(req).await;
    let out = out_rx.next().await.unwrap();
    let out = serde_json::to_string(&out).unwrap();
    assert_eq!(out, r#"{"jsonrpc":"2.0","id":"hi","result":"FOO"}"#);

    Ok(())
}