mirrors/zed
1
0
Fork 0
mirror of https://github.com/zed-industries/zed.git synced 2025-01-15 06:40:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 10

Leave joined worktrees when guest loses connection

Browse source
This commit is contained in:
Antonio Scandurra 2021-09-22 15:33:13 +02:00
parent b47422ae28
commit 3e65fb4267
1 changed files with 12 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
server/src/rpc/store.rs
View file

@ -81,7 +81,7 @@ impl Store {
&mut self,
connection_id: ConnectionId,
) -> tide::Result<RemovedConnectionState> {
let connection = if let Some(connection) = self.connections.get(&connection_id) {
let connection = if let Some(connection) = self.connections.remove(&connection_id) {
connection
} else {
return Err(anyhow!("no such connection"))?;
@ -109,18 +109,17 @@ impl Store {
.collaborator_ids
.extend(worktree.collaborator_user_ids.iter().copied());
result.hosted_worktrees.insert(worktree_id, worktree);
} else {
if let Some(worktree) = self.worktrees.get(&worktree_id) {
result
.guest_worktree_ids
.insert(worktree_id, worktree.connection_ids());
result
.collaborator_ids
.extend(worktree.collaborator_user_ids.iter().copied());
}
} else if let Some(worktree) = self.leave_worktree(connection_id, worktree_id) {
result
.guest_worktree_ids
.insert(worktree_id, worktree.connection_ids);
result.collaborator_ids.extend(worktree.collaborator_ids);
}
}
#[cfg(test)]
self.check_invariants();
Ok(result)
}
@ -376,8 +375,9 @@ impl Store {
let replica_id = share.guest_connection_ids.remove(&connection_id)?;
share.active_replica_ids.remove(&replica_id);
let connection = self.connections.get_mut(&connection_id)?;
connection.worktrees.remove(&worktree_id);
if let Some(connection) = self.connections.get_mut(&connection_id) {
connection.worktrees.remove(&worktree_id);
}
let connection_ids = worktree.connection_ids();
let collaborator_ids = worktree.collaborator_user_ids.clone();