Add IntoIterator trait to SQL Model type
This commit is contained in:
parent
45c81ad02b
commit
9020618a0e
1 changed files with 48 additions and 0 deletions
|
@ -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 {
|
||||||
|
|
Loading…
Reference in a new issue