remove on document removal

yorkie-team · Jun 12, 2024 · 84c5054 · 84c5054
1 parent deb893f
commit 84c5054
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/yorkie/src/main/kotlin/dev/yorkie/core/Client.kt b/yorkie/src/main/kotlin/dev/yorkie/core/Client.kt
@@ -270,6 +270,7 @@ public class Client @VisibleForTesting internal constructor(
                             // be disconnected to not receive an event for that document.
                             if (document.status == DocumentStatus.Removed) {
                                 attachments.value -= document.key
+                                mutexForDocuments.remove(document.key)
                             }
                         }
                     }.onFailure {
@@ -561,6 +562,7 @@ public class Client @VisibleForTesting internal constructor(
                 if (document.status != DocumentStatus.Removed) {
                     document.status = DocumentStatus.Detached
                     attachments.value -= document.key
+                    mutexForDocuments.remove(document.key)
                 }
             }
             SUCCESS
@@ -618,6 +620,7 @@ public class Client @VisibleForTesting internal constructor(
                 val pack = response.changePack.toChangePack()
                 document.applyChangePack(pack)
                 attachments.value -= document.key
+                mutexForDocuments.remove(document.key)
             }
             SUCCESS
         }