rpcnet 0.1.0

RPC library based on QUIC+TLS encryption
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101

#![allow(dead_code)]
#![allow(unused_imports)]

use super::types::*;
use async_trait::async_trait;
use rpcnet::{RpcConfig, RpcError, RpcServer};
use std::sync::Arc;
/// Handler trait that users implement for the service.
#[async_trait]
pub trait FileTransferHandler: Send + Sync + 'static {
    async fn upload_chunk(
        &self,
        request: UploadChunkRequest,
    ) -> Result<UploadChunkResponse, FileTransferError>;
    async fn download_chunk(
        &self,
        request: DownloadChunkRequest,
    ) -> Result<DownloadChunkResponse, FileTransferError>;
    async fn get_file_info(
        &self,
        request: FileInfoRequest,
    ) -> Result<FileInfoResponse, FileTransferError>;
}
/// Generated server that manages RPC registration and routing.
pub struct FileTransferServer<H: FileTransferHandler> {
    handler: Arc<H>,
    rpc_server: RpcServer,
}
impl<H: FileTransferHandler> FileTransferServer<H> {
    /// Creates a new server with the given handler and configuration.
    pub fn new(handler: H, config: RpcConfig) -> Self {
        Self {
            handler: Arc::new(handler),
            rpc_server: RpcServer::new(config),
        }
    }
    /// Registers all service methods with the RPC server.
    pub async fn register_all(&mut self) {
        {
            let handler = self.handler.clone();
            self.rpc_server
                .register("FileTransfer.upload_chunk", move |params| {
                    let handler = handler.clone();
                    async move {
                        let request: UploadChunkRequest =
                            bincode::deserialize(&params).map_err(RpcError::SerializationError)?;
                        match handler.upload_chunk(request).await {
                            Ok(response) => {
                                bincode::serialize(&response).map_err(RpcError::SerializationError)
                            }
                            Err(e) => Err(RpcError::StreamError(format!("{:?}", e))),
                        }
                    }
                })
                .await;
        }
        {
            let handler = self.handler.clone();
            self.rpc_server
                .register("FileTransfer.download_chunk", move |params| {
                    let handler = handler.clone();
                    async move {
                        let request: DownloadChunkRequest =
                            bincode::deserialize(&params).map_err(RpcError::SerializationError)?;
                        match handler.download_chunk(request).await {
                            Ok(response) => {
                                bincode::serialize(&response).map_err(RpcError::SerializationError)
                            }
                            Err(e) => Err(RpcError::StreamError(format!("{:?}", e))),
                        }
                    }
                })
                .await;
        }
        {
            let handler = self.handler.clone();
            self.rpc_server
                .register("FileTransfer.get_file_info", move |params| {
                    let handler = handler.clone();
                    async move {
                        let request: FileInfoRequest =
                            bincode::deserialize(&params).map_err(RpcError::SerializationError)?;
                        match handler.get_file_info(request).await {
                            Ok(response) => {
                                bincode::serialize(&response).map_err(RpcError::SerializationError)
                            }
                            Err(e) => Err(RpcError::StreamError(format!("{:?}", e))),
                        }
                    }
                })
                .await;
        }
    }
    /// Starts the server and begins accepting connections.
    pub async fn serve(mut self) -> Result<(), RpcError> {
        self.register_all().await;
        let quic_server = self.rpc_server.bind()?;
        println!("Server listening on: {:?}", self.rpc_server.socket_addr);
        self.rpc_server.start(quic_server).await
    }
}