rbatis 4.8.3

The Rust SQL Toolkit and ORM Library. An async, pure Rust SQL crate featuring compile-time Dynamic SQL
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

#[cfg(test)]
mod test {
    use rbatis_macro_driver::rb_py;
    use rbs::value::map::ValueMap;
    use rbs::Value;

    #[rb_py(
        "
    SELECT * FROM biz_activity
    if  name != null:
      AND delete_flag = #{del}
      AND version = 1
      if  age!=1:
        AND version = 1
      AND version = 1
    AND a = 0
      yes
    and id in (
    trim ',': for item in ids:
      #{item},
    )
    and id in (
    trim ',': for index,item in ids:
      #{item},
    )
    trim 'AND':
      AND delete_flag = #{map.k}
    choose:
        when age==27:
          AND age = 27
        otherwise:
          AND age = 0
    trim end='where':
      ` where`
      if 1==2:
        `true`;
    "
    )]
    pub fn py_select_by_condition(arg: &rbs::Value, _tag: char) {}

    #[test]
    fn test_rbatis_codegen() {
        let mut arg = ValueMap::new();
        arg.insert("name".into(), "ss".into());
        arg.insert("del".into(), 1.into());
        arg.insert("del2".into(), 2.into());
        arg.insert(
            "ids".into(),
            vec![Value::I32(1), Value::I32(2), Value::I32(3)].into(),
        );
        arg.insert("map".into(), {
            let mut m = ValueMap::new();
            m.insert("k".into(), 1.into());
            Value::Map(m)
        });
        let (sql, args) = py_select_by_condition(rbs::Value::Map(arg), '$');
        println!("py->sql: {}", sql);
        println!("py->args: {}", serde_json::to_string(&args).unwrap());
    }

    #[rb_py(
        "insert into ${table_name} (
             trim ',':
               for k,v2 in table:
                 ${k},
             ) VALUES (
             trim ',':
               for k,v1 in table:
                 #{v1},
             )"
    )]
    pub fn save(arg: &rbs::Value, _tag: char) {}

    #[test]
    fn test_save() {
        let mut arg = ValueMap::new();
        arg.insert("table".into(), {
            let mut arg = ValueMap::new();
            arg.insert("table_name".into(), "a".into());
            arg.insert("table".into(), "a".into());
            Value::Map(arg)
        });
        let (sql, args) = save(rbs::Value::Map(arg), '$');
        println!("py->sql: {}", sql);
        println!("py->args: {}", serde_json::to_string(&args).unwrap());
    }

    #[test]
    fn test_two_eq() {
        #[rb_py("select * from test where  coalesce(user_id,#{data.user_id})=#{data.user_id}")]
        pub fn test_py_sql(arg: &rbs::Value, _tag: char) {}
        let mut data = ValueMap::new();
        data.insert(
            "data".into(),
            rbs::Value::Map({
                let mut m = ValueMap::new();
                m.insert("user_id".to_string().into(), 1.into());
                m
            }),
        );
        let res = test_py_sql(rbs::Value::Map(data), '?');
        println!("res={:?}", res.0);
        assert_eq!("select * from test where  coalesce(user_id,?)=?", res.0);
    }
}