feat: add gen_prime generator

README.md

@@ -21,7 +21,7 @@
     - [domain](#domain)
     - [http method](#http-method)
     - [ipv4](#ipv4)
-  - [Generators with arguments](#generators-with-arguments)
+    - [gen_prime](#gen_prime)
     - [enum](#enum)
     - [int](#int)
     - [private ipv4](#private-ipv4)
@@ -153,6 +153,20 @@ let ip: String = gen_ipv4();
 // ip = "168.11.40.75"
 ```
 
+#### gen_prime
+
+Returns one of the first 1000 prime numners, randomely.
+
+Function signature
+```rust
+gen_prime() -> usize
+``` 
+
+Example call
+```rust
+let prime: usize = gen_prime();
+// prime = 6323
+```
 ### Generators with arguments
 [⬆️ Back to Top](#table-of-contents)
 

src/data.rs

@@ -9,6 +9,7 @@ use std::error::Error;
 
 pub mod corpora;
 pub mod tlds;
+pub mod primes;
 
 /// JSONDataset represents a generic data structure for storing the parsed JSON. Each JSON taken
 /// from Corpora has a `data` field which is an Array of Strings in JSON (= Vec<String> in Rust).
@@ -79,3 +80,11 @@ pub fn gen_switch(name: String) -> String {
 pub fn gen_corpora_switch(name: String) -> String {
     return gen_switch(name);
 }
+
+// gen_prime returns a random of the first 1000 prime numbers
+pub fn gen_prime() -> usize {
+    let mut rnd = rand::thread_rng();
+    let index = rnd.gen_range(0..primes::DATA_PRIMES.len() - 1);
+
+    primes::DATA_PRIMES[index]
+}

src/data/corpora.rs

@@ -3902,4 +3902,4 @@ pub const DATA_MOOD: &str = r#"
 		"zealous"
 	]
 }
-"#;
+"#;