diesel_derives/
identifiable.rs

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

diesel_derives/
identifiable.rs
