1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20

mod random;
mod round_robin;
mod round_robin_sync;
mod single_node;

pub use crate::load_balancing::random::Random;
pub use crate::load_balancing::round_robin::RoundRobin;
pub use crate::load_balancing::round_robin_sync::RoundRobinSync;
pub use crate::load_balancing::single_node::SingleNode;

pub trait LoadBalancingStrategy<N>: Sized {
    fn init(&mut self, cluster: Vec<N>);
    fn next(&self) -> Option<&N>;
    fn remove_node<F>(&mut self, _filter: F)
    where
        F: FnMut(&N) -> bool,
    {
        // default implementation does nothing
    }
}