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

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

#[cfg(feature = "generate")]
static mut COUNTER: AtomicI64 = AtomicI64::new(0);

#[derive(Debug, Clone)]
#[cfg_attr(feature = "schema", derive(JsonSchema))]
#[cfg_attr(feature = "serialize", derive(Serialize, Deserialize))]
pub struct Counter {
    pub step: Option<i64>,
    pub 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::counter::{Counter, COUNTER};
    use crate::schema::transform::Transform;
    use crate::util::types::Result;
    use std::sync::atomic::Ordering;

    impl IntoGenerated for Counter {
        fn into_generated(self, _schema: CurrentSchemaRef) -> Result<GeneratedSchema> {
            let value = unsafe { COUNTER.fetch_add(1, Ordering::SeqCst) };
            Ok(GeneratedSchema::Integer(value))
        }

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