Add IntoIterator trait to SQL Model type

This commit is contained in:
Tobie Morgan Hitchcock 2022-05-30 16:05:05 +01:00
parent 45c81ad02b
commit 9020618a0e

View file

@ -1,18 +1,66 @@
use crate::sql::common::take_u64; use crate::sql::common::take_u64;
use crate::sql::error::IResult; use crate::sql::error::IResult;
use crate::sql::escape::escape_ident; use crate::sql::escape::escape_ident;
use crate::sql::id::Id;
use crate::sql::ident::ident_raw; use crate::sql::ident::ident_raw;
use crate::sql::thing::Thing;
use nom::branch::alt; use nom::branch::alt;
use nom::character::complete::char; use nom::character::complete::char;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use std::fmt; use std::fmt;
pub struct IntoIter {
model: Model,
index: u64,
}
impl Iterator for IntoIter {
type Item = Thing;
fn next(&mut self) -> Option<Thing> {
match &self.model {
Model::Count(tb, c) => {
if self.index < *c {
self.index += 1;
Some(Thing {
tb: tb.to_string(),
id: Id::rand(),
})
} else {
None
}
}
Model::Range(tb, b, e) => {
if self.index + b <= *e {
self.index += 1;
Some(Thing {
tb: tb.to_string(),
id: Id::from(self.index),
})
} else {
None
}
}
}
}
}
#[derive(Clone, Debug, Eq, PartialEq, PartialOrd, Serialize, Deserialize)] #[derive(Clone, Debug, Eq, PartialEq, PartialOrd, Serialize, Deserialize)]
pub enum Model { pub enum Model {
Count(String, u64), Count(String, u64),
Range(String, u64, u64), Range(String, u64, u64),
} }
impl IntoIterator for Model {
type Item = Thing;
type IntoIter = IntoIter;
fn into_iter(self) -> Self::IntoIter {
IntoIter {
model: self,
index: 0,
}
}
}
impl fmt::Display for Model { impl fmt::Display for Model {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
match self { match self {