Reduce serializability of project delete (#11628)

This may reduce locks when deleting projects.

Release Notes:

- N/A
This commit is contained in:
Conrad Irwin 2024-05-09 16:17:13 -06:00 committed by GitHub
parent aa5113cd92
commit a3e75540af
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 37 additions and 21 deletions

View file

@ -415,7 +415,7 @@ impl Database {
if is_serialization_error(error) && prev_attempt_count < SLEEPS.len() { if is_serialization_error(error) && prev_attempt_count < SLEEPS.len() {
let base_delay = SLEEPS[prev_attempt_count]; let base_delay = SLEEPS[prev_attempt_count];
let randomized_delay = base_delay * self.rng.lock().await.gen_range(0.5..=2.0); let randomized_delay = base_delay * self.rng.lock().await.gen_range(0.5..=2.0);
log::info!( log::warn!(
"retrying transaction after serialization error. delay: {} ms.", "retrying transaction after serialization error. delay: {} ms.",
randomized_delay randomized_delay
); );

View file

@ -130,13 +130,21 @@ impl Database {
.await .await
} }
pub async fn delete_project(&self, project_id: ProjectId) -> Result<()> {
self.weak_transaction(|tx| async move {
project::Entity::delete_by_id(project_id).exec(&*tx).await?;
Ok(())
})
.await
}
/// Unshares the given project. /// Unshares the given project.
pub async fn unshare_project( pub async fn unshare_project(
&self, &self,
project_id: ProjectId, project_id: ProjectId,
connection: ConnectionId, connection: ConnectionId,
user_id: Option<UserId>, user_id: Option<UserId>,
) -> Result<TransactionGuard<(Option<proto::Room>, Vec<ConnectionId>)>> { ) -> Result<TransactionGuard<(bool, Option<proto::Room>, Vec<ConnectionId>)>> {
self.project_transaction(project_id, |tx| async move { self.project_transaction(project_id, |tx| async move {
let guest_connection_ids = self.project_guest_connection_ids(project_id, &tx).await?; let guest_connection_ids = self.project_guest_connection_ids(project_id, &tx).await?;
let project = project::Entity::find_by_id(project_id) let project = project::Entity::find_by_id(project_id)
@ -149,10 +157,7 @@ impl Database {
None None
}; };
if project.host_connection()? == connection { if project.host_connection()? == connection {
project::Entity::delete(project.into_active_model()) return Ok((true, room, guest_connection_ids));
.exec(&*tx)
.await?;
return Ok((room, guest_connection_ids));
} }
if let Some(dev_server_project_id) = project.dev_server_project_id { if let Some(dev_server_project_id) = project.dev_server_project_id {
if let Some(user_id) = user_id { if let Some(user_id) = user_id {
@ -169,7 +174,7 @@ impl Database {
}) })
.exec(&*tx) .exec(&*tx)
.await?; .await?;
return Ok((room, guest_connection_ids)); return Ok((false, room, guest_connection_ids));
} }
} }

View file

@ -2032,23 +2032,34 @@ async fn unshare_project_internal(
user_id: Option<UserId>, user_id: Option<UserId>,
session: &Session, session: &Session,
) -> Result<()> { ) -> Result<()> {
let (room, guest_connection_ids) = &*session let delete = {
.db() let room_guard = session
.await .db()
.unshare_project(project_id, connection_id, user_id) .await
.await?; .unshare_project(project_id, connection_id, user_id)
.await?;
let message = proto::UnshareProject { let (delete, room, guest_connection_ids) = &*room_guard;
project_id: project_id.to_proto(),
let message = proto::UnshareProject {
project_id: project_id.to_proto(),
};
broadcast(
Some(connection_id),
guest_connection_ids.iter().copied(),
|conn_id| session.peer.send(conn_id, message.clone()),
);
if let Some(room) = room {
room_updated(room, &session.peer);
}
*delete
}; };
broadcast( if delete {
Some(connection_id), let db = session.db().await;
guest_connection_ids.iter().copied(), db.delete_project(project_id).await?;
|conn_id| session.peer.send(conn_id, message.clone()),
);
if let Some(room) = room {
room_updated(room, &session.peer);
} }
Ok(()) Ok(())