crablog/site/src/db.rs

mod models;
mod schema;

use crate::config;
use diesel::prelude::*;
use diesel::sqlite::SqliteConnection;
use models::*;

fn establish_connection() -> SqliteConnection {
    let root_path = config::get_from_env("ROOT_PATH", true);
    let db_path = root_path + "/db.sqlite3";
    SqliteConnection::establish(&db_path)
        .unwrap_or_else(|_| panic!("Error, connection to {} failed.", &db_path))
}

pub fn get_posts() -> std::vec::Vec<Post> {
    use crate::db::schema::posts::dsl::*;
    let connection = establish_connection();
    posts
        .filter(published.eq(true))
        .order(id.desc())
        .limit(5)
        .load::<Post>(&connection)
        .expect("Error, couldn't load posts.")
}

pub fn get_post_by_id(_id: i32) -> Post {
    use crate::db::schema::posts::dsl::*;
    let connection = establish_connection();
    posts
        .find(_id)
        .get_result(&connection)
        .expect("Error, couldn't find post")
}

pub fn add_post(title: &str, body: &str) {
    use chrono::prelude::*;
    use schema::posts;

    let connection = establish_connection();

    let new_post = NewPost {
        title,
        body,
        published: &true,
        publish_date: &Utc::now().naive_utc(),
    };

    diesel::insert_into(posts::table)
        .values(&new_post)
        .execute(&connection)
        .unwrap_or_else(|_| panic!("Error, couldn't insert new Post."));
}