dbstruct-derive 0.1.0-alpha

Build a typed database by defining a struct
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

use syn::parse_quote;

use crate::model::Model;

use super::struct_def::Struct;

pub struct NewMethod {
    pub fields: Vec<syn::FieldValue>,
    pub vis: syn::Visibility,
}

fn as_len_value(ident: syn::Ident) -> syn::FieldValue {
    let colon: syn::token::Colon = syn::Token![:](proc_macro2::Span::call_site());
    syn::FieldValue {
        attrs: Vec::new(),
        member: syn::Member::Named(ident),
        colon_token: Some(colon),
        expr: parse_quote!(std::sync::Arc::new(std::sync::atomic::AtomicUsize::new(0))),
    }
}

impl NewMethod {
    pub fn from(model: &Model, struct_def: &Struct) -> Self {
        let fields: Vec<_> = struct_def
            .extra_vars
            .iter()
            .map(|def| def.ident.clone())
            .map(Option::unwrap)
            .map(as_len_value)
            .collect();
        Self {
            fields,
            vis: model.vis.clone(),
        }
    }
}