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

[lemmy.git] / crates / db_views / src / site_view.rs

diff --git a/crates/db_views/src/site_view.rs b/crates/db_views/src/site_view.rs
index e77d7c6ce770298a51bb84b6b3026199d54499e4..f80666bad15c29b550cb91df48dbb39bb106f29b 100644 (file)
--- a/crates/db_views/src/site_view.rs
+++ b/crates/db_views/src/site_view.rs
@@ -1,24 +1,37 @@
-use diesel::{result::Error, *};
+use crate::structs::SiteView;
+use diesel::{result::Error, ExpressionMethods, JoinOnDsl, QueryDsl};
+use diesel_async::RunQueryDsl;
 use lemmy_db_schema::{
-  aggregates::site_aggregates::SiteAggregates,
-  schema::{site, site_aggregates},
-  source::site::Site,
+  aggregates::structs::SiteAggregates,
+  schema::{local_site, local_site_rate_limit, site, site_aggregates},
+  source::{local_site::LocalSite, local_site_rate_limit::LocalSiteRateLimit, site::Site},
+  utils::{get_conn, DbPool},
 };
-use serde::{Deserialize, Serialize};
-
-#[derive(Debug, Serialize, Deserialize, Clone)]
-pub struct SiteView {
-  pub site: Site,
-  pub counts: SiteAggregates,
-}
 
 impl SiteView {
-  pub fn read(conn: &PgConnection) -> Result<Self, Error> {
-    let (site, counts) = site::table
+  pub async fn read_local(pool: &mut DbPool<'_>) -> Result<Self, Error> {
+    let conn = &mut get_conn(pool).await?;
+    let (mut site, local_site, local_site_rate_limit, counts) = site::table
+      .inner_join(local_site::table)
+      .inner_join(
+        local_site_rate_limit::table.on(local_site::id.eq(local_site_rate_limit::local_site_id)),
+      )
       .inner_join(site_aggregates::table)
-      .select((site::all_columns, site_aggregates::all_columns))
-      .first::<(Site, SiteAggregates)>(conn)?;
+      .select((
+        site::all_columns,
+        local_site::all_columns,
+        local_site_rate_limit::all_columns,
+        site_aggregates::all_columns,
+      ))
+      .first::<(Site, LocalSite, LocalSiteRateLimit, SiteAggregates)>(conn)
+      .await?;
 
-    Ok(SiteView { site, counts })
+    site.private_key = None;
+    Ok(SiteView {
+      site,
+      local_site,
+      local_site_rate_limit,
+      counts,
+    })
   }
 }