db_code_macro 0.1.2

Generate Db code
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

use proc_macro::TokenStream;

use quote::quote;
use syn::{DeriveInput, parse_macro_input};

mod dao;
mod db_meta;
mod kits;

const CARGO_BUILD_DIR_SQL: &str = "CARGO_BUILD_DIR_SQL_";

#[proc_macro_derive(DbDao, attributes(db_sub))]
pub fn db(input: TokenStream) -> TokenStream {
    let ast = parse_macro_input!(input as DeriveInput);
    let mut meta = db_meta::DbMeta::get()
        .lock()
        .expect("db_meta::DbMeta::get().lock()");
    (*meta).push(&ast);
    TokenStream::from(quote! {})
}

#[proc_macro_derive(DbSub)]
pub fn db_sub(input: TokenStream) -> TokenStream {
    let ast = parse_macro_input!(input as DeriveInput);
    let mut meta = db_meta::DbMeta::get()
        .lock()
        .expect("db_meta::DbMeta::get().lock()");
    (*meta).push_sub_struct(&ast);
    TokenStream::from(quote! {})
}