dinoco_engine 2.0.5

Database adapters, query execution, and migration engine components for Dinoco.
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

use chrono::{DateTime, NaiveDate, NaiveDateTime, Utc};

use crate::{DeadpoolPostgresRow, DinocoMysql, DinocoPostgres, DinocoSqlite, MysqlRow, PostgresRow, SqliteRow};

/// Wraps a single scalar column read from position `0` of a result row.
///
/// Backs both [`crate::DinocoAdapter::query`]-style scalar projections such as
/// `.pluck(...)` and single-value aggregates such as `exists(...)`, so the
/// same [`crate::DinocoRowModel`] machinery used for entities can decode a
/// bare `SELECT <field> ...` / `SELECT EXISTS(...) ...` result.
#[derive(Debug, Clone, PartialEq)]
pub struct PluckValue<T>(pub T);

macro_rules! impl_pluck_scalar {
    ($($ty:ty),* $(,)?) => {
        $(
            impl DinocoSqlite for PluckValue<$ty> {
                fn from_sqlite_row(row: &SqliteRow<'_>) -> Option<Self> {
                    row.get::<_, $ty>(0).ok().map(PluckValue)
                }
            }

            impl DinocoSqlite for PluckValue<Option<$ty>> {
                fn from_sqlite_row(row: &SqliteRow<'_>) -> Option<Self> {
                    row.get::<_, Option<$ty>>(0).ok().map(PluckValue)
                }
            }

            impl DinocoPostgres for PluckValue<$ty> {
                fn from_deadpool_posgres_row(row: &DeadpoolPostgresRow) -> Option<Self> {
                    row.try_get::<_, $ty>(0).ok().map(PluckValue)
                }

                fn from_postgres_row(row: &PostgresRow) -> Option<Self> {
                    Self::from_deadpool_posgres_row(row)
                }
            }

            impl DinocoPostgres for PluckValue<Option<$ty>> {
                fn from_deadpool_posgres_row(row: &DeadpoolPostgresRow) -> Option<Self> {
                    row.try_get::<_, Option<$ty>>(0).ok().map(PluckValue)
                }

                fn from_postgres_row(row: &PostgresRow) -> Option<Self> {
                    Self::from_deadpool_posgres_row(row)
                }
            }

            impl DinocoMysql for PluckValue<$ty> {
                fn from_mysql_row(row: &MysqlRow) -> Option<Self> {
                    let mut row = row.clone();
                    row.take::<$ty, _>(0).map(PluckValue)
                }
            }

            impl DinocoMysql for PluckValue<Option<$ty>> {
                fn from_mysql_row(row: &MysqlRow) -> Option<Self> {
                    let mut row = row.clone();
                    row.take::<Option<$ty>, _>(0).map(PluckValue)
                }
            }
        )*
    };
}

impl_pluck_scalar!(String, bool, i64, f64, Vec<u8>, NaiveDate, crate::serde_json::Value);

impl DinocoSqlite for PluckValue<DateTime<Utc>> {
    fn from_sqlite_row(row: &SqliteRow<'_>) -> Option<Self> {
        row.get::<_, DateTime<Utc>>(0).ok().map(PluckValue)
    }
}

impl DinocoSqlite for PluckValue<Option<DateTime<Utc>>> {
    fn from_sqlite_row(row: &SqliteRow<'_>) -> Option<Self> {
        row.get::<_, Option<DateTime<Utc>>>(0).ok().map(PluckValue)
    }
}

impl DinocoPostgres for PluckValue<DateTime<Utc>> {
    fn from_deadpool_posgres_row(row: &DeadpoolPostgresRow) -> Option<Self> {
        row.try_get::<_, DateTime<Utc>>(0).ok().map(PluckValue)
    }

    fn from_postgres_row(row: &PostgresRow) -> Option<Self> {
        Self::from_deadpool_posgres_row(row)
    }
}

impl DinocoPostgres for PluckValue<Option<DateTime<Utc>>> {
    fn from_deadpool_posgres_row(row: &DeadpoolPostgresRow) -> Option<Self> {
        row.try_get::<_, Option<DateTime<Utc>>>(0).ok().map(PluckValue)
    }

    fn from_postgres_row(row: &PostgresRow) -> Option<Self> {
        Self::from_deadpool_posgres_row(row)
    }
}

impl DinocoMysql for PluckValue<DateTime<Utc>> {
    fn from_mysql_row(row: &MysqlRow) -> Option<Self> {
        let mut row = row.clone();
        row.take::<NaiveDateTime, _>(0).map(|value| PluckValue(value.and_utc()))
    }
}

impl DinocoMysql for PluckValue<Option<DateTime<Utc>>> {
    fn from_mysql_row(row: &MysqlRow) -> Option<Self> {
        let mut row = row.clone();
        row.take::<Option<NaiveDateTime>, _>(0).map(|value| PluckValue(value.map(|value| value.and_utc())))
    }
}