co-orm-macros 0.3.15

Derive macro for co-orm
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

/*
 * @Author: plucky
 * @Date: 2022-10-20 00:29:48
 *
 */

#[cfg(not(any(feature = "postgres", feature = "sqlite", feature = "mssql")))]
pub(crate) mod db {
    use quote::{__private::TokenStream, quote};

    pub fn db_pool_token() -> (TokenStream, TokenStream, TokenStream) {
        let pool = quote!(sqlx::MySqlPool);
        let query_result = quote!(sqlx::mysql::MySqlQueryResult);
        let arguments = quote!(sqlx::mysql::MySqlArguments);
        (pool, query_result, arguments)
    }

    pub fn db_row_token() -> TokenStream {
        quote!(sqlx::mysql::MySqlRow)
    }

    pub fn db_placeholder(_index: usize) -> String {
        "?".into()
    }
}

#[cfg(feature = "postgres")]
pub(crate) mod db {
    use quote::{__private::TokenStream, quote};

    pub fn db_pool_token() -> (TokenStream, TokenStream, TokenStream) {
        let pool = quote!(sqlx::PgPool);
        let query_result = quote!(sqlx::postgres::PgQueryResult);
        let arguments = quote!(sqlx::postgres::PgArguments);
        (pool, query_result, arguments)
    }

    pub fn db_row_token() -> TokenStream {
        quote!(sqlx::postgres::PgRow)
    }

    pub fn db_placeholder(index: usize) -> String {
        format!("${}", index)
    }
}

#[cfg(feature = "sqlite")]
pub(crate) mod db {
    use quote::{__private::TokenStream, quote};

    pub fn db_pool_token() -> (TokenStream, TokenStream, TokenStream) {
        let pool = quote!(sqlx::SqlitePool);
        let query_result = quote!(sqlx::sqlite::SqliteQueryResult);
        let arguments = quote!(sqlx::sqlite::SqliteArguments);
        (pool, query_result, arguments)
    }

    pub fn db_row_token() -> TokenStream {
        quote!(sqlx::sqlite::SqliteRow)
    }

    pub fn db_placeholder(index: usize) -> String {
        format!("?{}", index)
    }
}

#[cfg(feature = "mssql")]
pub(crate) mod db {
    use quote::{__private::TokenStream, quote};

    pub fn db_pool_token() -> (TokenStream, TokenStream, TokenStream) {
        let pool = quote!(sqlx::Pool<sqlx::Mssql>);
        let query_result = quote!(sqlx::mssql::MssqlQueryResult);
        let arguments = quote!(sqlx::mssql::MssqlArguments);
        (pool, query_result, arguments)
    }

    pub fn db_row_token() -> TokenStream {
        quote!(sqlx::mssql::MssqlRow)
    }

    pub fn db_placeholder(_index: usize) -> String {
        "?".into()
    }
}

// #[cfg(feature = "any")]
// pub(crate) mod db {
//     use quote::{__private::TokenStream, quote};

//     pub fn db_pool_token() -> (TokenStream, TokenStream, TokenStream) {
//         let pool = quote!(sqlx::Pool<sqlx::Any>);
//         let query_result = quote!(sqlx::any::AnyQueryResult);
//         let arguments = quote!(sqlx::any::AnyArguments<'static>);
//         (pool, query_result, arguments)
//     }

//     pub fn db_row_token() -> TokenStream {
//         quote!(sqlx::any::AnyRow)
//     }

//     pub fn db_placeholder(_index: usize) -> String {
//         "?".into()
//     }
// }