blog/backend/src/cache.rs

use anyhow::Result;
use chrono::prelude::*;
use diesel::prelude::*;
use lazy_static::lazy_static;
use r2d2_redis::{r2d2, redis, RedisConnectionManager};
use serde::{Deserialize, Serialize};

use crate::{config, db, markdown, web};

pub type RedisPool = r2d2::Pool<RedisConnectionManager>;
pub type ConnectionPool = r2d2::PooledConnection<RedisConnectionManager>;
pub type Connection = redis::Connection;

pub fn establish_connection() -> Result<redis::Connection> {
    Ok(redis::Client::open(config::CONFIG.redis_url.as_str())?.get_connection()?)
}

pub fn pool() -> Result<RedisPool> {
    Ok(r2d2::Pool::builder().build(RedisConnectionManager::new(
        config::CONFIG.redis_url.as_str(),
    )?)?)
}

lazy_static! {
    pub static ref POOL: RedisPool = pool().unwrap();
}

pub mod keys {
    pub const POSTS: &str = "posts";

    pub const POST: &str = "post:";

    pub fn post(id: i32) -> String {
        format!("{}{}", POST, id)
    }

    pub const TAG_POSTS: &str = "tag_posts:";

    pub fn tag_posts(id: i32) -> String {
        format!("{}{}", TAG_POSTS, id)
    }
}

pub fn clear(redis_conn: &mut Connection) -> Result<()> {
    redis::cmd("UNLINK").arg(keys::POSTS).query(redis_conn)?;

    for key in redis::cmd("KEYS")
        .arg(format!("{}*", keys::POST))
        .query::<Vec<String>>(redis_conn)?
    {
        redis::cmd("UNLINK").arg(key).query(redis_conn)?;
    }

    for key in redis::cmd("KEYS")
        .arg(format!("{}*", keys::TAG_POSTS))
        .query::<Vec<String>>(redis_conn)?
    {
        redis::cmd("UNLINK").arg(key).query(redis_conn)?;
    }

    Ok(())
}

pub fn cache_posts(
    db_conn: &mut db::Connection,
    redis_conn: &mut Connection,
) -> Result<Vec<web::templates::PostIndexPost>> {
    if let Some(s) = redis::cmd("GET")
        .arg(keys::POSTS)
        .query::<Option<String>>(redis_conn)?
    {
        Ok(serde_json::from_str(&s)?)
    } else {
        let posts_cache: Vec<web::templates::PostIndexPost> = db::schema::posts::table
            .select((
                db::schema::posts::id,
                db::schema::posts::name,
                db::schema::posts::slug,
                db::schema::posts::description,
                db::schema::posts::published_at,
                db::schema::posts::edited_at,
            ))
            .filter(db::schema::posts::active)
            .order(db::schema::posts::published_at.desc())
            .load::<(i32, String, String, String, NaiveDate, Option<NaiveDate>)>(db_conn)?
            .into_iter()
            .map(
                |p: (i32, String, String, String, NaiveDate, Option<NaiveDate>)| -> Result<_> {
                    let (id, name, slug, description, published_at, edited_at) = p;
                    let tags = web::get_tags_by_post(id, db_conn)?;

                    Ok(web::templates::PostIndexPost {
                        name,
                        slug,
                        description,
                        published_at,
                        edited_at,
                        tags,
                    })
                },
            )
            .collect::<Result<_>>()?;

        redis::cmd("SET")
            .arg(keys::POSTS)
            .arg(serde_json::to_string(&posts_cache)?)
            .query(redis_conn)?;
        redis::cmd("EXPIRE")
            .arg(keys::POSTS)
            .arg(config::CONFIG.cache_ttl)
            .query(redis_conn)?;

        Ok(posts_cache)
    }
}

#[derive(Serialize, Deserialize, Debug)]
pub struct Post {
    pub name: String,
    pub slug: String,
    pub published_at: NaiveDate,
    pub edited_at: Option<NaiveDate>,
    pub tags: Vec<String>,
    pub content: String,
}

pub fn cache_post(
    id: i32,
    db_conn: &mut db::Connection,
    redis_conn: &mut Connection,
) -> Result<Post> {
    let key = keys::post(id);

    if let Some(s) = redis::cmd("GET")
        .arg(&key)
        .query::<Option<String>>(redis_conn)?
    {
        Ok(serde_json::from_str(&s)?)
    } else {
        let post = db::schema::posts::table
            .filter(db::schema::posts::id.eq(id))
            .first::<db::models::Post>(db_conn)?;

        let post_cache = Post {
            name: post.name,
            slug: post.slug,
            published_at: post.published_at,
            edited_at: post.edited_at,
            tags: web::get_tags_by_post(id, db_conn)?,
            content: markdown::to_html(&post.content),
        };

        redis::cmd("SET")
            .arg(&key)
            .arg(serde_json::to_string(&post_cache)?)
            .query(redis_conn)?;
        redis::cmd("EXPIRE")
            .arg(key)
            .arg(config::CONFIG.cache_ttl)
            .query(redis_conn)?;

        Ok(post_cache)
    }
}

pub fn cache_tag_posts(
    id: i32,
    db_conn: &mut db::Connection,
    redis_conn: &mut Connection,
) -> Result<Vec<web::templates::PostIndexPost>> {
    let key = keys::tag_posts(id);

    if let Some(s) = redis::cmd("GET")
        .arg(&key)
        .query::<Option<String>>(redis_conn)?
    {
        Ok(serde_json::from_str(&s)?)
    } else {
        let post_tag_post_ids = db::schema::post_tags::table
            .select(db::schema::post_tags::post_id)
            .filter(db::schema::post_tags::tag_id.eq(id))
            .load::<i32>(db_conn)?;

        let posts: Vec<web::templates::PostIndexPost> = db::schema::posts::table
            .select((
                db::schema::posts::id,
                db::schema::posts::name,
                db::schema::posts::slug,
                db::schema::posts::description,
                db::schema::posts::published_at,
                db::schema::posts::edited_at,
            ))
            .filter(db::schema::posts::id.eq_any(post_tag_post_ids))
            .filter(db::schema::posts::active)
            .order(db::schema::posts::published_at.desc())
            .load::<(i32, String, String, String, NaiveDate, Option<NaiveDate>)>(db_conn)?
            .into_iter()
            .map(
                |p: (i32, String, String, String, NaiveDate, Option<NaiveDate>)| -> Result<_> {
                    let (id, name, slug, description, published_at, edited_at) = p;
                    let tags = web::get_tags_by_post(id, db_conn)?;

                    Ok(web::templates::PostIndexPost {
                        name,
                        slug,
                        description,
                        published_at,
                        edited_at,
                        tags,
                    })
                },
            )
            .collect::<Result<_>>()?;

        redis::cmd("SET")
            .arg(&key)
            .arg(serde_json::to_string(&posts)?)
            .query(redis_conn)?;
        redis::cmd("EXPIRE")
            .arg(key)
            .arg(config::CONFIG.cache_ttl)
            .query(redis_conn)?;

        Ok(posts)
    }
}