alopex-sql 0.6.0

SQL parser components for the Alopex DB dialect
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91

use crate::planner::typed_expr::TypedExpr;
use crate::planner::types::ResolvedType;

/// Supported aggregate function types.
#[derive(Debug, Clone, PartialEq)]
pub enum AggregateFunction {
    Count,
    Sum,
    Total,
    Avg,
    Min,
    Max,
    GroupConcat { separator: Option<String> },
    StringAgg { separator: Option<String> },
}

/// Aggregate expression definition.
#[derive(Debug, Clone)]
pub struct AggregateExpr {
    pub function: AggregateFunction,
    pub arg: Option<TypedExpr>,
    pub distinct: bool,
    pub result_type: ResolvedType,
}

impl AggregateExpr {
    pub fn count_star() -> Self {
        Self {
            function: AggregateFunction::Count,
            arg: None,
            distinct: false,
            result_type: ResolvedType::BigInt,
        }
    }

    pub fn count(arg: TypedExpr, distinct: bool) -> Self {
        Self {
            function: AggregateFunction::Count,
            arg: Some(arg),
            distinct,
            result_type: ResolvedType::BigInt,
        }
    }

    pub fn sum(arg: TypedExpr) -> Self {
        Self {
            function: AggregateFunction::Sum,
            arg: Some(arg),
            distinct: false,
            result_type: ResolvedType::Double,
        }
    }

    pub fn total(arg: TypedExpr) -> Self {
        Self {
            function: AggregateFunction::Total,
            arg: Some(arg),
            distinct: false,
            result_type: ResolvedType::Double,
        }
    }

    pub fn avg(arg: TypedExpr) -> Self {
        Self {
            function: AggregateFunction::Avg,
            arg: Some(arg),
            distinct: false,
            result_type: ResolvedType::Double,
        }
    }

    pub fn min(arg: TypedExpr) -> Self {
        let result_type = arg.resolved_type.clone();
        Self {
            function: AggregateFunction::Min,
            arg: Some(arg),
            distinct: false,
            result_type,
        }
    }

    pub fn max(arg: TypedExpr) -> Self {
        let result_type = arg.resolved_type.clone();
        Self {
            function: AggregateFunction::Max,
            arg: Some(arg),
            distinct: false,
            result_type,
        }
    }
}