datagen-rs 0.1.0

A random data generator written 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

use crate::schema::transform::Transform;
#[cfg(feature = "schema")]
use schemars::JsonSchema;
#[cfg(feature = "serialize")]
use serde::{Deserialize, Serialize};

#[derive(Debug, Clone)]
#[cfg_attr(feature = "schema", derive(JsonSchema))]
#[cfg_attr(feature = "serialize", derive(Serialize, Deserialize))]
#[cfg_attr(feature = "serialize", serde(rename_all = "camelCase"))]
pub struct Reference {
    pub reference: String,
    pub except: Option<Vec<StringOrNumber>>,
    pub keep_all: Option<bool>,
    pub transform: Option<Vec<Transform>>,
}

#[derive(Debug, Clone, PartialEq)]
#[cfg_attr(feature = "schema", derive(JsonSchema))]
#[cfg_attr(feature = "serialize", derive(Serialize, Deserialize))]
#[cfg_attr(feature = "serialize", serde(untagged, deny_unknown_fields))]
pub enum StringOrNumber {
    String(String),
    Number(f64),
}

#[cfg(feature = "map-schema")]
pub mod generate {
    use crate::generate::current_schema::CurrentSchemaRef;
    use crate::generate::generated_schema::generate::IntoGeneratedArc;
    use crate::generate::generated_schema::GeneratedSchema;
    use crate::schema::reference::{Reference, StringOrNumber};
    use crate::schema::transform::Transform;
    use crate::util::types::Result;
    use rand::prelude::SliceRandom;
    use std::sync::Arc;

    impl IntoGeneratedArc for Reference {
        fn into_generated_arc(self, schema: CurrentSchemaRef) -> Result<Arc<GeneratedSchema>> {
            let mut reference = self.reference;
            if !reference.starts_with("ref:") {
                reference = format!("ref:{reference}");
            }

            let resolved = schema.resolve_ref(reference)?;
            let Some(except) = self.except else {
                return resolved.into_random();
            };

            let Some(resolved) = resolved.into_vec() else {
                return Ok(Arc::new(GeneratedSchema::None));
            };

            let except = except
                .into_iter()
                .map(|x| match x {
                    StringOrNumber::String(string) => {
                        Ok(schema.resolve_ref(string)?.into_vec().unwrap_or(vec![]))
                    }
                    StringOrNumber::Number(number) => {
                        Ok(vec![Arc::new(GeneratedSchema::Number(number.into()))])
                    }
                })
                .collect::<Result<Vec<_>>>()?
                .into_iter()
                .flatten()
                .collect::<Vec<_>>();

            let resolved = resolved
                .iter()
                .filter(|x| !except.contains(x))
                .cloned()
                .collect::<Vec<_>>();

            Ok(if self.keep_all.unwrap_or(false) {
                Arc::new(GeneratedSchema::Array(resolved))
            } else if let Some(resolved) = resolved.choose(&mut rand::thread_rng()) {
                resolved.clone()
            } else {
                Arc::new(GeneratedSchema::None)
            })
        }

        fn get_transform(&self) -> Option<Vec<Transform>> {
            self.transform.clone()
        }
    }
}