meilibridge 0.1.6

High-performance PostgreSQL to Meilisearch connector
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

use crate::health::HealthRegistry;
use crate::pipeline::PipelineOrchestrator;
use crate::source::postgres::StatementCache;
use crate::sync::SyncTaskManager;
use std::sync::Arc;
use tokio::sync::RwLock;

/// Shared state for the API server
#[derive(Clone)]
pub struct ApiState {
    pub orchestrator: Arc<RwLock<PipelineOrchestrator>>,
    pub task_manager: Arc<RwLock<SyncTaskManager>>,
    pub health_registry: Option<Arc<HealthRegistry>>,
    pub postgres_cache: Option<Arc<StatementCache>>,
}

impl ApiState {
    pub fn new(
        orchestrator: Arc<RwLock<PipelineOrchestrator>>,
        task_manager: Arc<RwLock<SyncTaskManager>>,
    ) -> Self {
        Self {
            orchestrator,
            task_manager,
            health_registry: None,
            postgres_cache: None,
        }
    }

    pub fn with_health_registry(mut self, registry: Arc<HealthRegistry>) -> Self {
        self.health_registry = Some(registry);
        self
    }

    pub fn with_postgres_cache(mut self, cache: Arc<StatementCache>) -> Self {
        self.postgres_cache = Some(cache);
        self
    }

    pub fn health_registry(&self) -> Option<&Arc<HealthRegistry>> {
        self.health_registry.as_ref()
    }
}