mrpc 0.1.0

MessagePack-RPC for Rust
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
83

//! Conformance tests for mrpc interoperability with the msgpack-rpc crate.

#![allow(clippy::tests_outside_test_module)]

use std::{error::Error, sync::Arc};

use async_trait::async_trait;
use mrpc::{Connection, Result as MrpcResult, RpcError, RpcSender, Server, Value};
use tempfile::tempdir;
use tokio::{net::UnixStream, sync::Mutex};
use tokio_util::compat::TokioAsyncReadCompatExt;

const PINGS: u32 = 3;

#[derive(Clone, Default)]
struct PingPongService {
    pong_counter: Arc<Mutex<u32>>,
}

#[async_trait]
impl Connection for PingPongService {
    async fn handle_request(
        &self,
        _client: RpcSender,
        method: &str,
        params: Vec<Value>,
    ) -> MrpcResult<Value> {
        match method {
            "ping" => {
                let _ = params.first().and_then(|v| v.as_i64()).ok_or_else(|| {
                    RpcError::Protocol("Expected integer parameter for ping".into())
                })?;
                let mut count = self.pong_counter.lock().await;
                *count += 1;
                Ok(Value::Boolean(true))
            }
            _ => Err(RpcError::Protocol(
                format!("PingPongService: Unknown method: {}", method).into(),
            )),
        }
    }
}

#[tokio::test]
async fn test_mrpc_compatibility_with_msgpack_rpc() -> Result<(), Box<dyn Error>> {
    let temp_dir = tempdir()?;
    let socket_path = temp_dir.path().join("pingpong.sock");

    // Set up the mrpc server
    let pong_counter = Arc::new(Mutex::new(0_u32));
    let pong_counter_clone = pong_counter.clone();

    let server: Server<PingPongService> = Server::from_fn(move || PingPongService {
        pong_counter: pong_counter_clone.clone(),
    })
    .unix(&socket_path)
    .await?;
    let server_handle = server.spawn().await?;

    // Create a msgpack-rpc client
    let socket = UnixStream::connect(&socket_path).await?;
    let client = msgpack_rpc::Client::new(socket.compat());

    for i in 0..PINGS {
        client
            .request("ping", &[msgpack_rpc::Value::Integer(i.into())])
            .await
            .unwrap();
    }

    // Check if our server received 10 pings
    let final_count = *pong_counter.lock().await;
    assert_eq!(
        final_count, PINGS,
        "Expected {} pongs, but got {}",
        PINGS, final_count
    );

    drop(client);
    server_handle.shutdown();
    server_handle.join().await?;
    Ok(())
}