worktable_macros 0.9.0

Proc-macro companion crate for worktable: the worktable! macro and its derives. Formerly published as worktable_codegen.
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

use crate::common::name_generator::WorktableNameGenerator;
use crate::persist_table::generator::Generator;
use proc_macro2::TokenStream;
use quote::quote;

impl Generator {
    pub fn get_persistence_task_type(&self) -> TokenStream {
        if self.attributes.read_only {
            return quote! {};
        }
        let name_generator = WorktableNameGenerator::from_struct_ident(&self.struct_def.ident);
        let ident = name_generator.get_persistence_task_ident();
        let primary_key_type = name_generator.get_primary_key_type_ident();
        let space_secondary_indexes_events = name_generator.get_space_secondary_index_events_ident();
        let avt_index_ident = name_generator.get_available_indexes_ident();

        quote! {
            pub type #ident = PersistenceTask<
                <<#primary_key_type as TablePrimaryKey>::Generator as PrimaryKeyGeneratorState>::State,
                #primary_key_type,
                #space_secondary_indexes_events,
                #avt_index_ident,
            >;
        }
    }

    pub fn get_persistence_engine_type(&self) -> TokenStream {
        if self.attributes.read_only {
            return quote! {};
        }
        let name_generator = WorktableNameGenerator::from_struct_ident(&self.struct_def.ident);
        let ident = name_generator.get_persistence_engine_ident();
        let primary_key_type = name_generator.get_primary_key_type_ident();
        let inner_const_name = name_generator.get_page_inner_size_const_ident();
        let const_name = name_generator.get_page_size_const_ident();
        let space_secondary_indexes = name_generator.get_space_secondary_index_ident();
        let space_secondary_indexes_events = name_generator.get_space_secondary_index_events_ident();
        let avt_index_ident = name_generator.get_available_indexes_ident();
        let space_index_type = if self.attributes.pk_unsized {
            quote! {
                SpaceIndexUnsized<#primary_key_type, { #inner_const_name as u32 }>,
            }
        } else {
            quote! {
                SpaceIndex<#primary_key_type, { #inner_const_name as u32 }>,
            }
        };

        quote! {
            pub type #ident = DiskPersistenceEngine<
                SpaceData<
                    <<#primary_key_type as TablePrimaryKey>::Generator as PrimaryKeyGeneratorState>::State,
                    { #inner_const_name},
                    { #const_name as u32 }
                >,
                #space_index_type
                #space_secondary_indexes,
                #primary_key_type,
                #space_secondary_indexes_events,
                #avt_index_ident,
            >;
        }
    }
}