78 lines
1.5 KiB
Rust
78 lines
1.5 KiB
Rust
use crate::sql::{escape::escape_ident, Id, Thing};
|
|
use revision::revisioned;
|
|
use serde::{Deserialize, Serialize};
|
|
use std::fmt;
|
|
|
|
pub(crate) const TOKEN: &str = "$surrealdb::private::sql::Mock";
|
|
|
|
pub struct IntoIter {
|
|
model: Mock,
|
|
index: u64,
|
|
}
|
|
|
|
impl Iterator for IntoIter {
|
|
type Item = Thing;
|
|
fn next(&mut self) -> Option<Thing> {
|
|
match &self.model {
|
|
Mock::Count(tb, c) => {
|
|
if self.index < *c {
|
|
self.index += 1;
|
|
Some(Thing {
|
|
tb: tb.to_string(),
|
|
id: Id::rand(),
|
|
})
|
|
} else {
|
|
None
|
|
}
|
|
}
|
|
Mock::Range(tb, b, e) => {
|
|
if self.index == 0 {
|
|
self.index = *b - 1;
|
|
}
|
|
if self.index < *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, Hash)]
|
|
#[serde(rename = "$surrealdb::private::sql::Mock")]
|
|
#[revisioned(revision = 1)]
|
|
#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]
|
|
pub enum Mock {
|
|
Count(String, u64),
|
|
Range(String, u64, u64),
|
|
// Add new variants here
|
|
}
|
|
|
|
impl IntoIterator for Mock {
|
|
type Item = Thing;
|
|
type IntoIter = IntoIter;
|
|
fn into_iter(self) -> Self::IntoIter {
|
|
IntoIter {
|
|
model: self,
|
|
index: 0,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl fmt::Display for Mock {
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
match self {
|
|
Mock::Count(tb, c) => {
|
|
write!(f, "|{}:{}|", escape_ident(tb), c)
|
|
}
|
|
Mock::Range(tb, b, e) => {
|
|
write!(f, "|{}:{}..{}|", escape_ident(tb), b, e)
|
|
}
|
|
}
|
|
}
|
|
}
|