ogurpchik 0.1.0

A transport-agnostic RPC framework for stream and memory-based communication. Built with high-performance primitives to deliver medium-performance results.
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

use crate::transport::raw::peer::PeerConfig;
use crate::transport::raw::stream::{GenericStreamBuilder, GenericStreamConnector};
use crate::transport::stream::vsock::{VsockAcceptorBuilder, VsockConnector};

pub struct VsockTransport {
    cid: u32,
    base_port: u32,
    config: PeerConfig,
}

impl VsockTransport {
    pub fn new(cid: u32, base_port: u32, config: PeerConfig) -> Self {
        Self {
            cid,
            base_port,
            config,
        }
    }

    pub fn server_builder(&self, core_id: usize) -> GenericStreamBuilder<VsockAcceptorBuilder> {
        let port = self.base_port + core_id as u32;
        GenericStreamBuilder::new(
            VsockAcceptorBuilder::new(self.cid, port),
            self.config.clone(),
        )
    }

    pub fn client_connector(&self, core_id: usize) -> GenericStreamConnector<VsockConnector> {
        let port = self.base_port + core_id as u32;
        GenericStreamConnector::new(VsockConnector::new(self.cid, port), self.config.clone())
    }
}