gluesql-core 0.20.0

GlueSQL - Open source SQL database engine fully written in Rust with pure functional execution layer, easily swappable storage and web assembly support!
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
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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107

use {
    super::ExprNode,
    crate::{
        ast::OrderByExpr,
        parse_sql::parse_order_by_expr,
        plan::OrderByExprPlan,
        result::Result,
        translate::{NO_PARAMS, translate_order_by_expr},
    },
};

#[derive(Clone, Debug)]
pub enum OrderByExprNode<'a> {
    Text(String),
    Expr {
        expr: ExprNode<'a>,
        asc: Option<bool>,
    },
}

impl From<&str> for OrderByExprNode<'_> {
    fn from(expr: &str) -> Self {
        Self::Text(expr.to_owned())
    }
}

impl<'a> From<ExprNode<'a>> for OrderByExprNode<'a> {
    fn from(expr_node: ExprNode<'a>) -> Self {
        Self::Expr {
            expr: expr_node,
            asc: None,
        }
    }
}

impl OrderByExprNode<'_> {
    pub(super) fn build_order_by_expr(self) -> Result<OrderByExpr> {
        match self {
            OrderByExprNode::Text(expr) => {
                parse_order_by_expr(expr).and_then(|op| translate_order_by_expr(&op, NO_PARAMS))
            }
            OrderByExprNode::Expr { expr, asc } => {
                let expr = expr.build_expr()?;

                Ok(OrderByExpr { expr, asc })
            }
        }
    }

    pub(super) fn build_order_by_expr_plan(self) -> Result<OrderByExprPlan> {
        match self {
            OrderByExprNode::Text(expr) => parse_order_by_expr(expr)
                .and_then(|op| translate_order_by_expr(&op, NO_PARAMS).map(Into::into)),
            OrderByExprNode::Expr { expr, asc } => {
                let expr = expr.build_expr_plan()?;

                Ok(OrderByExprPlan { expr, asc })
            }
        }
    }
}

#[cfg(test)]
mod tests {
    use {
        crate::{
            parse_sql::parse_order_by_expr,
            plan::OrderByExprPlan,
            query_builder::{OrderByExprNode, col},
            translate::{NO_PARAMS, translate_order_by_expr},
        },
        pretty_assertions::assert_eq,
    };

    fn test(actual: OrderByExprNode, expected: &str) {
        let parsed = &parse_order_by_expr(expected).expect(expected);
        let expected = translate_order_by_expr(parsed, NO_PARAMS).map(OrderByExprPlan::from);
        assert_eq!(actual.build_order_by_expr_plan(), expected);
    }

    #[test]
    fn order_by_expr() {
        let actual = OrderByExprNode::Text("foo".into());
        let expected = "foo";
        test(actual, expected);

        let actual = OrderByExprNode::Text("foo asc".into());
        let expected = "foo ASC";
        test(actual, expected);

        let actual = OrderByExprNode::Text("foo desc".into());
        let expected = "foo DESC";
        test(actual, expected);

        let actual = OrderByExprNode::from(col("foo"));
        let expected = "foo";
        test(actual, expected);

        let actual = col("foo").asc();
        let expected = "foo ASC";
        test(actual, expected);

        let actual = col("foo").desc();
        let expected = "foo DESC";
        test(actual, expected);
    }
}