Make functions work with both connection and pool (#3420)

[lemmy.git] / crates / db_schema / src / aggregates / person_post_aggregates.rs

diff --git a/crates/db_schema/src/aggregates/person_post_aggregates.rs b/crates/db_schema/src/aggregates/person_post_aggregates.rs
index 5a2fb17d8610f0395a4320d63f83d586b728636c..1cbaa242215b4cd1df4c8a010171442ab0e80a95 100644 (file)
--- a/crates/db_schema/src/aggregates/person_post_aggregates.rs
+++ b/crates/db_schema/src/aggregates/person_post_aggregates.rs
@@ -2,14 +2,17 @@ use crate::{
   aggregates::structs::{PersonPostAggregates, PersonPostAggregatesForm},
   diesel::BoolExpressionMethods,
   newtypes::{PersonId, PostId},
-  schema::person_post_aggregates::dsl::*,
+  schema::person_post_aggregates::dsl::{person_id, person_post_aggregates, post_id},
   utils::{get_conn, DbPool},
 };
 use diesel::{insert_into, result::Error, ExpressionMethods, QueryDsl};
 use diesel_async::RunQueryDsl;
 
 impl PersonPostAggregates {
-  pub async fn upsert(pool: &DbPool, form: &PersonPostAggregatesForm) -> Result<Self, Error> {
+  pub async fn upsert(
+    pool: &mut DbPool<'_>,
+    form: &PersonPostAggregatesForm,
+  ) -> Result<Self, Error> {
     let conn = &mut get_conn(pool).await?;
     insert_into(person_post_aggregates)
       .values(form)
@@ -19,7 +22,11 @@ impl PersonPostAggregates {
       .get_result::<Self>(conn)
       .await
   }
-  pub async fn read(pool: &DbPool, person_id_: PersonId, post_id_: PostId) -> Result<Self, Error> {
+  pub async fn read(
+    pool: &mut DbPool<'_>,
+    person_id_: PersonId,
+    post_id_: PostId,
+  ) -> Result<Self, Error> {
     let conn = &mut get_conn(pool).await?;
     person_post_aggregates
       .filter(post_id.eq(post_id_).and(person_id.eq(person_id_)))