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

use proc_macro2::{ Ident, Span, TokenStream };
use quote::quote;
use crate::derives::{
  TableField,
  utils::{ set_ident, type_value }
};

pub fn set_setters(table_list: Vec<TableField>) -> Vec<TokenStream> {
  Setters::new(table_list).set_setters()
}
#[derive(Debug, Clone, Default)]
pub struct Setters {
  pub data: Vec<TableField>,
  pub ts_setters: Vec<TokenStream>,
}

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

  pub fn set_setters(&mut self) -> Vec<TokenStream> {
    for field in &self.data {
      if field.exist {
        let set_name = set_ident(&format!("set_{}", &field.name));
        let name = set_ident(&field.name);
        let param_type = type_value(field._option, &field.field_type);
        self.ts_setters.push(
          quote! {
            pub fn #set_name(&mut self, param: #param_type) -> &mut Self {
              self.#name = param;
              self
            }
          }
        );
      }
    }
    self.ts_setters.clone()
  }
}