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

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

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

impl RowMapper {
    pub fn map_row_try_get(&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() {
            quote! {
                if let Some(_) = &selection.#field_ident {
                    selected.#field_ident = Some(sqlx::Row::try_get(&row, i)?);
                    i += 1;
                }
            }
        } else {
            quote! {
                selected.#field_ident = Some(sqlx::Row::try_get(&row, i)?);
                i += 1;
            }
        }
    }
    pub fn map_selected_default(&self, field: &FieldDef) -> TokenStream {
        let field_name = &field.struct_field.get_name();
        let field_ident = format_ident!("{}", field_name);
        quote! {
            #field_ident: Some(None),
        }
    }
}