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

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(untagged, deny_unknown_fields))]
pub enum Integer {
    Random {
        #[cfg_attr(feature = "serialize", serde(skip_serializing_if = "Option::is_none"))]
        min: Option<i64>,
        #[cfg_attr(feature = "serialize", serde(skip_serializing_if = "Option::is_none"))]
        max: Option<i64>,
        transform: Option<Vec<Transform>>,
    },
    Constant {
        value: i64,
        transform: Option<Vec<Transform>>,
    },
}

#[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::integer::Integer;
    use crate::schema::transform::Transform;
    use crate::util::types::Result;
    use rand::Rng;

    impl IntoGenerated for Integer {
        fn into_generated(self, _: CurrentSchemaRef) -> Result<GeneratedSchema> {
            Ok(match self {
                Integer::Constant { value, .. } => GeneratedSchema::Integer(value),
                Integer::Random { min, max, .. } => {
                    let mut rng = rand::thread_rng();
                    let min = min.unwrap_or(i64::MIN);
                    let max = max.unwrap_or(i64::MAX);
                    let value = rng.gen_range(min..=max);
                    GeneratedSchema::Integer(value)
                }
            })
        }

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