taitan-orm-trait 0.1.12

Next Generation ORM based on sqlx
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

use sqlx::{Arguments, Database, Postgres};
use std::borrow::Cow;
use sqlx::postgres::{PgArguments};
use taitan_orm_trait::traits::Mutation;
use taitan_orm_trait::result::Result;
use time::macros::datetime;
use time::PrimitiveDateTime;

use taitan_orm_trait::traits::Parameter;

#[derive(Debug)]
struct UserMutation {
    name: Option<String>,
    created: Option<PrimitiveDateTime>,
}

impl Parameter<Postgres> for UserMutation {
    fn add_to_args<'a, 'b>(&'a self, args: &'b mut <Postgres as Database>::Arguments<'a>) -> Result<()> {
        if let Some(name) =  &self.name {
            args.add(name)?;
        }
        if let Some(created) = &self.created {
            args.add(created)?;
        }
        Ok(())
    }
}

impl Mutation<Postgres> for UserMutation {
    fn gen_update_set_sql<'a>(&self) -> Cow<'a, str> {
        let mut sql = String::from("UPDATE `user`");
        if self.name.is_some() {
            sql.push_str(" name = ?");
        }
        if self.created.is_some() {
            sql.push_str(" created = ?");
        }
        Cow::from(sql)
    }

    // fn add_update_set_args<'a>(&'a self, args: &mut <Postgres as Database>::Arguments<'a>) -> Result<()> {
    //     if let Optional::Some(name) =  &self.name {
    //         wrap_encode(args.add(name))?;
    //     }
    //     if let Optional::Some(created) = &self.created {
    //         wrap_encode(args.add(created))?;
    //     }
    //     Ok(())
    // }

    fn all_none(&self) -> bool {
        self.name.is_none() && self.created.is_none()
    }
}

#[test]
fn mutation_trait_spec() {
    let user = UserMutation {
        name: Some("Allen".to_string()),
        created: Some(datetime!(2019-01-01 0:00)),
    };
    let update_set_sql = user.gen_update_set_sql();
    let mut args = PgArguments::default();
    let update_set_args = user.add_to_args(&mut args).unwrap();
}