feat: clean-up unused orgs

NEAR-DevHub · Nov 5, 2024 · fd3ed5d · fd3ed5d
1 parent 5ff3119
commit fd3ed5d
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 0 deletions.
diff --git a/server/.sqlx/query-2e7cbaee4000581fb5c34e4ef5738f5e1522bc5aff1e946b3bf0d4d19ea37267.json b/server/.sqlx/query-2e7cbaee4000581fb5c34e4ef5738f5e1522bc5aff1e946b3bf0d4d19ea37267.json
diff --git a/server/src/contract_pull.rs b/server/src/contract_pull.rs
@@ -120,6 +120,7 @@ async fn fetch_and_store_repos(
 ) -> anyhow::Result<()> {
     let organizations = near_client.repos().await?;
     DB::remove_non_existent_repos(tx, &organizations).await?;
+    DB::remove_empty_organizations(tx).await?;
     for org in organizations {
         let organization_id = DB::upsert_organization(tx, &org.organization)
             .await

diff --git a/server/src/db/mod.rs b/server/src/db/mod.rs
@@ -278,6 +278,17 @@ impl DB {
         Ok(())
     }
 
+    pub async fn remove_empty_organizations(
+        tx: &mut Transaction<'static, Postgres>,
+    ) -> anyhow::Result<()> {
+        sqlx::query!(
+            r#"DELETE FROM organizations WHERE id NOT IN (SELECT DISTINCT organization_id FROM repos)"#
+        )
+        .execute(tx.as_mut())
+        .await?;
+        Ok(())
+    }
+
     pub async fn upsert_pull_request(
         tx: &mut Transaction<'static, Postgres>,
         repo_id: i32,