rs-zero 0.1.1

Rust-first microservice framework inspired by go-zero engineering practices
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

use crate::discovery::{Discovery, DiscoveryError, DiscoveryResult, ServiceInstance};
use async_trait::async_trait;

use crate::discovery_kube::{KubeDiscoveryConfig, map_endpoints};

/// Kubernetes discovery facade backed by a refreshable fixture snapshot.
#[derive(Debug, Clone)]
pub struct KubeDiscovery {
    config: KubeDiscoveryConfig,
    service: String,
    snapshot: String,
}

impl KubeDiscovery {
    /// Creates a discovery facade from a fixture snapshot.
    pub fn from_snapshot(
        config: KubeDiscoveryConfig,
        service: impl Into<String>,
        snapshot: impl Into<String>,
    ) -> Self {
        Self {
            config,
            service: service.into(),
            snapshot: snapshot.into(),
        }
    }
}

#[async_trait]
impl Discovery for KubeDiscovery {
    async fn discover(&self, service: &str) -> DiscoveryResult<Vec<ServiceInstance>> {
        if service != self.service {
            return Err(DiscoveryError::NoInstances {
                service: service.to_string(),
            });
        }
        let instances = map_endpoints(&self.config, &self.snapshot).map_err(|error| {
            DiscoveryError::Resolve {
                host: "kubernetes".to_string(),
                message: error.to_string(),
            }
        })?;
        if instances.is_empty() {
            Err(DiscoveryError::NoInstances {
                service: service.to_string(),
            })
        } else {
            Ok(instances)
        }
    }
}