path: root/src/models/post.rs

use ::schema::posts;
use diesel::prelude::*;
use diesel::{self, ExecuteDsl};

#[derive(AsChangeset, FromForm, Identifiable, Serialize, Queryable)]
pub struct Post {
    pub id: i32,
    pub title: String,
    pub body: String,
    pub published: bool,
}

#[derive(Default, FromForm, Insertable)]
#[table_name="posts"]
pub struct NewPost {
    pub title: String,
    pub body: String,
    pub published: bool,
}

impl Post {
    pub fn get_all(conn: ::DbConn) -> Vec<Post> {
        use ::schema::posts::dsl::*;
        posts.filter(published.eq(false))
            .limit(5)
            .load::<Post>(&*conn)
            .expect("Error loading posts")
    }

    fn get_internal(post_id: i32, conn: &::DbConn) -> Post {
        use ::schema::posts::dsl::*;
        posts.find(post_id)
            .get_result(&**conn)
            .expect(&format!("Unable to find post with id={}", post_id))
    }

    pub fn get(post_id: i32, conn: ::DbConn) -> Post {
        Post::get_internal(post_id, &conn)
    }

    pub fn create(new_post: &NewPost, conn: ::DbConn) {
        diesel::insert(new_post)
            .into(posts::table)
            .execute(&*conn)
            .expect("Error saving post.");
    }

    pub fn update(updated_post: &Post, conn: ::DbConn) {
        let p = Post::get_internal(updated_post.id, &conn);
        diesel::update(&p)
            .set(updated_post)
            .execute(&*conn)
            .expect("Error saving post.");
    }
}