taitan-orm-parser 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

use crate::FieldDef;
use proc_macro2::TokenStream;
use quote::{format_ident, quote};

#[derive(Clone, Debug, Default)]
pub struct ArgsMapper;

impl ArgsMapper {
    pub fn map_add_to_args(&self, field: &FieldDef) -> TokenStream {
        let field_name = &field.struct_field.get_name();
        let field_ident = format_ident!("{}", field_name);
        if field.struct_field.is_option() {
            if field.struct_field.is_location_expr {
                quote! {
                    if let Some(f) =  &self.#field_ident {
                        sqlx::Arguments::add(args, &f.val)?
                    }
                }
            } else {
                quote! {
                    if let Some(f) =  &self.#field_ident {
                        sqlx::Arguments::add(args, f)?
                    }
                }
            }
        } else {
            if field.struct_field.is_location_expr {
                quote! {
                    sqlx::Arguments::add(args, &self.#field_ident.val)?
                }
            } else {
                quote! {
                    sqlx::Arguments::add(args, &self.#field_ident)?
                }
            }
        }
    }

    pub fn map_enum_add_to_args(&self, field: &FieldDef) -> TokenStream {
        let field_name = &field.struct_field.get_name();
        let field_ident = format_ident!("{}", field_name);
        if field.struct_field.is_location_expr {
            quote! {
                sqlx::Arguments::add(args, &#field_ident.val)?
            }
        } else {
            quote! {
                sqlx::Arguments::add(args, #field_ident)?
            }
        }

    }
}