mockiapi 0.1.8

A flexible, feature-rich mock API server with an intuitive frontend that simplifies development.
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

use std::collections::HashMap;
use tokio::sync::RwLock;
use crate::models::grpc::GrpcMockResponse;

type MethodKey = String; // format: "MyService.MyMethod"

#[derive(Default)]
pub struct GrpcRegistry {
    mocks: RwLock<HashMap<MethodKey, GrpcMockResponse>>,
}

impl GrpcRegistry {
    pub fn new() -> Self {
        Self::default()
    }

    pub async fn register_mock(&self, service: &str, method: &str, response: GrpcMockResponse) {
        let key = format!("{}.{}", service, method);
        self.mocks.write().await.insert(key, response);
    }

    pub async fn get_mock(&self, service: &str, method: &str) -> Option<GrpcMockResponse> {
        let registry = self.mocks.read().await;

        let service_prefix = format!("{}.", service);
        let mut service_exists = false;

        for key in registry.keys() {
            if key.starts_with(&service_prefix) {
                service_exists = true;
                if key == &format!("{}.{}", service, method) {
                    return registry.get(key).cloned();
                }
            }
        }

        if !service_exists {
            log::warn!("gRPC service '{}' not found", service);
        } else {
            log::warn!("gRPC method '{}' not found in service '{}'", method, service);
        }

        None
    }
}