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

#![cfg_attr(feature = "clippy", feature(plugin))]
#![cfg_attr(feature = "clippy", plugin(clippy))]

extern crate diesel;

use diesel::prelude::*;
use diesel::associations::HasTable;

pub trait Model<'a>
where
    &'a Self: Identifiable,
    Self: Sized + 'a,
{
    fn save(self, conn: &PgConnection) -> QueryResult<Self>;
    fn find_all(conn: &PgConnection) -> QueryResult<Vec<Self>>;
    fn find_one(
        conn: &PgConnection,
        id: <&'a Self as Identifiable>::Id,
    ) -> QueryResult<Option<Self>>;
    fn exists(conn: &PgConnection, id: <&'a Self as Identifiable>::Id) -> QueryResult<bool>;
    fn count_all(conn: &PgConnection) -> QueryResult<i64>;
    fn destroy(self, conn: &PgConnection) -> QueryResult<()>;
}

pub trait NewModel<'a, T>
where
    &'a T: HasTable,
    T: 'a,
    &'a Self: Insertable<<&'a T as HasTable>::Table>,
    Self: 'a,
{
    fn save(self, conn: &PgConnection) -> QueryResult<T>;
}