encapsulate Document

yorkie-team · May 27, 2024 · 9a2e440 · 9a2e440
1 parent 677fa29
commit 9a2e440
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/yorkie/src/main/kotlin/dev/yorkie/core/Client.kt b/yorkie/src/main/kotlin/dev/yorkie/core/Client.kt
@@ -489,7 +489,7 @@ public class Client @VisibleForTesting internal constructor(
                 return@async SUCCESS
             }
 
-            document.status = DocumentStatus.Attached
+            document.setStatus(DocumentStatus.Attached)
             attachments.value += document.key to Attachment(
                 document,
                 response.documentId,
@@ -535,7 +535,7 @@ public class Client @VisibleForTesting internal constructor(
             val pack = response.changePack.toChangePack()
             document.applyChangePack(pack)
             if (document.status != DocumentStatus.Removed) {
-                document.status = DocumentStatus.Detached
+                document.setStatus(DocumentStatus.Detached)
                 attachments.value -= document.key
             }
             SUCCESS

diff --git a/yorkie/src/main/kotlin/dev/yorkie/document/Document.kt b/yorkie/src/main/kotlin/dev/yorkie/document/Document.kt
@@ -94,7 +94,7 @@ public class Document(
 
     @Volatile
     public var status = DocumentStatus.Detached
-        internal set
+        private set
 
     @VisibleForTesting
     public val garbageLength: Int
@@ -451,6 +451,10 @@ public class Document(
         onlineClients.value -= actorID
     }
 
+    internal fun setStatus(newStatus: DocumentStatus) {
+        this.status = newStatus
+    }
+
     /**
      * Deletes elements that were removed before the given time.
      */