surrealpatch/src/dbs/iterate.rs

use crate::cnf::ID_CHARS;
use crate::dbs::Iterator;
use crate::dbs::Options;
use crate::dbs::Runtime;
use crate::dbs::Transaction;
use crate::err::Error;
use crate::sql::array::Array;
use crate::sql::model::Model;
use crate::sql::table::Table;
use crate::sql::thing::Thing;
use crate::sql::value::Value;
use async_recursion::async_recursion;
use nanoid::nanoid;

impl Value {
	#[async_recursion]
	pub async fn iterate(
		self,
		ctx: &Runtime,
		opt: &Options,
		txn: &Transaction<'_>,
		ite: &mut Iterator<'_>,
	) -> Result<(), Error> {
		match self {
			Value::Array(v) => v.iterate(ctx, opt, txn, ite).await?,
			Value::Model(v) => v.iterate(ctx, opt, txn, ite).await?,
			Value::Thing(v) => v.iterate(ctx, opt, txn, ite).await?,
			Value::Table(v) => v.iterate(ctx, opt, txn, ite).await?,
			v => ite.process(ctx, opt, txn, None, v).await,
		}
		Ok(())
	}
}

impl Array {
	#[async_recursion]
	pub async fn iterate(
		self,
		ctx: &Runtime,
		opt: &Options,
		txn: &Transaction<'_>,
		ite: &mut Iterator<'_>,
	) -> Result<(), Error> {
		for v in self.value.into_iter() {
			match v {
				Value::Array(v) => v.iterate(ctx, opt, txn, ite).await?,
				Value::Model(v) => v.iterate(ctx, opt, txn, ite).await?,
				Value::Thing(v) => v.iterate(ctx, opt, txn, ite).await?,
				Value::Table(v) => v.iterate(ctx, opt, txn, ite).await?,
				v => ite.process(ctx, opt, txn, None, v).await,
			}
		}
		Ok(())
	}
}

impl Model {
	pub async fn iterate(
		self,
		ctx: &Runtime,
		opt: &Options,
		txn: &Transaction<'_>,
		ite: &mut Iterator<'_>,
	) -> Result<(), Error> {
		if ctx.is_ok() {
			if let Some(c) = self.count {
				for _ in 0..c {
					Thing {
						tb: self.table.to_string(),
						id: nanoid!(20, &ID_CHARS),
					}
					.iterate(ctx, opt, txn, ite)
					.await?;
				}
			}
			if let Some(r) = self.range {
				for x in r.0..r.1 {
					Thing {
						tb: self.table.to_string(),
						id: x.to_string(),
					}
					.iterate(ctx, opt, txn, ite)
					.await?;
				}
			}
		}
		Ok(())
	}
}

impl Thing {
	pub async fn iterate(
		self,
		ctx: &Runtime,
		opt: &Options,
		txn: &Transaction<'_>,
		ite: &mut Iterator<'_>,
	) -> Result<(), Error> {
		Ok(())
	}
}

impl Table {
	pub async fn iterate(
		self,
		ctx: &Runtime,
		opt: &Options,
		txn: &Transaction<'_>,
		ite: &mut Iterator<'_>,
	) -> Result<(), Error> {
		Ok(())
	}
}