surql_parser/upstream/sql/
mock.rs

1use crate::compat::val::range::TypedRange;
2use crate::upstream::fmt::EscapeKwFreeIdent;
3use std::ops::Bound;
4use surrealdb_types::{SqlFormat, ToSql, write_sql};
5#[derive(Clone, Debug, Eq, PartialEq, PartialOrd, Hash)]
6#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]
7pub enum Mock {
8	Count(String, i64),
9	Range(String, TypedRange<i64>),
10}
11impl ToSql for Mock {
12	fn fmt_sql(&self, f: &mut String, fmt: SqlFormat) {
13		match self {
14			Mock::Count(tb, c) => {
15				write_sql!(f, fmt, "|{}:{}|", EscapeKwFreeIdent(tb), c);
16			}
17			Mock::Range(tb, r) => {
18				write_sql!(f, fmt, "|{}:", EscapeKwFreeIdent(tb));
19				match r.start {
20					Bound::Included(x) => write_sql!(f, fmt, "{x}.."),
21					Bound::Excluded(x) => write_sql!(f, fmt, "{x}>.."),
22					Bound::Unbounded => f.push_str(".."),
23				}
24				match r.end {
25					Bound::Included(x) => write_sql!(f, fmt, "={x}|"),
26					Bound::Excluded(x) => write_sql!(f, fmt, "{x}|"),
27					Bound::Unbounded => f.push('|'),
28				}
29			}
30		}
31	}
32}
surql_parser/upstream/sql/mock.rs

surql_parser/upstream/sql/
mock.rs
