From 532a5992394d96dfaf9bb8921aab8036368a23b6 Mon Sep 17 00:00:00 2001
From: Antonio Scandurra <me@as-cii.com>
Date: Thu, 17 Nov 2022 11:38:00 +0100
Subject: [PATCH] Use `Db::get_guest_connection_ids` in other db methods

---
 crates/collab/src/db.rs | 57 +++++------------------------------------
 1 file changed, 6 insertions(+), 51 deletions(-)

diff --git a/crates/collab/src/db.rs b/crates/collab/src/db.rs
index 6741afab7e..9485d1aae0 100644
--- a/crates/collab/src/db.rs
+++ b/crates/collab/src/db.rs
@@ -1771,24 +1771,9 @@ where
                 query.execute(&mut tx).await?;
             }
 
-            let connection_ids = sqlx::query_scalar::<_, i32>(
-                "
-                SELECT connection_id
-                FROM project_collaborators
-                WHERE project_id = $1 AND connection_id != $2
-                ",
-            )
-            .bind(project_id)
-            .bind(connection_id.0 as i32)
-            .fetch_all(&mut tx)
-            .await?;
-
+            let connection_ids = self.get_guest_connection_ids(project_id, &mut tx).await?;
             tx.commit().await?;
-
-            Ok(connection_ids
-                .into_iter()
-                .map(|connection_id| ConnectionId(connection_id as u32))
-                .collect())
+            Ok(connection_ids)
         })
         .await
     }
@@ -1846,24 +1831,9 @@ where
             .execute(&mut tx)
             .await?;
 
-            let connection_ids = sqlx::query_scalar::<_, i32>(
-                "
-                SELECT connection_id
-                FROM project_collaborators
-                WHERE project_id = $1 AND connection_id != $2
-                ",
-            )
-            .bind(project_id)
-            .bind(connection_id.0 as i32)
-            .fetch_all(&mut tx)
-            .await?;
-
+            let connection_ids = self.get_guest_connection_ids(project_id, &mut tx).await?;
             tx.commit().await?;
-
-            Ok(connection_ids
-                .into_iter()
-                .map(|connection_id| ConnectionId(connection_id as u32))
-                .collect())
+            Ok(connection_ids)
         })
         .await
     }
@@ -1908,24 +1878,9 @@ where
             .execute(&mut tx)
             .await?;
 
-            let connection_ids = sqlx::query_scalar::<_, i32>(
-                "
-                SELECT connection_id
-                FROM project_collaborators
-                WHERE project_id = $1 AND connection_id != $2
-                ",
-            )
-            .bind(project_id)
-            .bind(connection_id.0 as i32)
-            .fetch_all(&mut tx)
-            .await?;
-
+            let connection_ids = self.get_guest_connection_ids(project_id, &mut tx).await?;
             tx.commit().await?;
-
-            Ok(connection_ids
-                .into_iter()
-                .map(|connection_id| ConnectionId(connection_id as u32))
-                .collect())
+            Ok(connection_ids)
         })
         .await
     }