use crate::Generator;
pub struct StringIter<'a> {
c: &'a Generator,
n: u128,
i: u128,
}
impl<'a> Iterator for StringIter<'a> {
type Item = String;
fn next(&mut self) -> Option<Self::Item> {
if self.i == self.n {
None
} else {
self.i += 1;
Some(self.c.generate_one(self.i - 1))
}
}
}
#[cfg(feature = "with_rand")]
impl<'a> StringIter<'a> {
pub fn random(&self) -> String {
let num = rand::random::<u128>() % self.n;
self.c.generate_one(num)
}
}
impl<'a> From<&'a Generator> for StringIter<'a> {
fn from(c: &'a Generator) -> Self {
Self {
c,
n: c.len(),
i: 0,
}
}
}