mockforge-core 0.3.115

Shared logic for MockForge - routing, validation, latency, proxy
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

//! Consumer registry for tracking API consumers
//!
//! This module provides functionality for registering and querying consumers.

use crate::consumer_contracts::types::{Consumer, ConsumerIdentifier};
use std::collections::HashMap;
use std::sync::Arc;
use tokio::sync::RwLock;
use uuid::Uuid;

/// Registry for tracking consumers
#[derive(Debug, Clone)]
pub struct ConsumerRegistry {
    /// Consumers indexed by ID
    consumers: Arc<RwLock<HashMap<String, Consumer>>>,
    /// Index by identifier for fast lookup
    identifier_index: Arc<RwLock<HashMap<ConsumerIdentifier, String>>>,
}

impl ConsumerRegistry {
    /// Create a new consumer registry
    pub fn new() -> Self {
        Self {
            consumers: Arc::new(RwLock::new(HashMap::new())),
            identifier_index: Arc::new(RwLock::new(HashMap::new())),
        }
    }

    /// Register a consumer
    pub async fn register(&self, consumer: Consumer) {
        let id = consumer.id.clone();
        let identifier = consumer.identifier.clone();

        // Store consumer
        {
            let mut consumers = self.consumers.write().await;
            consumers.insert(id.clone(), consumer);
        }

        // Update identifier index
        {
            let mut index = self.identifier_index.write().await;
            index.insert(identifier, id);
        }
    }

    /// Get consumer by ID
    pub async fn get_by_id(&self, id: &str) -> Option<Consumer> {
        let consumers = self.consumers.read().await;
        consumers.get(id).cloned()
    }

    /// Get consumer by identifier
    pub async fn get_by_identifier(&self, identifier: &ConsumerIdentifier) -> Option<Consumer> {
        let consumer_id = {
            let index = self.identifier_index.read().await;
            index.get(identifier).cloned()
        };

        if let Some(consumer_id) = consumer_id {
            self.get_by_id(&consumer_id).await
        } else {
            None
        }
    }

    /// List all consumers
    pub async fn list_all(&self) -> Vec<Consumer> {
        let consumers = self.consumers.read().await;
        consumers.values().cloned().collect()
    }

    /// Remove a consumer
    pub async fn remove(&self, id: &str) -> Option<Consumer> {
        let consumer = {
            let mut consumers = self.consumers.write().await;
            consumers.remove(id)
        };

        if let Some(ref consumer) = consumer {
            let mut index = self.identifier_index.write().await;
            index.remove(&consumer.identifier);
        }

        consumer
    }

    /// Create or get consumer by identifier
    pub async fn get_or_create(
        &self,
        identifier: ConsumerIdentifier,
        name: String,
        workspace_id: Option<String>,
    ) -> Consumer {
        // Try to get existing consumer
        if let Some(consumer) = self.get_by_identifier(&identifier).await {
            return consumer;
        }

        // Create new consumer
        let id = Uuid::new_v4().to_string();
        let consumer = Consumer::new(id, identifier, name, workspace_id);
        self.register(consumer.clone()).await;
        consumer
    }
}

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