path: root/src/models/post.rs

use diesel;
use diesel::prelude::*;
use schema::posts;
use utils::DbConn;

#[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)
            .unwrap_or_else(|_| panic!("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) {
        use schema::posts::dsl::*;
        diesel::insert_into(posts)
            .values(new_post)
            .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.");
    }

    pub fn delete(post_id: i32, conn: &DbConn) {
        use schema::posts::dsl::*;
        diesel::delete(posts.filter(id.eq(post_id)))
            .execute(&**conn)
            .unwrap_or_else(|_| panic!("Could not delete post with id {}", post_id));
    }
}