use serde_json::{Value as Json};
use std::marker::PhantomData;
use crate::traits::ModelAble;
use crate::statements::StatementAble;
use crate::nodes::SqlLiteral;
use crate::methods;
#[derive(Clone, Debug)]
pub struct Update<M: ModelAble> {
value: Json,
_marker: PhantomData<M>,
}
impl<M> StatementAble<M> for Update<M> where M: ModelAble {
fn json_value(&self) -> Option<&Json> {
Some(&self.value)
}
fn to_sql_literals(&self) -> Vec<SqlLiteral> {
let mut vec = vec![];
if let Some(json_value) = self.json_value() {
match json_value {
Json::Object(json_object) => {
for column_name in json_object.keys() {
let table_column_name = methods::table_column_name::<M>(column_name);
let json_value = json_object.get(column_name).unwrap();
vec.push(SqlLiteral::new(format!("{} = {}", table_column_name, self.json_value_sql(json_value))));
}
},
_ => vec.append(&mut StatementAble::to_sql_literals_default(self))
}
}
vec
}
fn to_sql(&self) -> String {
self.to_sql_with_concat(", ")
}
}
impl<M> Update<M> where M: ModelAble {
pub fn new(value: Json) -> Self {
Self {
value,
_marker: PhantomData,
}
}
}
#[cfg(test)]
mod tests {
use super::*;
use serde_json::{json};
#[test]
fn to_sql() {
#[derive(Clone, Debug)]
struct User {}
impl ModelAble for User {}
let update = Update::<User>::new(json!({
"name": "Tome",
"age": 18,
"active": true,
"profile": null
}));
assert_eq!(update.to_sql(), "`users`.`active` = 1, `users`.`age` = 18, `users`.`name` = 'Tome', `users`.`profile` = null");
let update = Update::<User>::new(json!("users.active = 1"));
assert_eq!(update.to_sql(), "users.active = 1");
let update = Update::<User>::new(json!(["users.active = ?", 1]));
assert_eq!(update.to_sql(), "users.active = 1");
}
}