qail_core/ast/
operators.rs

1use serde::{Deserialize, Serialize};
2
3/// The action type (SQL operation).
4#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
5pub enum Action {
6    Get,
7    Cnt,
8    Set,
9    Del,
10    Add,
11    Gen,
12    Make,
13    Drop,
14    Mod,
15    Over,
16    With,
17    Index,
18    DropIndex,
19    Alter,
20    AlterDrop,
21    AlterType,
22    TxnStart,
23    TxnCommit,
24    TxnRollback,
25    Put,
26    DropCol,
27    RenameCol,
28    JsonTable,
29    Export,
30    Truncate,
31    Explain,
32    ExplainAnalyze,
33    Lock,
34    CreateMaterializedView,
35    RefreshMaterializedView,
36    DropMaterializedView,
37    Listen,
38    Notify,
39    Unlisten,
40    Savepoint,
41    ReleaseSavepoint,
42    RollbackToSavepoint,
43    CreateView,
44    DropView,
45    Search,
46    Upsert,
47    Scroll,
48    CreateCollection,
49    DeleteCollection,
50    CreateFunction,
51    DropFunction,
52    CreateTrigger,
53    DropTrigger,
54    RedisGet,
55    RedisSet,
56    RedisDel,
57    RedisIncr,
58    RedisDecr,
59    RedisTtl,
60    RedisExpire,
61    RedisExists,
62    RedisMGet,
63    RedisMSet,
64    RedisPing,
65    CreateExtension,
66    DropExtension,
67    CommentOn,
68    CreateSequence,
69    DropSequence,
70    CreateEnum,
71    DropEnum,
72    AlterEnumAddValue,
73    AlterSetNotNull,
74    AlterDropNotNull,
75    AlterSetDefault,
76    AlterDropDefault,
77    AlterEnableRls,
78    AlterDisableRls,
79    AlterForceRls,
80    AlterNoForceRls,
81}
82
83impl std::fmt::Display for Action {
84    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
85        match self {
86            Action::Get => write!(f, "GET"),
87            Action::Cnt => write!(f, "CNT"),
88            Action::Set => write!(f, "SET"),
89            Action::Del => write!(f, "DEL"),
90            Action::Add => write!(f, "ADD"),
91            Action::Gen => write!(f, "GEN"),
92            Action::Make => write!(f, "MAKE"),
93            Action::Drop => write!(f, "DROP"),
94            Action::Mod => write!(f, "MOD"),
95            Action::Over => write!(f, "OVER"),
96            Action::With => write!(f, "WITH"),
97            Action::Index => write!(f, "INDEX"),
98            Action::DropIndex => write!(f, "DROP_INDEX"),
99            Action::Alter => write!(f, "ALTER"),
100            Action::AlterDrop => write!(f, "ALTER_DROP"),
101            Action::AlterType => write!(f, "ALTER_TYPE"),
102            Action::TxnStart => write!(f, "TXN_START"),
103            Action::TxnCommit => write!(f, "TXN_COMMIT"),
104            Action::TxnRollback => write!(f, "TXN_ROLLBACK"),
105            Action::Put => write!(f, "PUT"),
106            Action::DropCol => write!(f, "DROP_COL"),
107            Action::RenameCol => write!(f, "RENAME_COL"),
108            Action::JsonTable => write!(f, "JSON_TABLE"),
109            Action::Export => write!(f, "EXPORT"),
110            Action::Truncate => write!(f, "TRUNCATE"),
111            Action::Explain => write!(f, "EXPLAIN"),
112            Action::ExplainAnalyze => write!(f, "EXPLAIN_ANALYZE"),
113            Action::Lock => write!(f, "LOCK"),
114            Action::CreateMaterializedView => write!(f, "CREATE_MATERIALIZED_VIEW"),
115            Action::RefreshMaterializedView => write!(f, "REFRESH_MATERIALIZED_VIEW"),
116            Action::DropMaterializedView => write!(f, "DROP_MATERIALIZED_VIEW"),
117            Action::Listen => write!(f, "LISTEN"),
118            Action::Notify => write!(f, "NOTIFY"),
119            Action::Unlisten => write!(f, "UNLISTEN"),
120            Action::Savepoint => write!(f, "SAVEPOINT"),
121            Action::ReleaseSavepoint => write!(f, "RELEASE_SAVEPOINT"),
122            Action::RollbackToSavepoint => write!(f, "ROLLBACK_TO_SAVEPOINT"),
123            Action::CreateView => write!(f, "CREATE_VIEW"),
124            Action::DropView => write!(f, "DROP_VIEW"),
125            Action::Search => write!(f, "SEARCH"),
126            Action::Upsert => write!(f, "UPSERT"),
127            Action::Scroll => write!(f, "SCROLL"),
128            Action::CreateCollection => write!(f, "CREATE_COLLECTION"),
129            Action::DeleteCollection => write!(f, "DELETE_COLLECTION"),
130            Action::CreateFunction => write!(f, "CREATE_FUNCTION"),
131            Action::DropFunction => write!(f, "DROP_FUNCTION"),
132            Action::CreateTrigger => write!(f, "CREATE_TRIGGER"),
133            Action::DropTrigger => write!(f, "DROP_TRIGGER"),
134            Action::RedisGet => write!(f, "REDIS_GET"),
135            Action::RedisSet => write!(f, "REDIS_SET"),
136            Action::RedisDel => write!(f, "REDIS_DEL"),
137            Action::RedisIncr => write!(f, "REDIS_INCR"),
138            Action::RedisDecr => write!(f, "REDIS_DECR"),
139            Action::RedisTtl => write!(f, "REDIS_TTL"),
140            Action::RedisExpire => write!(f, "REDIS_EXPIRE"),
141            Action::RedisExists => write!(f, "REDIS_EXISTS"),
142            Action::RedisMGet => write!(f, "REDIS_MGET"),
143            Action::RedisMSet => write!(f, "REDIS_MSET"),
144            Action::RedisPing => write!(f, "REDIS_PING"),
145            Action::CreateExtension => write!(f, "CREATE_EXTENSION"),
146            Action::DropExtension => write!(f, "DROP_EXTENSION"),
147            Action::CommentOn => write!(f, "COMMENT_ON"),
148            Action::CreateSequence => write!(f, "CREATE_SEQUENCE"),
149            Action::DropSequence => write!(f, "DROP_SEQUENCE"),
150            Action::CreateEnum => write!(f, "CREATE_ENUM"),
151            Action::DropEnum => write!(f, "DROP_ENUM"),
152            Action::AlterEnumAddValue => write!(f, "ALTER_ENUM_ADD_VALUE"),
153            Action::AlterSetNotNull => write!(f, "ALTER_SET_NOT_NULL"),
154            Action::AlterDropNotNull => write!(f, "ALTER_DROP_NOT_NULL"),
155            Action::AlterSetDefault => write!(f, "ALTER_SET_DEFAULT"),
156            Action::AlterDropDefault => write!(f, "ALTER_DROP_DEFAULT"),
157            Action::AlterEnableRls => write!(f, "ALTER_ENABLE_RLS"),
158            Action::AlterDisableRls => write!(f, "ALTER_DISABLE_RLS"),
159            Action::AlterForceRls => write!(f, "ALTER_FORCE_RLS"),
160            Action::AlterNoForceRls => write!(f, "ALTER_NO_FORCE_RLS"),
161        }
162    }
163}
164
165/// Logical operator between conditions.
166#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize, Default)]
167pub enum LogicalOp {
168    #[default]
169    And,
170    Or,
171}
172
173#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
174pub enum SortOrder {
175    Asc,
176    Desc,
177    AscNullsFirst,
178    AscNullsLast,
179    DescNullsFirst,
180    DescNullsLast,
181}
182
183#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
184pub enum Operator {
185    Eq,
186    Ne,
187    Gt,
188    Gte,
189    Lt,
190    Lte,
191    Fuzzy,
192    In,
193    NotIn,
194    IsNull,
195    IsNotNull,
196    Contains,
197    KeyExists,
198    JsonExists,
199    JsonQuery,
200    JsonValue,
201    Like,
202    NotLike,
203    ILike,
204    NotILike,
205    Between,
206    NotBetween,
207    Exists,
208    NotExists,
209    Regex,
210    RegexI,
211    SimilarTo,
212    ContainedBy,
213    Overlaps,
214    TextSearch,
215}
216
217impl Operator {
218    /// For simple operators, returns the symbol directly.
219    /// For complex operators (BETWEEN, EXISTS), returns the keyword.
220    pub fn sql_symbol(&self) -> &'static str {
221        match self {
222            Operator::Eq => "=",
223            Operator::Ne => "!=",
224            Operator::Gt => ">",
225            Operator::Gte => ">=",
226            Operator::Lt => "<",
227            Operator::Lte => "<=",
228            Operator::Fuzzy => "ILIKE",
229            Operator::In => "IN",
230            Operator::NotIn => "NOT IN",
231            Operator::IsNull => "IS NULL",
232            Operator::IsNotNull => "IS NOT NULL",
233            Operator::Contains => "@>",
234            Operator::KeyExists => "?",
235            Operator::JsonExists => "JSON_EXISTS",
236            Operator::JsonQuery => "JSON_QUERY",
237            Operator::JsonValue => "JSON_VALUE",
238            Operator::Like => "LIKE",
239            Operator::NotLike => "NOT LIKE",
240            Operator::ILike => "ILIKE",
241            Operator::NotILike => "NOT ILIKE",
242            Operator::Between => "BETWEEN",
243            Operator::NotBetween => "NOT BETWEEN",
244            Operator::Exists => "EXISTS",
245            Operator::NotExists => "NOT EXISTS",
246            Operator::Regex => "~",
247            Operator::RegexI => "~*",
248            Operator::SimilarTo => "SIMILAR TO",
249            Operator::ContainedBy => "<@",
250            Operator::Overlaps => "&&",
251            Operator::TextSearch => "@@",
252        }
253    }
254
255    /// IS NULL, IS NOT NULL, EXISTS, NOT EXISTS don't need values.
256    pub fn needs_value(&self) -> bool {
257        !matches!(
258            self,
259            Operator::IsNull | Operator::IsNotNull | Operator::Exists | Operator::NotExists
260        )
261    }
262
263    pub fn is_simple_binary(&self) -> bool {
264        matches!(
265            self,
266            Operator::Eq
267                | Operator::Ne
268                | Operator::Gt
269                | Operator::Gte
270                | Operator::Lt
271                | Operator::Lte
272                | Operator::Like
273                | Operator::NotLike
274                | Operator::ILike
275                | Operator::NotILike
276        )
277    }
278}
279
280#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
281pub enum AggregateFunc {
282    Count,
283    Sum,
284    Avg,
285    Min,
286    Max,
287    ArrayAgg,
288    StringAgg,
289    JsonAgg,
290    JsonbAgg,
291    BoolAnd,
292    BoolOr,
293}
294
295impl std::fmt::Display for AggregateFunc {
296    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
297        match self {
298            AggregateFunc::Count => write!(f, "COUNT"),
299            AggregateFunc::Sum => write!(f, "SUM"),
300            AggregateFunc::Avg => write!(f, "AVG"),
301            AggregateFunc::Min => write!(f, "MIN"),
302            AggregateFunc::Max => write!(f, "MAX"),
303            AggregateFunc::ArrayAgg => write!(f, "ARRAY_AGG"),
304            AggregateFunc::StringAgg => write!(f, "STRING_AGG"),
305            AggregateFunc::JsonAgg => write!(f, "JSON_AGG"),
306            AggregateFunc::JsonbAgg => write!(f, "JSONB_AGG"),
307            AggregateFunc::BoolAnd => write!(f, "BOOL_AND"),
308            AggregateFunc::BoolOr => write!(f, "BOOL_OR"),
309        }
310    }
311}
312
313/// Join Type
314#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
315pub enum JoinKind {
316    Inner,
317    Left,
318    Right,
319    Lateral,
320    Full,
321    Cross,
322}
323
324/// Set operation type for combining queries
325#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
326pub enum SetOp {
327    Union,
328    UnionAll,
329    Intersect,
330    Except,
331}
332
333#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
334pub enum ModKind {
335    Add,
336    Drop,
337}
338
339/// GROUP BY mode for advanced aggregations
340#[derive(Debug, Clone, PartialEq, Eq, Default, Serialize, Deserialize)]
341pub enum GroupByMode {
342    #[default]
343    Simple,
344    Rollup,
345    Cube,
346    GroupingSets(Vec<Vec<String>>),
347}
348
349impl GroupByMode {
350    /// Check if this is the default Simple mode (for serde skip)
351    pub fn is_simple(&self) -> bool {
352        matches!(self, GroupByMode::Simple)
353    }
354}
355
356/// Row locking mode for SELECT...FOR UPDATE/SHARE
357#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
358pub enum LockMode {
359    Update,
360    NoKeyUpdate,
361    Share,
362    KeyShare,
363}
364
365/// OVERRIDING clause for INSERT with GENERATED columns
366#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
367pub enum OverridingKind {
368    SystemValue,
369    UserValue,
370}
371
372/// TABLESAMPLE sampling method
373#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
374pub enum SampleMethod {
375    Bernoulli,
376    System,
377}
378
379/// Distance metric for vector similarity
380#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize, Default)]
381pub enum Distance {
382    #[default]
383    Cosine,
384    Euclid,
385    Dot,
386}
qail_core/ast/operators.rs

qail_core/ast/
operators.rs
