ruroonga_expr 0.2.0

A tiny Groonga expression builder and generator.
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
108
109
110
111
112
113
114
115
116
117
118
119
120

//! Create grouping Groonga expression.
//!
//! Put parentheses into outer places:
//!
//! `expr1 expr2` =(grouping)=> `(expr1 expr2)`

use groupable::{Fragmentable, Groupable, Query};

#[derive(Debug, Clone, PartialEq, Eq)]
pub struct GroupBuilder {
    lhs: Query,
    rhs: Query,
}

impl GroupBuilder {
    pub fn new(lhs: Query, rhs: Query) -> GroupBuilder {
        GroupBuilder {
            lhs: lhs,
            rhs: rhs,
        }
    }

    pub fn build(self) -> String {
        format!("\'{} {}\'",
                self.lhs.into_iter().map(|e| e).collect::<String>(),
                self.rhs.into_iter().map(|e| e).collect::<String>())
    }
}

impl Fragmentable for GroupBuilder {
    fn to_fragment(self) -> Query {
        vec![format!("{} {}",
                     self.lhs.into_iter().map(|e| e).collect::<String>(),
                     self.rhs.into_iter().map(|e| e).collect::<String>())]
    }
}

impl Groupable for GroupBuilder {
    fn to_group(self) -> Query {
        vec![format!("({})",
                     self.to_fragment().into_iter().map(|e| e).collect::<String>())]
    }
}

// TODO: More std::ops::* definition is needed?

#[cfg(test)]
mod tests {
    use super::*;
    use phrase_expr::PhraseExpr;
    use greater_equal_expr::GreaterEqualExpr;
    use groupable::{Groupable, Fragmentable};
    use groupable::logical_or_builder::LogicalOrBuilder;
    use groupable::logical_and_builder::LogicalAndBuilder;
    use groupable::logical_not_builder::LogicalNotBuilder;

    #[test]
    fn test_build_with_logical_or() {
        let lexpr = PhraseExpr::new("Rust lang").column("language").prepare().unwrap().to_fragment();
        let rexpr = PhraseExpr::new("Haskell lang").column("language").prepare().unwrap().to_fragment();
        let logical_or_expr = LogicalOrBuilder::new(lexpr, rexpr).to_group();
        let comb_lexpr = GreaterEqualExpr::new("n_likes", "10").prepare().to_fragment();
        let result = GroupBuilder::new(comb_lexpr, logical_or_expr).build();
        let expected = concat!("\'n_likes:>=10 ",
                               "(language:@\"Rust lang\" OR language:@\"Haskell lang\")\'");
        assert_eq!(expected, result);
    }

    #[test]
    fn test_build_with_logical_and() {
        let lexpr = PhraseExpr::new("Rust lang").column("language").prepare().unwrap().to_fragment();
        let rexpr = PhraseExpr::new("Haskell lang").column("language").prepare().unwrap().to_fragment();
        let logical_or_expr = LogicalAndBuilder::new(lexpr, rexpr).to_group();
        let comb_lexpr = GreaterEqualExpr::new("n_likes", "10").prepare().to_fragment();
        let result = GroupBuilder::new(comb_lexpr, logical_or_expr).build();
        let expected = concat!("\'n_likes:>=10 ",
                               "(language:@\"Rust lang\" + language:@\"Haskell lang\")\'");
        assert_eq!(expected, result);
    }

    #[test]
    fn test_build_with_logical_not() {
        let lexpr = PhraseExpr::new("Rust lang").column("language").prepare().unwrap().to_fragment();
        let rexpr = PhraseExpr::new("Haskell lang").column("language").prepare().unwrap().to_fragment();
        let logical_or_expr = LogicalNotBuilder::new(lexpr, rexpr).to_group();
        let comb_lexpr = GreaterEqualExpr::new("n_likes", "10").prepare().to_fragment();
        let result = GroupBuilder::new(comb_lexpr, logical_or_expr).build();
        let expected = concat!("\'n_likes:>=10 ",
                               "(language:@\"Rust lang\" - language:@\"Haskell lang\")\'");
        assert_eq!(expected, result);
    }

    #[test]
    fn test_build_with_nested_logicals() {
        let llexpr = PhraseExpr::new("Rust lang").column("language").prepare().unwrap().to_fragment();
        let lrexpr = PhraseExpr::new("Haskell lang").column("language").prepare().unwrap().to_fragment();
        let llogical_or_expr = LogicalOrBuilder::new(llexpr, lrexpr).to_group();
        let lcomb_lexpr = GreaterEqualExpr::new("n_likes", "10").prepare().to_fragment();
        let llogical_expr = GroupBuilder::new(lcomb_lexpr, llogical_or_expr);
        assert_eq!(concat!("\'n_likes:>=10 ",
                           "(language:@\"Rust lang\" OR language:@\"Haskell lang\")\'"),
                   llogical_expr.clone().build());

        let rlexpr = PhraseExpr::new("Ruby lang").column("language").prepare().unwrap().to_fragment();
        let rrexpr = PhraseExpr::new("Python lang").column("language").prepare().unwrap().to_fragment();
        let rlogical_or_expr = LogicalOrBuilder::new(rlexpr, rrexpr).to_group();
        let rcomb_lexpr = GreaterEqualExpr::new("n_likes", "10").prepare().to_fragment();
        let rlogical_expr = GroupBuilder::new(rcomb_lexpr, rlogical_or_expr);
        assert_eq!(concat!("\'n_likes:>=10 ",
                           "(language:@\"Ruby lang\" OR language:@\"Python lang\")\'"),
                   rlogical_expr.clone().build());

        let result = LogicalAndBuilder::new(llogical_expr.to_group(), rlogical_expr.to_group()).build();
        let expected = concat!("\'(n_likes:>=10 ",
                               "(language:@\"Rust lang\" OR language:@\"Haskell lang\")) ",
                               "+ (n_likes:>=10 ",
                               "(language:@\"Ruby lang\" OR language:@\"Python lang\"))\'");
        assert_eq!(expected, result);
    }
}