data_generator 0.1.119

RDF data shapes implementation in Rust
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101

use crate::field_generators::{FieldGenerator, FieldGeneratorFactory};
use crate::{DataGeneratorError, Result};
use std::collections::HashMap;

/// Registry for managing field generators
pub struct FieldGeneratorRegistry {
    generators: HashMap<String, Box<dyn FieldGenerator>>,
    factories: HashMap<String, Box<dyn FieldGeneratorFactory>>,
    datatype_mappings: HashMap<String, String>,
}

impl Default for FieldGeneratorRegistry {
    fn default() -> Self {
        Self::new()
    }
}

impl FieldGeneratorRegistry {
    pub fn new() -> Self {
        Self {
            generators: HashMap::new(),
            factories: HashMap::new(),
            datatype_mappings: HashMap::new(),
        }
    }

    /// Register a field generator instance
    pub fn register_generator(&mut self, generator: Box<dyn FieldGenerator>) {
        let name = generator.name().to_string();

        // Register mappings for supported datatypes
        for datatype in generator.supported_datatypes() {
            self.datatype_mappings.insert(datatype, name.clone());
        }

        self.generators.insert(name, generator);
    }

    /// Register a field generator factory
    pub fn register_factory(&mut self, factory: Box<dyn FieldGeneratorFactory>) {
        let name = factory.generator_name().to_string();
        self.factories.insert(name, factory);
    }

    /// Get a generator by name
    pub fn get_generator(&self, name: &str) -> Result<&dyn FieldGenerator> {
        self.generators
            .get(name)
            .map(|g| g.as_ref())
            .ok_or_else(|| {
                DataGeneratorError::FieldGeneration(format!("Generator '{name}' not found"))
            })
    }

    /// Get the default generator for a datatype
    pub fn get_default_generator(&self, datatype: &str) -> Result<&dyn FieldGenerator> {
        let generator_name = self.datatype_mappings.get(datatype).ok_or_else(|| {
            DataGeneratorError::FieldGeneration(format!(
                "No generator found for datatype '{datatype}'"
            ))
        })?;

        self.get_generator(generator_name)
    }

    /// Create a generator instance from a factory
    pub fn create_generator(&self, name: &str) -> Result<Box<dyn FieldGenerator>> {
        let factory = self.factories.get(name).ok_or_else(|| {
            DataGeneratorError::FieldGeneration(format!("Factory for generator '{name}' not found"))
        })?;

        factory.create()
    }

    /// Register all default generators
    pub fn register_default_generators(&mut self) -> Result<()> {
        use crate::field_generators::basic::*;
        use crate::field_generators::pattern::PatternGenerator;

        self.register_generator(Box::new(StringGenerator));
        self.register_generator(Box::new(IntegerGenerator));
        self.register_generator(Box::new(DecimalGenerator));
        self.register_generator(Box::new(BooleanGenerator));
        self.register_generator(Box::new(DateGenerator));
        self.register_generator(Box::new(DateTimeGenerator));
        self.register_generator(Box::new(UriGenerator));
        self.register_generator(Box::new(PatternGenerator));

        Ok(())
    }

    /// List all registered generators
    pub fn list_generators(&self) -> Vec<&str> {
        self.generators.keys().map(|s| s.as_str()).collect()
    }

    /// List all supported datatypes
    pub fn list_datatypes(&self) -> Vec<&str> {
        self.datatype_mappings.keys().map(|s| s.as_str()).collect()
    }
}