mockforge-core 0.3.114

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

//! Global risk assessment engine
//!
//! Similar to other global services, this provides a global singleton
//! for the risk assessment engine that can be accessed throughout the application.

use crate::security::risk_assessment::RiskAssessmentEngine;
use once_cell::sync::Lazy;
use std::sync::Arc;
use tokio::sync::RwLock;
use tracing::debug;

#[allow(clippy::type_complexity)]
static GLOBAL_RISK_ASSESSMENT_ENGINE: Lazy<Arc<RwLock<Option<Arc<RwLock<RiskAssessmentEngine>>>>>> =
    Lazy::new(|| Arc::new(RwLock::new(None)));

/// Initialize the global risk assessment engine
///
/// This should be called once during application startup.
/// Takes an Arc<RwLock<RiskAssessmentEngine>> to share the same instance.
pub async fn init_global_risk_assessment_engine(
    engine: Arc<RwLock<RiskAssessmentEngine>>,
) -> Result<(), crate::Error> {
    let mut global = GLOBAL_RISK_ASSESSMENT_ENGINE.write().await;
    if global.is_some() {
        return Err(crate::Error::already_initialized("Global risk assessment engine"));
    }
    *global = Some(engine);
    debug!("Global risk assessment engine initialized");
    Ok(())
}

/// Get the global risk assessment engine
///
/// Returns None if the engine has not been initialized.
pub async fn get_global_risk_assessment_engine() -> Option<Arc<RwLock<RiskAssessmentEngine>>> {
    GLOBAL_RISK_ASSESSMENT_ENGINE.read().await.clone()
}

/// Check if the global risk assessment engine is initialized
pub async fn is_risk_assessment_engine_initialized() -> bool {
    GLOBAL_RISK_ASSESSMENT_ENGINE.read().await.is_some()
}