scylla_orm_table_to_struct 0.1.1

Table mapper for the ORM for Scylla, fully compatible with Apache Cassandra™
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117

use heck::CamelCase;
use proc_macro2::{Ident, TokenStream};
use quote::format_ident;

pub fn primary_key_struct() -> Ident {
    format_ident!("PrimaryKey")
}

pub fn primary_key_owned() -> Ident {
    format_ident!("primary_key_owned")
}

pub fn primary_key_struct_ref() -> Ident {
    struct_ref(&primary_key_struct())
}

pub fn struct_ref(struct_ref: &Ident) -> Ident {
    format_ident!("{}Ref", struct_ref)
}

pub fn to_ref() -> TokenStream {
    "to_ref".parse().unwrap()
}

pub fn primary_key_struct_parameter() -> Ident {
    format_ident!("primary_key")
}

pub fn updatable_column() -> Ident {
    format_ident!("UpdatableColumn")
}

pub fn get_updatable_column_field(field: &Ident) -> Ident {
    format_ident!("updatable_column_{}", field)
}

pub fn updatable_column_ref() -> Ident {
    struct_ref(&updatable_column())
}

pub fn create_variant(ident: &Ident) -> Ident {
    format_ident!("{}", ident.to_string().to_camel_case())
}

pub fn update_dyn() -> Ident {
    format_ident!("update_dyn")
}

pub fn update_dyn_multiple() -> Ident {
    format_ident!("update_dyn_multiple")
}

pub fn in_memory_update() -> Ident {
    format_ident!("in_memory_update")
}

pub fn in_memory_updates() -> Ident {
    format_ident!("in_memory_updates")
}

macro_rules! write_query {
    ($fn_name: ident, $content: expr) => {
        pub fn $fn_name() -> Ident {
            format_ident!($content)
        }
    };
    ($fn_name: ident, $content: expr, $fn_name_constant: ident) => {
        write_query!($fn_name, $content);

        pub fn $fn_name_constant() -> Ident {
            format_ident!("{}_QUERY", $content.to_uppercase())
        }
    };
}

write_query!(insert_or_delete_fn_name, "insert_or_delete");
write_query!(insert_fn_name, "insert", insert_constant);
write_query!(insert_ttl_fn_name, "insert_ttl", insert_ttl_constant);
write_query!(truncate_fn_name, "truncate", truncate_constant);
write_query!(
    select_unique_fn_name,
    "select_unique",
    select_unique_constant
);
write_query!(select_unique_expect_fn_name, "select_unique_expect");
write_query!(select_all_fn_name, "select_all", select_all_constant);
write_query!(
    select_all_count_fn_name,
    "select_all_count",
    select_all_count_constant
);
write_query!(delete_fn_name, "delete", delete_constant);

pub fn base_table(ident: &Ident) -> Ident {
    format_ident!("{}_base_table", ident)
}
pub fn base_table_query(ident: &Ident) -> Ident {
    format_ident!("{}_BASE_TABLE", ident)
}

pub fn qv(ident: &Ident) -> Ident {
    format_ident!("{}_qv", ident)
}

pub fn all_in_memory(ident: &Ident) -> Ident {
    format_ident!("{}_in_memory", ident)
}

pub fn update_field(ident: &Ident) -> (Ident, Ident) {
    let update_string = format!("update_{}", ident);
    let constant = update_string.to_uppercase() + "_QUERY";

    (
        format_ident!("{}", update_string),
        format_ident!("{}", constant),
    )
}