ragit-server 0.4.2

poc server implementation of ragit
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

/// It's a very thin wrapper around `sqlx::query!`. If `log_sql` feature is set,
/// this macro calls `write_log((query, args))` after calling `sqlx::query!`.
/// `log_sql` is very expensive. Some tables store blobs where each blob is 100KiB ~ 2MiB.
/// Rust will try to dump the blob in `Vec<u8>` format, which is a few MiB per blob.
#[macro_export]
#[cfg(feature = "log_sql")]
macro_rules! query {
    ($query: expr, $($args:tt)*) => {{
        // `write_log` consumes the args but `sqlx::query!` doesn't. So we have to call
        // `write_log` after `sqlx::query!`.
        let r = sqlx::query!($query, $($args)*);
        ragit_fs::write_log(
            "sql",
            // TODO: how about using `trim_long_string` here?
            &format!("query: {:?}, args: {:?}", $query, ($($args)*)),
        );
        r
    }};
    ($query: expr) => {{
        let r = sqlx::query!($query);
        ragit_fs::write_log(
            "sql",
            &format!("query: {:?}, args: ()", $query),
        );
        r
    }};
}

#[macro_export]
#[cfg(feature = "log_sql")]
macro_rules! query_as {
    ($out_struct: path, $query: expr, $($args:tt)*) => {{
        let r = sqlx::query_as!($out_struct, $query, $($args)*);
        ragit_fs::write_log(
            "sql",
            // TODO: how about using `trim_long_string` here?
            &format!("out_struct: {}, query: {:?}, args: {:?}", stringify!($out_struct), $query, ($($args)*)),
        );
        r
    }};
    ($out_struct: path, $query: expr) => {{
        let r = sqlx::query_as!($out_struct, $query);
        ragit_fs::write_log(
            "sql",
            // TODO: how about using `trim_long_string` here?
            &format!("out_struct: {}, query: {:?}, args: ()", stringify!($out_struct), $query),
        );
        r
    }};
}

#[macro_export]
#[cfg(not(feature = "log_sql"))]
macro_rules! query {
    ($($args:tt)*) => { sqlx::query!($($args)*) };
}

#[macro_export]
#[cfg(not(feature = "log_sql"))]
macro_rules! query_as {
    ($($args:tt)*) => { sqlx::query_as!($($args)*) };
}