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

use super::{ChooseOptions, Chooser, ChooserFeedback, ChosenResults, IpAddrWithPort};

/// 直接选择器
///
/// 不做任何筛选，也不接受任何反馈，直接将给出的 IP 地址列表返回
#[derive(Clone, Copy, Debug, Default)]
pub struct DirectChooser;

impl Chooser for DirectChooser {
    #[inline]
    fn choose(&self, ips: &[IpAddrWithPort], _opts: ChooseOptions) -> ChosenResults {
        ips.to_owned().into()
    }

    #[inline]
    fn feedback(&self, _feedback: ChooserFeedback) {}
}

#[cfg(test)]
mod tests {
    use super::*;
    use anyhow::Result;
    use once_cell::sync::Lazy;
    use std::{
        net::{IpAddr, Ipv4Addr},
        num::NonZeroU16,
    };

    static IPS_WITHOUT_PORT: Lazy<Vec<IpAddrWithPort>> = Lazy::new(|| {
        vec![
            IpAddrWithPort::new(IpAddr::V4(Ipv4Addr::new(192, 168, 1, 1)), None),
            IpAddrWithPort::new(IpAddr::V4(Ipv4Addr::new(192, 168, 1, 2)), None),
            IpAddrWithPort::new(IpAddr::V4(Ipv4Addr::new(192, 168, 1, 3)), None),
        ]
    });

    static IPS_WITH_PORT: Lazy<Vec<IpAddrWithPort>> = Lazy::new(|| {
        vec![
            IpAddrWithPort::new(IpAddr::V4(Ipv4Addr::new(192, 168, 1, 1)), NonZeroU16::new(443)),
            IpAddrWithPort::new(IpAddr::V4(Ipv4Addr::new(192, 168, 1, 2)), NonZeroU16::new(443)),
            IpAddrWithPort::new(IpAddr::V4(Ipv4Addr::new(192, 168, 1, 3)), NonZeroU16::new(443)),
        ]
    });

    #[test]
    fn test_direct_chooser() -> Result<()> {
        let chooser = DirectChooser;
        assert_eq!(
            chooser.choose(&IPS_WITHOUT_PORT, Default::default()).into_ip_addrs(),
            IPS_WITHOUT_PORT.to_vec(),
        );
        assert_eq!(
            chooser.choose(&IPS_WITH_PORT, Default::default()).into_ip_addrs(),
            IPS_WITH_PORT.to_vec(),
        );
        Ok(())
    }

    #[cfg(feature = "async")]
    #[tokio::test]
    async fn async_test_direct_chooser() -> Result<()> {
        let chooser = DirectChooser;
        assert_eq!(
            chooser
                .async_choose(&IPS_WITHOUT_PORT, Default::default())
                .await
                .into_ip_addrs(),
            IPS_WITHOUT_PORT.to_vec(),
        );
        assert_eq!(
            chooser
                .async_choose(&IPS_WITH_PORT, Default::default())
                .await
                .into_ip_addrs(),
            IPS_WITH_PORT.to_vec(),
        );
        Ok(())
    }
}