use std::borrow::Cow;
use sqlx::{Arguments, Database, Sqlite};
use time::PrimitiveDateTime;
use time::macros::datetime;
use taitan_orm_trait::traits::{Entity, SqliteEntity};
use taitan_orm_trait::traits::Parameter;
use taitan_orm_trait::result::Result;
use sqlx::mysql::MySqlArguments;
use sqlx::sqlite::SqliteArguments;
use sqlx::postgres::PgArguments;
fn call_entity(entity: &dyn SqliteEntity) -> Cow<'_, str> {
entity.gen_create_sql()
}
#[derive(Debug)]
struct User {
name: String,
created: PrimitiveDateTime
}
impl Parameter<Sqlite> for User {
fn add_to_args<'a, 'b>(&'a self, args: &'b mut <Sqlite as Database>::Arguments<'a>) -> Result<()> {
args.add(&self.name)?;
args.add(&self.created)?;
Ok(())
}
}
impl Entity<Sqlite> for User {
fn gen_insert_sql<'a>(&self) -> Cow<'a, str> {
let sql = "insert into users (name, created) values (?, ?)";
Cow::from(sql)
}
fn gen_upsert_sql<'a>(&self) -> Cow<'a, str> {
let sql = "insert into users (name, created) values (?, ?) on conflict (name) do update set created = ?";
Cow::from(sql)
}
fn gen_create_sql<'a>(&self) -> Cow<'a, str> {
let sql = "insert into users (name, created) values (?, ?)";
Cow::from(sql)
}
}
#[test]
fn entity_trait_spec() {
let user = User{name: "Allen".to_string(), created: datetime!(2019-01-01 0:00)};
let insert_args = <User as Parameter<Sqlite>>::gen_args(&user).unwrap();
call_entity(&user);
}