satex-load-balancer 0.5.2

Satex Load Balancer Library
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

use crate::Backend;
use arc_swap::ArcSwap;
use async_trait::async_trait;
use satex_core::Error;
use std::collections::{BTreeSet, HashMap};
use std::sync::Arc;

#[async_trait]
pub trait Discovery {
    async fn discover(&self) -> Result<(BTreeSet<Backend>, HashMap<u64, bool>), Error>;
}

pub struct StaticFixedDiscovery {
    backends: ArcSwap<BTreeSet<Backend>>,
}

impl StaticFixedDiscovery {
    pub fn new(backends: BTreeSet<Backend>) -> Self {
        Self {
            backends: ArcSwap::new(Arc::new(backends)),
        }
    }
}

#[async_trait]
impl Discovery for StaticFixedDiscovery {
    async fn discover(&self) -> Result<(BTreeSet<Backend>, HashMap<u64, bool>), Error> {
        Ok((BTreeSet::clone(&self.backends.load()), HashMap::new()))
    }
}