net-pool 0.2.0

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

use crate::backend::{Address, BackendState};

#[derive(Debug, Clone, Eq, PartialEq)]
pub enum Strategy {
    /// 普通哈希
    Hash,

    /// 一致性哈希
    ConsistentHash,

    /// 轮询
    RoundRobin,
}

/// 策略trait
pub trait LbStrategy: Send + Sync {
    /// 策略类型
    fn strategy(&self) -> Strategy;

    /// 判断一个后端地址是否存在
    fn contain(&self, addr: &Address) -> bool;

    /// 添加一个后端地址
    fn add_backend(&self, addr: Address);

    /// 移除一个后端地址
    fn remove_backend(&self, addr: &Address) -> bool;

    /// 获取一个后端地址
    fn get_backend(&self, key: &str) -> Option<BackendState>;

    /// 获取所有后端地址
    fn get_backends(&self) -> Vec<BackendState>;
    
    /// 获取一个后端地址, 通过通用key
    fn backend<KEY: AsRef<str>>(this: Box<dyn LbStrategy>, key: KEY) -> Option<BackendState>
    where
        Self: Sized,
    {
        this.get_backend(key.as_ref())
    }
}