Fix get()

Exelord · Mar 27, 2024 · 5f97a11 · 5f97a11
1 parent 4a35817
commit 5f97a11
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/src/primitives/map.ts b/src/primitives/map.ts
@@ -59,6 +59,7 @@ export class SignaledMap<K, V> extends Map<K, V> {
   }
 
   get(key: K): V | undefined {
+    track(key, this.valuesCache);
     return super.get(key);
   }
 
@@ -100,7 +101,6 @@ export class SignaledMap<K, V> extends Map<K, V> {
       batch(() => {
         dirtyAll(this.keysCache);
         dirtyAll(this.valuesCache);
-        dirty(OBJECT_KEYS, this.keysCache);
       });
     }
   }

diff --git a/src/primitives/set.ts b/src/primitives/set.ts
@@ -11,15 +11,15 @@ export class SignaledSet<T> extends Set<T> {
     if (values) for (const v of values) super.add(v);
   }
 
+  [Symbol.iterator](): IterableIterator<T> {
+    return this.values();
+  }
+
   get size(): number {
     track(OBJECT_KEYS, this.valuesCache);
     return super.size;
   }
 
-  [Symbol.iterator](): IterableIterator<T> {
-    return this.values();
-  }
-
   keys(): IterableIterator<T> {
     return this.values();
   }
@@ -80,7 +80,6 @@ export class SignaledSet<T> extends Set<T> {
       super.clear();
       batch(() => {
         dirtyAll(this.valuesCache);
-        dirty(OBJECT_KEYS, this.valuesCache);
       });
     }
   }