dojo-orm 0.2.2

A simple ORM for Rust
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

use std::fmt::Debug;
use std::marker::PhantomData;

use bb8::Pool;
use bb8_postgres::PostgresConnectionManager;
use postgres_types::ToSql;
use tokio_postgres::NoTls;

use crate::execution::Execution;
use crate::query_builder::{QueryBuilder, QueryType};
use crate::Model;

pub struct InsertOperation<'a, T>
where
    T: Model + Debug,
{
    pub(crate) pool: &'a Pool<PostgresConnectionManager<NoTls>>,
    pub(crate) data: &'a T,
}

impl<'a, T> InsertOperation<'a, T>
where
    T: Model + Debug,
{
    pub async fn exec(&self) -> anyhow::Result<T>
    where
        T: Model + Debug,
    {
        let params = self.data.params();
        let qb = QueryBuilder::builder()
            .table_name(T::NAME)
            .columns(T::COLUMNS)
            .params(&params)
            .ty(QueryType::Insert)
            .is_returning(true)
            .build();

        let execution = Execution::new(&self.pool, &qb);
        execution.first_or_throw().await
    }
}

pub struct InsertManyOperation<'a, T> {
    pub(crate) pool: &'a Pool<PostgresConnectionManager<NoTls>>,
    pub(crate) data: &'a [T],
}

impl<'a, T> InsertManyOperation<'a, T> {
    pub async fn exec(&self) -> anyhow::Result<Vec<T>>
    where
        T: Model + Debug,
    {
        if self.data.is_empty() {
            return Ok(vec![]);
        }

        let mut params = vec![];
        for data in self.data {
            params.extend(data.params());
        }

        let qb = QueryBuilder::builder()
            .table_name(T::NAME)
            .columns(T::COLUMNS)
            .params(&params)
            .ty(QueryType::Insert)
            .is_returning(true)
            .build();

        let execution = Execution::new(&self.pool, &qb);
        execution.all().await
    }
}