@colyseus/schema 3.0.36 → 3.0.37

package/build/esm/index.mjs

@@ -1030,19 +1030,35 @@ class ChangeTree {
     setRoot(root) {
         this.root = root;
         this.checkIsFiltered(this.parent, this.parentIndex);
+        //
+        // TODO: refactor and possibly unify .setRoot() and .setParent()
+        //
         // Recursively set root on child structures
         const metadata = this.ref.constructor[Symbol.metadata];
         if (metadata) {
             metadata[$refTypeFieldIndexes]?.forEach((index) => {
                 const field = metadata[index];
-                const value = this.ref[field.name];
-                value?.[$changes].setRoot(root);
+                const changeTree = this.ref[field.name]?.[$changes];
+                if (changeTree) {
+                    if (changeTree.root !== root) {
+                        changeTree.setRoot(root);
+                    }
+                    else {
+                        root.add(changeTree); // increment refCount
+                    }
+                }
             });
         }
         else if (this.ref[$childType] && typeof (this.ref[$childType]) !== "string") {
             // MapSchema / ArraySchema, etc.
             this.ref.forEach((value, key) => {
-                value[$changes].setRoot(root);
+                const changeTree = value[$changes];
+                if (changeTree.root !== root) {
+                    changeTree.setRoot(root);
+                }
+                else {
+                    root.add(changeTree); // increment refCount
+                }
             });
         }
     }
@@ -1066,14 +1082,19 @@ class ChangeTree {
         if (metadata) {
             metadata[$refTypeFieldIndexes]?.forEach((index) => {
                 const field = metadata[index];
-                const value = this.ref[field.name];
-                value?.[$changes].setParent(this.ref, root, index);
+                const changeTree = this.ref[field.name]?.[$changes];
+                if (changeTree && changeTree.root !== root) {
+                    changeTree.setParent(this.ref, root, index);
+                }
             });
         }
         else if (this.ref[$childType] && typeof (this.ref[$childType]) !== "string") {
             // MapSchema / ArraySchema, etc.
             this.ref.forEach((value, key) => {
-                value[$changes].setParent(this.ref, root, this.indexes[key] ?? key);
+                const changeTree = value[$changes];
+                if (changeTree.root !== root) {
+                    changeTree.setParent(this.ref, root, this.indexes[key] ?? key);
+                }
             });
         }
     }
@@ -1305,12 +1326,11 @@ class ChangeTree {
                 this.allFilteredChanges.indexes = {};
                 this.allFilteredChanges.operations.length = 0;
             }
-            // remove children references
-            this.forEachChild((changeTree, _) => this.root?.remove(changeTree));
         }
     }
     /**
      * Recursively discard all changes from this, and child structures.
+     * (Used in tests only)
      */
     discardAll() {
         const keys = Object.keys(this.indexedOperations);
@@ -1981,8 +2001,7 @@ class ArraySchema {
     push(...values) {
         let length = this.tmpItems.length;
         const changeTree = this[$changes];
-        // values.forEach((value, i) => {
-        for (let i = 0, l = values.length; i < values.length; i++, length++) {
+        for (let i = 0, l = values.length; i < l; i++, length++) {
             const value = values[i];
             if (value === undefined || value === null) {
                 // skip null values
@@ -2001,8 +2020,6 @@ class ArraySchema {
             //
             value[$changes]?.setParent(this, changeTree.root, length);
         }
-        //     length++;
-        // });
         return length;
     }
     /**
@@ -2081,21 +2098,9 @@ class ArraySchema {
         }
         // discard previous operations.
         const changeTree = this[$changes];
-        // discard children
-        changeTree.forEachChild((changeTree, _) => {
-            changeTree.discard(true);
-            //
-            // TODO: add tests with instance sharing + .clear()
-            // FIXME: this.root? is required because it is being called at decoding time.
-            //
-            // TODO: do not use [$changes] at decoding time.
-            //
-            const root = changeTree.root;
-            if (root !== undefined) {
-                root.removeChangeFromChangeSet("changes", changeTree);
-                root.removeChangeFromChangeSet("allChanges", changeTree);
-                root.removeChangeFromChangeSet("allFilteredChanges", changeTree);
-            }
+        // remove children references
+        changeTree.forEachChild((childChangeTree, _) => {
+            changeTree.root?.remove(childChangeTree);
         });
         changeTree.discard(true);
         changeTree.operation(OPERATION.CLEAR);
@@ -2680,6 +2685,10 @@ class MapSchema {
         // discard previous operations.
         changeTree.discard(true);
         changeTree.indexes = {};
+        // remove children references
+        changeTree.forEachChild((childChangeTree, _) => {
+            changeTree.root?.remove(childChangeTree);
+        });
         // clear previous indexes
         this.$indexes.clear();
         // clear items
@@ -3277,6 +3286,10 @@ class Schema {
         }
         return obj;
     }
+    /**
+     * Used in tests only
+     * @internal
+     */
     discardAllChanges() {
         this[$changes].discardAll();
     }
@@ -3299,8 +3312,11 @@ class Schema {
         const contents = (showContents) ? ` - ${JSON.stringify(ref.toJSON())}` : "";
         const changeTree = ref[$changes];
         const refId = changeTree.refId;
-        let output = "";
-        output += `${getIndent(level)}${ref.constructor.name} (refId: ${refId})${contents}\n`;
+        // log reference count if > 1
+        const refCount = (changeTree.root?.refCount?.[refId] > 1)
+            ? ` [×${changeTree.root.refCount[refId]}]`
+            : '';
+        let output = `${getIndent(level)}${ref.constructor.name} (refId: ${refId})${refCount}${contents}\n`;
         changeTree.forEachChild((childChangeTree) => output += this.debugRefIds(childChangeTree.ref, showContents, level + 1));
         return output;
     }
@@ -3485,6 +3501,10 @@ class CollectionSchema {
         // discard previous operations.
         changeTree.discard(true);
         changeTree.indexes = {};
+        // remove children references
+        changeTree.forEachChild((childChangeTree, _) => {
+            changeTree.root?.remove(childChangeTree);
+        });
         // clear previous indexes
         this.$indexes.clear();
         // clear items
@@ -3817,6 +3837,7 @@ class Root {
                 this.removeChangeFromChangeSet("filteredChanges", changeTree);
             }
             this.refCount[changeTree.refId] = 0;
+            changeTree.forEachChild((child, _) => this.remove(child));
         }
         else {
             this.refCount[changeTree.refId] = refCount;
@@ -3838,7 +3859,6 @@ class Root {
                 enqueueChangeTree(this, changeTree, "changes");
             }
         }
-        changeTree.forEachChild((child, _) => this.remove(child));
         return refCount;
     }
     removeChangeFromChangeSet(changeSetName, changeTree) {