crates/db_schema/src/impls/custom_emoji.rs

   1 use crate::{
   2   newtypes::CustomEmojiId,
   3   schema::{
   4     custom_emoji::dsl::custom_emoji,
   5     custom_emoji_keyword::dsl::{custom_emoji_id, custom_emoji_keyword},
   6   },
   7   source::{
   8     custom_emoji::{CustomEmoji, CustomEmojiInsertForm, CustomEmojiUpdateForm},
   9     custom_emoji_keyword::{CustomEmojiKeyword, CustomEmojiKeywordInsertForm},
  10   },
  11   utils::{get_conn, DbPool},
  12 };
  13 use diesel::{dsl::insert_into, result::Error, ExpressionMethods, QueryDsl};
  14 use diesel_async::RunQueryDsl;
  15
  16 impl CustomEmoji {
  17   pub async fn create(pool: &DbPool, form: &CustomEmojiInsertForm) -> Result<Self, Error> {
  18     let conn = &mut get_conn(pool).await?;
  19     insert_into(custom_emoji)
  20       .values(form)
  21       .get_result::<Self>(conn)
  22       .await
  23   }
  24   pub async fn update(
  25     pool: &DbPool,
  26     emoji_id: CustomEmojiId,
  27     form: &CustomEmojiUpdateForm,
  28   ) -> Result<Self, Error> {
  29     let conn = &mut get_conn(pool).await?;
  30     diesel::update(custom_emoji.find(emoji_id))
  31       .set(form)
  32       .get_result::<Self>(conn)
  33       .await
  34   }
  35   pub async fn delete(pool: &DbPool, emoji_id: CustomEmojiId) -> Result<usize, Error> {
  36     let conn = &mut get_conn(pool).await?;
  37     diesel::delete(custom_emoji.find(emoji_id))
  38       .execute(conn)
  39       .await
  40   }
  41 }
  42
  43 impl CustomEmojiKeyword {
  44   pub async fn create(
  45     pool: &DbPool,
  46     form: Vec<CustomEmojiKeywordInsertForm>,
  47   ) -> Result<Vec<Self>, Error> {
  48     let conn = &mut get_conn(pool).await?;
  49     insert_into(custom_emoji_keyword)
  50       .values(form)
  51       .get_results::<Self>(conn)
  52       .await
  53   }
  54   pub async fn delete(pool: &DbPool, emoji_id: CustomEmojiId) -> Result<usize, Error> {
  55     let conn = &mut get_conn(pool).await?;
  56     diesel::delete(custom_emoji_keyword.filter(custom_emoji_id.eq(emoji_id)))
  57       .execute(conn)
  58       .await
  59   }
  60 }