diesel_derives/
identifiable.rs

Help
1use proc_macro2::TokenStream;
2use quote::quote;
3use syn::parse_quote;
4use syn::DeriveInput;
5use syn::Result;
6
7use crate::attrs::AttributeSpanWrapper;
8use crate::field::Field;
9use crate::model::Model;
10use crate::util::wrap_in_dummy_mod;
11
12pub fn derive(item: DeriveInput) -> Result<TokenStream> {
13    let model = Model::from_item(&item, false, false)?;
14
15    let struct_name = &item.ident;
16    let table_name = &model.table_names()[0];
17
18    let (impl_generics, ty_generics, where_clause) = item.generics.split_for_impl();
19    let mut ref_generics = item.generics.clone();
20    ref_generics.params.push(parse_quote!('ident));
21    let (ref_generics, ..) = ref_generics.split_for_impl();
22
23    let mut field_ty = Vec::new();
24    let mut field_name = Vec::new();
25    for pk in model.primary_key_names.iter() {
26        let Field {
27            ty,
28            name,
29            serialize_as,
30            ..
31        } = &model.find_column(pk)?;
32        if let Some(AttributeSpanWrapper { item: ty, .. }) = serialize_as.as_ref() {
33            field_ty.push(quote!(#ty));
34            field_name.push(quote!(::std::convert::Into::<#ty>::into(self.#name.clone())));
35        } else {
36            field_ty.push(quote!(&'ident #ty));
37            field_name.push(quote!(&self.#name));
38        }
39    }
40
41    Ok(wrap_in_dummy_mod(quote! {
42        impl #impl_generics diesel::associations::HasTable for #struct_name #ty_generics
43        #where_clause
44        {
45            type Table = #table_name::table;
46
47            fn table() -> <Self as diesel::associations::HasTable>::Table {
48                #table_name::table
49            }
50        }
51
52        impl #ref_generics diesel::associations::Identifiable for &'ident #struct_name #ty_generics
53        #where_clause
54        {
55            type Id = (#(#field_ty),*);
56
57            fn id(self) -> <Self as diesel::associations::Identifiable>::Id {
58                (#(#field_name),*)
59            }
60        }
61
62        impl #ref_generics diesel::associations::Identifiable for &'_ &'ident #struct_name #ty_generics
63            #where_clause
64        {
65            type Id = (#(#field_ty),*);
66
67            fn id(self) -> <Self as diesel::associations::Identifiable>::Id {
68                (#(#field_name),*)
69            }
70        }
71    }))
72}
diesel_derives/identifiable.rs

diesel_derives/
identifiable.rs
