net-pool 0.5.1

A set of types for network connection pool.
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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149

use crate::{Address, BackendState};
use std::collections::HashMap;
use std::collections::hash_map::Entry;

#[derive(Debug, Clone, Default)]
pub(super) struct BackendStates {
    bbs: Vec<BackendState>,
    codes: HashMap<u64, usize>,
    ids: HashMap<u32, usize>,
}

impl BackendStates {
    pub fn new() -> Self {
        Self {
            bbs: vec![],
            codes: HashMap::new(),
            ids: HashMap::new(),
        }
    }

    /// 添加一个地址
    pub fn add(&mut self, id: Option<u32>, addr: Address) {
        let bs = BackendState::new(id, addr);
        let code = bs.hash_code();
        let index = match self.codes.entry(code) {
            Entry::Occupied(e) => {
                let index = *e.get();
                let old = &mut self.bbs[index];
                if old.id() != id {
                    if let Some(old) = old.id() {
                        // 删除掉旧id
                        if let Entry::Occupied(e) = self.ids.entry(old) {
                            if index == *e.get() {
                                e.remove();
                            }
                        }
                    }
                }
                // 替换掉新的bs
                *old = bs;
                index
            }
            Entry::Vacant(e) => {
                self.bbs.push(bs);
                let index = self.bbs.len() - 1;
                e.insert(index);
                index
            }
        };

        if let Some(id) = id {
            // 插入新id
            self.ids.insert(id, index);
        }
    }

    /// 移除一个地址
    pub fn remove(&mut self, addr: &Address) -> bool {
        let code = addr.hash_code();
        if let Some(index) = self.codes.remove(&code) {
            if let Some(id) = self.bbs.swap_remove(index).id() {
                // 移掉id
                if let Entry::Occupied(e) = self.ids.entry(id) {
                    if index == *e.get() {
                        e.remove();
                    }
                }
            }
            if self.bbs.len() > index {
                // 原尾部的更新索引
                let bs = &self.bbs[index];
                self.codes.insert(bs.hash_code(), index);
                if let Some(id) = bs.id() {
                    self.ids.insert(id, index);
                }
            }
            true
        } else {
            false
        }
    }

    /// 是否包含
    pub fn contain(&self, addr: &Address) -> bool {
        self.codes.contains_key(&addr.hash_code())
    }

    /// nth 获取第n个
    pub fn nth(&self, mut n: usize) -> Option<&BackendState> {
        if self.bbs.is_empty() {
            None
        } else {
            n = n % self.bbs.len();
            Some(&self.bbs[n])
        }
    }

    /// 通过id获取
    pub fn get_by_id(&self, id: u32) -> Option<&BackendState> {
        if let Some(index) = self.ids.get(&id) {
            self.nth(*index)
        } else {
            None
        }
    }

    /// 通过code获取
    pub fn get_by_code(&self, code: u64) -> Option<&BackendState> {
        if let Some(index) = self.codes.get(&code) {
            self.nth(*index)
        } else {
            None
        }
    }

    /// 获取所有的
    pub fn get_all(&self) -> &[BackendState] {
        self.bbs.as_slice()
    }
}

#[test]
fn test() {
    let mut bss = BackendStates::new();
    bss.add(None, Address::from("localhost:10001"));
    println!("{:?}", bss);
    bss.add(Some(1), Address::from("localhost:10001"));
    println!("{:?}", bss);
    bss.add(Some(2), Address::from("localhost:10001"));
    println!("{:?}", bss);
    bss.add(Some(2), Address::from("localhost:10002"));
    println!("{:?}", bss);
    bss.remove(&Address::from("localhost:10003"));
    println!("{:?}", bss);
    bss.remove(&Address::from("localhost:10001"));
    println!("{:?}", bss);
    bss.remove(&Address::from("localhost:10002"));
    println!("{:?}", bss);
    bss.add(Some(1), Address::from("localhost:10001"));
    println!("{:?}", bss);
    bss.add(Some(2), Address::from("localhost:10002"));
    println!("{:?}", bss);
    bss.remove(&Address::from("localhost:10002"));
    println!("{:?}", bss);
    assert!(bss.contain(&Address::from("localhost:10001")));
    println!("{:?}", bss.get_by_id(2));
    println!("{:?}", bss.get_by_id(1));
    println!("{:?}", bss.get_by_code(7139969718446815058));
}