revoke-registry 0.3.0

Service registry and discovery for Revoke microservices framework
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

use async_trait::async_trait;
use revoke_core::{HealthStatus, Result, ServiceInfo, ServiceRegistry, Status};
use std::collections::HashMap;
use std::sync::Arc;
use tokio::sync::RwLock;
use tracing::info;
use uuid::Uuid;

pub struct MemoryRegistry {
    services: Arc<RwLock<HashMap<String, Vec<ServiceInfo>>>>,
    health_status: Arc<RwLock<HashMap<Uuid, HealthStatus>>>,
}

impl MemoryRegistry {
    pub fn new() -> Self {
        Self {
            services: Arc::new(RwLock::new(HashMap::new())),
            health_status: Arc::new(RwLock::new(HashMap::new())),
        }
    }
}

impl Default for MemoryRegistry {
    fn default() -> Self {
        Self::new()
    }
}

#[async_trait]
impl ServiceRegistry for MemoryRegistry {
    async fn register(&self, service: ServiceInfo) -> Result<()> {
        let mut services = self.services.write().await;
        services
            .entry(service.name.clone())
            .or_insert_with(Vec::new)
            .push(service.clone());
        
        info!("Service {} registered with ID {}", service.name, service.id);
        Ok(())
    }

    async fn deregister(&self, service_id: Uuid) -> Result<()> {
        let mut services = self.services.write().await;
        
        for (_, service_list) in services.iter_mut() {
            service_list.retain(|s| s.id != service_id);
        }
        
        let mut health = self.health_status.write().await;
        health.remove(&service_id);
        
        info!("Service {} deregistered", service_id);
        Ok(())
    }

    async fn get_service(&self, name: &str) -> Result<Vec<ServiceInfo>> {
        let services = self.services.read().await;
        let health = self.health_status.read().await;
        
        Ok(services
            .get(name)
            .map(|list| {
                list.iter()
                    .filter(|s| {
                        health
                            .get(&s.id)
                            .map(|h| h.status == Status::Healthy)
                            .unwrap_or(true)
                    })
                    .cloned()
                    .collect()
            })
            .unwrap_or_default())
    }

    async fn update_health(&self, status: HealthStatus) -> Result<()> {
        let mut health = self.health_status.write().await;
        health.insert(status.service_id, status);
        Ok(())
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use revoke_core::Protocol;

    #[tokio::test]
    async fn test_memory_registry() {
        let registry = MemoryRegistry::new();
        
        let service = ServiceInfo {
            id: Uuid::new_v4(),
            name: "test-service".to_string(),
            version: "1.0.0".to_string(),
            address: "127.0.0.1".to_string(),
            port: 8080,
            protocol: Protocol::Http,
            metadata: HashMap::new(),
        };
        
        registry.register(service.clone()).await.unwrap();
        
        let services = registry.get_service("test-service").await.unwrap();
        assert_eq!(services.len(), 1);
        assert_eq!(services[0].id, service.id);
        
        registry.deregister(service.id).await.unwrap();
        
        let services = registry.get_service("test-service").await.unwrap();
        assert_eq!(services.len(), 0);
    }
}