solverforge-macros 0.8.8

Derive macros for SolverForge constraint solver
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 std::collections::HashMap;
use std::sync::{Mutex, OnceLock};

#[derive(Clone, Debug)]
pub(crate) struct StandardVariableMetadata {
    pub field_name: String,
    pub allows_unassigned: bool,
    pub value_range_provider: Option<String>,
    pub countable_range: Option<(i64, i64)>,
    pub provider_is_entity_field: bool,
}

#[derive(Clone, Debug)]
pub(crate) struct StandardEntityMetadata {
    pub variables: Vec<StandardVariableMetadata>,
}

static STANDARD_ENTITY_REGISTRY: OnceLock<Mutex<HashMap<String, StandardEntityMetadata>>> =
    OnceLock::new();

fn registry() -> &'static Mutex<HashMap<String, StandardEntityMetadata>> {
    STANDARD_ENTITY_REGISTRY.get_or_init(|| Mutex::new(HashMap::new()))
}

pub(crate) fn record_standard_entity_metadata(
    entity_name: &str,
    variables: Vec<StandardVariableMetadata>,
) {
    let mut registry = registry()
        .lock()
        .expect("solverforge standard metadata registry should be available");
    registry.insert(
        entity_name.to_string(),
        StandardEntityMetadata { variables },
    );
}

pub(crate) fn lookup_standard_entity_metadata(entity_name: &str) -> Option<StandardEntityMetadata> {
    let registry = registry()
        .lock()
        .expect("solverforge standard metadata registry should be available");
    registry.get(entity_name).cloned()
}