qail-core 0.27.8

AST-native query builder - type-safe expressions, zero SQL strings
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

//! CASE WHEN expression builders.

use crate::ast::{Condition, Expr};

/// Start a CASE WHEN expression
pub fn case_when(condition: Condition, then_expr: impl Into<Expr>) -> CaseBuilder {
    CaseBuilder {
        when_clauses: vec![(condition, Box::new(then_expr.into()))],
        else_value: None,
        alias: None,
    }
}

/// Builder for CASE expressions
#[derive(Debug, Clone)]
pub struct CaseBuilder {
    pub(crate) when_clauses: Vec<(Condition, Box<Expr>)>,
    pub(crate) else_value: Option<Box<Expr>>,
    pub(crate) alias: Option<String>,
}

impl CaseBuilder {
    /// Add another WHEN clause
    pub fn when(mut self, condition: Condition, then_expr: impl Into<Expr>) -> Self {
        self.when_clauses
            .push((condition, Box::new(then_expr.into())));
        self
    }

    /// Add ELSE clause
    pub fn otherwise(mut self, else_expr: impl Into<Expr>) -> Self {
        self.else_value = Some(Box::new(else_expr.into()));
        self
    }

    /// Add alias (AS name)
    pub fn alias(mut self, name: &str) -> Expr {
        self.alias = Some(name.to_string());
        self.build()
    }

    /// Build the final Expr
    pub fn build(self) -> Expr {
        Expr::Case {
            when_clauses: self.when_clauses,
            else_value: self.else_value,
            alias: self.alias,
        }
    }
}

impl From<CaseBuilder> for Expr {
    fn from(builder: CaseBuilder) -> Self {
        builder.build()
    }
}