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

use crate::util::types::Result;
use rand::seq::SliceRandom;

pub struct SequentialVec<T> {
    data: Vec<T>,
    index: usize,
}

impl<T> SequentialVec<T> {
    pub fn new(data: Vec<T>) -> Result<Self> {
        if data.is_empty() {
            Err("Cannot create SequentialVec from empty vec".into())
        } else {
            Ok(Self { data, index: 0 })
        }
    }

    pub fn random(&self) -> &T {
        self.data.choose(&mut rand::thread_rng()).unwrap()
    }

    pub fn next_value(&mut self) -> &T {
        if self.index >= self.data.len() {
            self.index = 0;
            &self.data[0]
        } else {
            let idx = self.index;
            self.index += 1;
            &self.data[idx]
        }
    }
}