arel 0.2.9

a sql orm base sqlx
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

use crate::prelude::Arel;
use crate::statements::ArelStatement;
use std::{fmt::Debug, marker::PhantomData, marker::Send};

#[derive(Debug)]
pub struct Increment<M: Arel, T: Debug + Send> {
    field: String,
    step: T,
    where_fields: Vec<String>,
    where_values: Vec<crate::Value>,
    _marker: PhantomData<M>,
}

impl<M: Arel, T: Debug + Send> ArelStatement for Increment<M, T> {
    fn to_sql(&self) -> Option<crate::Sql> {
        let table_name = M::table_name();
        let full_column_name = format!(r#""{}"."{}""#, table_name, self.field);
        let mut final_sql = crate::Sql::new(format!(r#"UPDATE "{}" SET "{}" = COALESCE({}, 0)"#, table_name, self.field, full_column_name));
        // if self.step >= 0 {
        //     final_sql.push_str(&format!(" + {}", self.step));
        // } else {
        //     final_sql.push_str(&format!(" - {}", self.step.abs()));
        // }
        final_sql.push_str(&format!(" + ({:?})", self.step));

        final_sql.push_str(" WHERE ");
        let len = self.where_fields.len();
        for (idx, field) in self.where_fields.iter().enumerate() {
            let value = &self.where_values[idx];
            let mut sql = crate::Sql::default();
            sql.push_str_with_prepare_value(format!(r#""{}" = {}"#, field, sql.prepare_symbol()), value.clone());
            final_sql.push_sql(sql);
            if idx < len - 1 {
                final_sql.push_str(" AND ");
            }
        }
        Some(final_sql)
    }
}

impl<M: Arel, T: Debug + Send> Increment<M, T> {
    pub fn new<F: Into<String>, V: Into<crate::Value>>(field: String, step: T, where_fields: Vec<F>, where_values: Vec<V>) -> Self {
        Self {
            field,
            step,
            where_fields: where_fields.into_iter().map(|f| f.into()).collect(),
            where_values: where_values.into_iter().map(|v| v.into()).collect(),
            _marker: PhantomData::<M>,
        }
    }
}