co_orm_macros/
lib.rs

1/*
2 * @Author: plucky
3 * @Date: 2022-09-04 00:01:24
4 * @LastEditTime: 2024-08-03 12:47:01
5 */
6
7extern crate proc_macro;
8
9use helper::check_attributes;
10use impl_by_field::generate_impl_from_row;
11use proc_macro::TokenStream;
12use impl_crud::generate_crud;
13use syn::{parse_macro_input, DeriveInput};
14
15
16mod util;
17mod impl_by_field;
18mod db_type;
19mod impl_crud;
20mod helper;
21
22
23
24/// `#[derive(Crud)]`
25/// generate method: get, get_by, query, query_by, update, delete, insert, insert_all, query_page_by
26/// 
27/// attributes:
28/// 
29/// `#[co_orm(id)]`
30/// default first field is primary key or set.
31/// 
32/// `#[co_orm(seq)]`
33/// sequence field, auto increment. insert will skip this field.
34/// 
35/// `#[co_orm(rename="name")]`
36/// rename table name or field name. 
37/// default table name by struct name to_table_case: UserDetail => user_detail. 
38/// default field name by field name to_snake_case: UserDetail => user_detail. 
39/// 
40/// `#[co_orm(skip)]`
41/// ignore field. using sqlx::FromRow, skip need `#[co_orm(skip)]` and `#[sqlx(skip)]`
42/// 
43/// `#[co_orm(update)]`
44/// generate method update_xxx. 
45/// 
46/// `#[co_orm(by)]`
47/// generate qet_by_field, query_by_field, update_by_field, delete_by_field.
48/// 
49/// `#[co_orm(skip_insert)]`
50/// insert will skip this field.
51#[proc_macro_derive(Crud, 
52    attributes(
53        co_orm, // co_orm(id), co_orm(seq), co_orm(rename="name"), co_orm(skip), co_orm(update), co_orm(by),
54        
55    )
56)]
57pub fn sql_derive_crud(input: TokenStream) -> TokenStream{
58    let input = parse_macro_input!(input as DeriveInput);
59    
60    if let Err(e) =  check_attributes(&input.attrs){
61       return e.to_compile_error().into();
62    }
63
64    generate_crud(input)
65    
66}
67
68/// `#[derive(FromRow)]`
69/// generate impl sqlx::FromRow for struct.
70/// 
71/// or use `#[derive(sqlx::FromRow)]`.
72/// 
73#[proc_macro_derive(FromRow)]
74pub fn sql_derive_form_row(input: TokenStream) -> TokenStream{
75    let input = parse_macro_input!(input as DeriveInput);
76
77    generate_impl_from_row(&input).into()
78
79}
80
co_orm_macros/lib.rs

co_orm_macros/
lib.rs
