mocra 0.3.0

A distributed, event-driven crawling and data collection framework
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

use crate::common::state::State;
use crate::engine::api::profile_store::ProfileControlPlaneStore;
use crate::engine::task::task_manager::TaskManager;
use crate::queue::QueueManager;
use metrics_exporter_prometheus::PrometheusHandle;
use std::sync::Arc;

#[derive(Clone)]
pub struct ApiState {
    pub(crate) queue_manager: Arc<QueueManager>,
    pub(crate) prometheus_handle: Option<PrometheusHandle>,
    pub(crate) state: Arc<State>,
    pub(crate) task_manager: Arc<TaskManager>,
    pub(crate) profile_store: Arc<ProfileControlPlaneStore>,
}

impl ApiState {
    pub fn new(
        queue_manager: Arc<QueueManager>,
        prometheus_handle: Option<PrometheusHandle>,
        state: Arc<State>,
        task_manager: Arc<TaskManager>,
    ) -> Self {
        let profile_store = state.profile_store.clone();
        Self {
            queue_manager,
            prometheus_handle,
            state,
            task_manager,
            profile_store,
        }
    }
}