deuterium 0.0.1 

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


use update_query::{
    UpdateQuery,
    FieldUpdate,
};

use sql::from::{FromToSql};
use sql::value::{ToPredicateValue};
use sql::{SqlContext, ToSql, QueryToSql};

impl<F: ToPredicateValue, T: ToPredicateValue> ToSql for FieldUpdate<F, T> {
    fn to_sql(&self, ctx: &mut SqlContext) -> String {
        format!("{} = {}", self.get_field().to_predicate_value(ctx), self.get_value().to_sql(ctx))
    }
}

impl<T, L, M> ToSql for UpdateQuery<T, L, M> {
    fn to_sql(&self, ctx: &mut SqlContext) -> String {
        let mut sql = "UPDATE".to_string();

        if self.only {
            sql = format!("{} ONLY", sql)
        }

        sql = format!("{} {}", sql, self.table.to_from_sql(ctx));

        let updates_str: Vec<String> = self.updates.iter().map(|upd| upd.to_sql(ctx)).collect();
        sql = format!("{} SET {}", sql, updates_str.connect(", "));

        if self.from.is_some() {
            let from = self.from.as_ref().unwrap();
            if !from.is_empty() {
                let tables_str: Vec<String> = from.iter().map(|v| v.as_sql().to_from_sql(ctx)).collect();
                sql = format!("{} FROM {}", sql, tables_str.connect(", "))
            }
        }

        match self.where_.as_ref() {
            Some(predicate) => {
                sql = format!("{} WHERE {}", sql, predicate.to_sql(false, ctx))
            },
            None if !self.all => {
                // http://devopsreactions.tumblr.com/post/47352638154/almost-ran-update-without-where
                sql = format!("{} WHERE true = false", sql)
            },
            _ => ()
        }

        match &self.returning {
            &Some(ref select) => sql = format!("{} RETURNING {}", sql, select.to_sql(ctx)),
            &None => ()
        }

        sql
    }
}

impl<T, L, M> QueryToSql for UpdateQuery<T, L, M> {}