mirror of
https://github.com/LemmyNet/lemmy.git
synced 2024-06-02 21:49:22 +00:00
40 lines
1.1 KiB
Rust
40 lines
1.1 KiB
Rust
use crate::{
|
|
schema::community_block::dsl::{community_block, community_id, person_id},
|
|
source::community_block::{CommunityBlock, CommunityBlockForm},
|
|
traits::Blockable,
|
|
utils::{DbPool, GetConn},
|
|
};
|
|
use diesel::{dsl::insert_into, result::Error, ExpressionMethods, QueryDsl};
|
|
use diesel_async::RunQueryDsl;
|
|
|
|
#[async_trait]
|
|
impl Blockable for CommunityBlock {
|
|
type Form = CommunityBlockForm;
|
|
async fn block(
|
|
mut pool: &mut impl GetConn,
|
|
community_block_form: &Self::Form,
|
|
) -> Result<Self, Error> {
|
|
let conn = &mut *pool.get_conn().await?;
|
|
insert_into(community_block)
|
|
.values(community_block_form)
|
|
.on_conflict((person_id, community_id))
|
|
.do_update()
|
|
.set(community_block_form)
|
|
.get_result::<Self>(conn)
|
|
.await
|
|
}
|
|
async fn unblock(
|
|
mut pool: &mut impl GetConn,
|
|
community_block_form: &Self::Form,
|
|
) -> Result<usize, Error> {
|
|
let conn = &mut *pool.get_conn().await?;
|
|
diesel::delete(
|
|
community_block
|
|
.filter(person_id.eq(community_block_form.person_id))
|
|
.filter(community_id.eq(community_block_form.community_id)),
|
|
)
|
|
.execute(conn)
|
|
.await
|
|
}
|
|
}
|