mydb 0.0.1

This is a plugin that works with mydb_sqlx to make database operation easier
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 proc_macro2::{ Ident, Span, TokenStream };
use quote::quote;
use crate::derives::{
  TableField,
  utils::{ set_ident, type_value }
};

pub fn set_getters(table_list: Vec<TableField>) -> Vec<TokenStream> {
  Getters::new(table_list).set_getters()
}

#[derive(Debug, Clone, Default)]
pub struct Getters {
  pub data: Vec<TableField>,
  pub ts_getters: Vec<TokenStream>,
}

impl Getters {
  pub fn new(data: Vec<TableField>) -> Self {
    Getters {
      data,
      ts_getters: Vec::new(),
    }
  }

  pub fn set_getters(&mut self) -> Vec<TokenStream> {
    for field in &self.data {
      if field.exist {
        let get_name = set_ident(&format!("get_{}", &field.name));
        let name = set_ident(&field.name);
        let reture_value = type_value(field._option, &field.field_type);
        if &field.field_type == "DateTime" {
          self.ts_getters.push(
            quote! {
              pub fn #get_name(&self) -> #reture_value {
                self.#name
              }
            }
          );
        } else {
          self.ts_getters.push(
            quote! {
              pub fn #get_name(&self) -> #reture_value {
                self.#name.clone()
              }
            }
          );
        }
      }
    }
    self.ts_getters.clone()
  }
}