spawnflake 0.2.9

Generate pattern based random data for mysql
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

use rand::Rng;

use crate::datastores::my_sql::const_types::db_types;

/// generates random alpha characters (latin)
/// character pool: abcdefghijklmnopqrstuvwxyz
pub fn generate_alphas(string_type: &str) -> Option<String> {
    let chars = "abcdefghijklmnopqrstuvwxyz";

    if string_type.eq(db_types::TEXT)
        || string_type.eq(db_types::LONG_TEXT)
        || string_type.eq(db_types::MEDIUM_TEXT)
    {
        let mut result = "".to_string();
        for _i in 0..255 {
            let rng = rand::thread_rng().gen_range(0..25);
            result = format!("{}{}", result, chars.chars().nth(rng).unwrap());
        }
        Some(result)
    } else {
        let start_bytes = string_type.find('(').unwrap_or(0);
        let end_bytes = string_type.find(')').unwrap_or(string_type.len());

        let v_size = &string_type[(start_bytes + 1)..end_bytes];
        let mut result = "".to_string();
        for _i in 0..v_size.parse::<i32>().unwrap_or(1) {
            let rng = rand::thread_rng().gen_range(0..25);
            result = format!("{}{}", result, chars.chars().nth(rng).unwrap());
        }
        Some(result)
    }
}

#[cfg(test)]
mod tests {
    use crate::string_generator::strings::generate_alphas;

    #[test]
    fn get_random_alpha_string_with_five_chars() {
        let alphas = generate_alphas("varchar(5)").expect("string");

        assert_eq!(alphas.len(), 5);
        assert_eq!(alphas.chars().all(char::is_alphabetic), true);
    }

    #[test]
    fn get_random_alpha_string_with_255_chars() {
        let alphas = generate_alphas("varchar(255)").expect("string");

        assert_eq!(alphas.len(), 255);
        assert_eq!(alphas.chars().all(char::is_alphabetic), true);
    }
}