toasty 0.2.0

An async ORM for Rust supporting SQL and NoSQL databases
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

use crate::{
    Result,
    engine::exec::{Action, Exec, Output, VarId},
};
use toasty_core::{
    driver::{Rows, operation},
    schema::db::{ColumnId, TableId},
    stmt::ValueStream,
};

/// Get a model by key
#[derive(Debug)]
pub(crate) struct GetByKey {
    /// Where to get the keys to load
    pub input: VarId,

    /// Where to store the result
    pub output: Output,

    /// Table to query
    pub table: TableId,

    /// Columns to get
    pub columns: Vec<ColumnId>,
}

impl Exec<'_> {
    pub(super) async fn action_get_by_key(&mut self, action: &GetByKey) -> Result<()> {
        let keys: Vec<_> = self
            .vars
            .load(action.input)
            .await?
            .collect_as_value()
            .await?
            .into_list_unwrap()
            .into_iter()
            .filter(|k| !k.is_null())
            .collect();

        let res = if keys.is_empty() {
            Rows::value_stream(ValueStream::default())
        } else {
            let op = operation::GetByKey {
                table: action.table,
                select: action.columns.clone(),
                keys,
            };

            let res = self.connection.exec(&self.engine.schema, op.into()).await?;
            res.rows
        };

        self.vars
            .store(action.output.var, action.output.num_uses, res);
        Ok(())
    }
}

impl From<GetByKey> for Action {
    fn from(src: GetByKey) -> Self {
        Self::GetByKey(src)
    }
}