volo 0.12.3

Volo is a high-performance and strong-extensibility Rust RPC framework that helps developers build microservices.
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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122

use std::io;

use super::{
    Address, DefaultIncoming, MakeIncoming,
    conn::{Conn, OwnedReadHalf, OwnedWriteHalf},
    dial::{DefaultMakeTransport, MakeTransport},
    incoming::Incoming,
};

pub struct ShmipcAddressWithFallback<MI> {
    pub shmipc_addr: Address,
    pub default_mi: MI,
}

impl<MI, I> MakeIncoming for ShmipcAddressWithFallback<MI>
where
    MI: MakeIncoming<Incoming = I> + Send,
    I: Incoming + Send,
{
    type Incoming = ShmipcIncoming<I>;

    async fn make_incoming(self) -> io::Result<Self::Incoming> {
        Ok(ShmipcIncoming {
            shmipc_listener: self.shmipc_addr.make_incoming().await?,
            default_incoming: self.default_mi.make_incoming().await?,
        })
    }
}

#[derive(Debug)]
pub struct ShmipcIncoming<I> {
    shmipc_listener: DefaultIncoming,
    default_incoming: I,
}

impl<I> Incoming for ShmipcIncoming<I>
where
    I: Incoming,
{
    async fn accept(&mut self) -> io::Result<Option<Conn>> {
        self.try_next().await
    }
}

impl<I> ShmipcIncoming<I>
where
    I: Incoming,
{
    async fn try_next(&mut self) -> io::Result<Option<Conn>> {
        tokio::select! {
            biased;
            conn = self.shmipc_listener.accept() => {
                tracing::trace!("recv a conn from shmipc");
                conn
            }
            conn = self.default_incoming.accept() => {
                tracing::trace!("recv a conn from default");
                conn
            }
        }
    }
}

#[derive(Clone, Debug)]
pub struct ShmipcMakeTransportWithFallback {
    pub shmipc_mkt: DefaultMakeTransport,
    pub default_mkt: DefaultMakeTransport,
    pub fallback_addr: Address,
}

impl ShmipcMakeTransportWithFallback {
    pub fn new(
        shmipc: DefaultMakeTransport,
        default_mkt: DefaultMakeTransport,
        fallback_addr: Address,
    ) -> Self {
        Self {
            shmipc_mkt: shmipc,
            default_mkt,
            fallback_addr,
        }
    }
}

impl MakeTransport for ShmipcMakeTransportWithFallback {
    type ReadHalf = OwnedReadHalf;
    type WriteHalf = OwnedWriteHalf;

    async fn make_transport(
        &self,
        mut addr: Address,
    ) -> io::Result<(Self::ReadHalf, Self::WriteHalf)> {
        if addr.is_shmipc() {
            match self.shmipc_mkt.make_transport(addr).await {
                Ok(ret) => return Ok(ret),
                Err(e) => {
                    tracing::info!(
                        "failed to connect to shmipc target: {e}, fallback to default target"
                    );
                    addr = self.fallback_addr.clone();
                }
            }
        }

        self.default_mkt.make_transport(addr).await
    }

    fn set_connect_timeout(&mut self, timeout: Option<std::time::Duration>) {
        self.default_mkt.set_connect_timeout(timeout);
        self.shmipc_mkt.set_connect_timeout(timeout);
    }

    fn set_read_timeout(&mut self, timeout: Option<std::time::Duration>) {
        self.default_mkt.set_read_timeout(timeout);
        self.shmipc_mkt.set_read_timeout(timeout);
    }

    fn set_write_timeout(&mut self, timeout: Option<std::time::Duration>) {
        self.default_mkt.set_write_timeout(timeout);
        self.shmipc_mkt.set_write_timeout(timeout);
    }
}