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
use async_trait::async_trait;
use crate::either::{EitherName, EitherOutput};
use crate::transport::TransportError;
use crate::upgrade::{UpgradeInfo, Upgrader};
#[derive(Debug, Copy, Clone)]
pub struct Selector<A, B>(A, B);
impl<A, B> Selector<A, B> {
pub fn new(a: A, b: B) -> Self {
Selector(a, b)
}
}
impl<A, B> UpgradeInfo for Selector<A, B>
where
A: UpgradeInfo,
B: UpgradeInfo,
{
type Info = EitherName<A::Info, B::Info>;
fn protocol_info(&self) -> Vec<Self::Info> {
let mut v = Vec::default();
v.extend(self.0.protocol_info().into_iter().map(EitherName::A));
v.extend(self.1.protocol_info().into_iter().map(EitherName::B));
v
}
}
#[async_trait]
impl<A, B, C> Upgrader<C> for Selector<A, B>
where
A: Upgrader<C> + Send,
B: Upgrader<C> + Send,
C: Send + 'static,
{
type Output = EitherOutput<A::Output, B::Output>;
async fn upgrade_inbound(
self,
socket: C,
info: <Self as UpgradeInfo>::Info,
) -> Result<EitherOutput<A::Output, B::Output>, TransportError> {
match info {
EitherName::A(info) => Ok(EitherOutput::A(self.0.upgrade_inbound(socket, info).await?)),
EitherName::B(info) => Ok(EitherOutput::B(self.1.upgrade_inbound(socket, info).await?)),
}
}
async fn upgrade_outbound(
self,
socket: C,
info: <Self as UpgradeInfo>::Info,
) -> Result<EitherOutput<A::Output, B::Output>, TransportError> {
match info {
EitherName::A(info) => Ok(EitherOutput::A(self.0.upgrade_outbound(socket, info).await?)),
EitherName::B(info) => Ok(EitherOutput::B(self.1.upgrade_outbound(socket, info).await?)),
}
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::upgrade::dummy::DummyUpgrader;
#[test]
fn verify_basic() {
let m = Selector::new(DummyUpgrader::new(), DummyUpgrader::new());
libp2prs_runtime::task::block_on(async move {
let output = m.upgrade_outbound(100u32, EitherName::A(b"")).await.unwrap();
let mut _o = match output {
EitherOutput::A(a) => a,
EitherOutput::B(a) => a,
};
});
}
}