chronon-backend-sql-common 0.1.0

Shared SQL SchedulerStore for SQLite and PostgreSQL
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

//! Internal SQL query helpers for dialect-aware pool dispatch.
//!
//! **Audience:** internal — used by [`SqlSchedulerStore`](crate::SqlSchedulerStore) modules.

/// Execute a parameterized SQL statement on the store pool.
#[macro_export]
macro_rules! sql_execute {
    ($store:expr, $sql:expr, |$q:ident| $body:expr) => {{
        match &$store.pool {
            $crate::SqlPool::Sqlite(pool) => {
                let $q = sqlx::query($sql);
                let $q = $body;
                $q.execute(pool).await.map_err(|e| $crate::error_map::map_err(&e))?;
            }
            $crate::SqlPool::Postgres(pool) => {
                let $q = sqlx::query($sql);
                let $q = $body;
                $q.execute(pool).await.map_err(|e| $crate::error_map::map_err(&e))?;
            }
        }
        Ok::<(), chronon_core::error::ChrononError>(())
    }};
}

/// Fetch zero or one row and map it through a closure.
#[macro_export]
macro_rules! sql_fetch_optional_map {
    ($store:expr, $sql:expr, |$q:ident| $bind:expr, |$row:ident| $map:expr) => {{
        match &$store.pool {
            $crate::SqlPool::Sqlite(pool) => {
                let $q = sqlx::query($sql);
                let $q = $bind;
                match $q.fetch_optional(pool).await.map_err(|e| $crate::error_map::map_err(&e))? {
                    Some($row) => Ok(Some($map?)),
                    None => Ok(None),
                }
            }
            $crate::SqlPool::Postgres(pool) => {
                let $q = sqlx::query($sql);
                let $q = $bind;
                match $q.fetch_optional(pool).await.map_err(|e| $crate::error_map::map_err(&e))? {
                    Some($row) => Ok(Some($map?)),
                    None => Ok(None),
                }
            }
        }
    }};
}

/// Fetch exactly one row and map it through a closure.
#[macro_export]
macro_rules! sql_fetch_one_map {
    ($store:expr, $sql:expr, |$q:ident| $bind:expr, |$row:ident| $map:expr) => {{
        match &$store.pool {
            $crate::SqlPool::Sqlite(pool) => {
                let $q = sqlx::query($sql);
                let $q = $bind;
                let $row = $q.fetch_one(pool).await.map_err(|e| $crate::error_map::map_err(&e))?;
                $map
            }
            $crate::SqlPool::Postgres(pool) => {
                let $q = sqlx::query($sql);
                let $q = $bind;
                let $row = $q.fetch_one(pool).await.map_err(|e| $crate::error_map::map_err(&e))?;
                $map
            }
        }
    }};
}

/// Fetch all rows and map each through a closure.
#[macro_export]
macro_rules! sql_fetch_all_map {
    ($store:expr, $sql:expr, |$q:ident| $bind:expr, |$row:ident| $map:expr) => {{
        match &$store.pool {
            $crate::SqlPool::Sqlite(pool) => {
                let $q = sqlx::query($sql);
                let $q = $bind;
                let rows = $q.fetch_all(pool).await.map_err(|e| $crate::error_map::map_err(&e))?;
                rows.iter().map(|$row| $map).collect::<chronon_core::Result<Vec<_>>>()
            }
            $crate::SqlPool::Postgres(pool) => {
                let $q = sqlx::query($sql);
                let $q = $bind;
                let rows = $q.fetch_all(pool).await.map_err(|e| $crate::error_map::map_err(&e))?;
                rows.iter().map(|$row| $map).collect::<chronon_core::Result<Vec<_>>>()
            }
        }
    }};
}