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

use crate::schema::transform::Transform;
#[cfg(feature = "schema")]
use schemars::JsonSchema;

#[derive(Debug, Clone)]
#[cfg_attr(feature = "schema", derive(JsonSchema))]
#[cfg_attr(feature = "serialize", derive(Serialize, Deserialize))]
#[cfg_attr(feature = "serialize", serde(untagged, deny_unknown_fields))]
pub enum Number {
    Random {
        #[cfg_attr(feature = "serialize", serde(skip_serializing_if = "Option::is_none"))]
        min: Option<f64>,
        #[cfg_attr(feature = "serialize", serde(skip_serializing_if = "Option::is_none"))]
        max: Option<f64>,
        #[cfg_attr(feature = "serialize", serde(skip_serializing_if = "Option::is_none"))]
        precision: Option<u8>,
        transform: Option<Vec<Transform>>,
    },
    Constant {
        value: f64,
        transform: Option<Vec<Transform>>,
    },
}

#[cfg(feature = "serialize")]
use serde::{Deserialize, Serialize};

#[cfg(feature = "generate")]
pub mod generate {
    use crate::generate::current_schema::CurrentSchemaRef;
    use crate::generate::generated_schema::generate::IntoGenerated;
    use crate::generate::generated_schema::GeneratedSchema;
    use crate::schema::number::Number;
    use crate::schema::transform::Transform;
    use crate::util::types::Result;
    use rand::Rng;

    impl IntoGenerated for Number {
        fn into_generated(self, _: CurrentSchemaRef) -> Result<GeneratedSchema> {
            Ok(match self {
                Number::Constant { value, .. } => GeneratedSchema::Number(value.into()),
                Number::Random {
                    min,
                    max,
                    precision,
                    ..
                } => {
                    let mut rng = rand::thread_rng();
                    let min = min.unwrap_or(f64::MIN);
                    let max = max.unwrap_or(f64::MAX);
                    let mut value = rng.gen_range(min..=max);
                    if let Some(precision) = precision {
                        value = (value * 10.0_f64.powi(precision as i32)).round()
                            / 10.0_f64.powi(precision as i32);
                    }
                    GeneratedSchema::Number(value.into())
                }
            })
        }

        fn get_transform(&self) -> Option<Vec<Transform>> {
            match self {
                Number::Constant { transform, .. } => transform.clone(),
                Number::Random { transform, .. } => transform.clone(),
            }
        }
    }
}