sntl 0.1.0

Compile-time guarded ORM for PostgreSQL — your data's guardian from compile to production
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

//! Async execution methods for query builders.
//!
//! These methods require a live database connection and are excluded from
//! unit-test coverage. They will be covered by integration tests.

use super::{DeleteQuery, InsertQuery, SelectQuery};

/// Helper: convert `Vec<Value>` binds into driver params slice.
fn to_params(binds: &[crate::core::types::Value]) -> Vec<&(dyn driver::ToSql + Sync)> {
    binds
        .iter()
        .map(|v| v as &(dyn driver::ToSql + Sync))
        .collect()
}

impl SelectQuery {
    /// Execute this query and fetch all rows.
    pub async fn fetch_all(
        self,
        conn: &mut driver::Connection,
    ) -> crate::core::error::Result<Vec<driver::Row>> {
        let (sql, binds) = self.build();
        Ok(conn.query(&sql, &to_params(&binds)).await?)
    }

    /// Execute this query and fetch exactly one row.
    ///
    /// Returns `Error::NotFound` if no rows are returned.
    pub async fn fetch_one(
        self,
        conn: &mut driver::Connection,
    ) -> crate::core::error::Result<driver::Row> {
        let (sql, binds) = self.build();
        conn.query_one(&sql, &to_params(&binds))
            .await
            .map_err(Into::into)
    }

    /// Execute this query and fetch an optional row.
    pub async fn fetch_optional(
        self,
        conn: &mut driver::Connection,
    ) -> crate::core::error::Result<Option<driver::Row>> {
        let (sql, binds) = self.build();
        Ok(conn.query_opt(&sql, &to_params(&binds)).await?)
    }
}

impl InsertQuery {
    /// Execute this INSERT and return all rows via RETURNING clause.
    pub async fn fetch_returning(
        self,
        conn: &mut driver::Connection,
    ) -> crate::core::error::Result<Vec<driver::Row>> {
        let (sql, binds) = self.build();
        Ok(conn.query(&sql, &to_params(&binds)).await?)
    }

    /// Execute this INSERT and return the number of rows affected.
    pub async fn execute(self, conn: &mut driver::Connection) -> crate::core::error::Result<u64> {
        let (sql, binds) = self.build();
        Ok(conn.execute(&sql, &to_params(&binds)).await?)
    }
}

impl super::UpdateQuery {
    /// Execute this UPDATE and return all rows via RETURNING clause.
    pub async fn fetch_returning(
        self,
        conn: &mut driver::Connection,
    ) -> crate::core::error::Result<Vec<driver::Row>> {
        let (sql, binds) = self.build();
        Ok(conn.query(&sql, &to_params(&binds)).await?)
    }

    /// Execute this UPDATE and return the number of rows affected.
    pub async fn execute(self, conn: &mut driver::Connection) -> crate::core::error::Result<u64> {
        let (sql, binds) = self.build();
        Ok(conn.execute(&sql, &to_params(&binds)).await?)
    }
}

impl DeleteQuery {
    /// Execute this DELETE and return the number of rows affected.
    pub async fn execute(self, conn: &mut driver::Connection) -> crate::core::error::Result<u64> {
        let (sql, binds) = self.build();
        Ok(conn.execute(&sql, &to_params(&binds)).await?)
    }
}