fix: add atoms to pending during restore

pmndrs · May 20, 2024 · 4d2e949 · 4d2e949
1 parent 43a2416
commit 4d2e949
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/vanilla/store2.ts b/src/vanilla/store2.ts
@@ -684,8 +684,15 @@ export const createStore = (): Store => {
         const pendingPair = createPending()
         for (const [atom, value] of values) {
           if (hasInitialValue(atom)) {
-            setAtomStateValueOrPromise(atom, getAtomState(atom), value)
-            recomputeDependents(pendingPair, atom)
+            const aState = getAtomState(atom)
+            const hasPrevValue = 'v' in aState
+            const prevValue = aState.v
+            setAtomStateValueOrPromise(atom, aState, value)
+            mountDependencies(pendingPair, atom, aState)
+            if (!hasPrevValue || !Object.is(prevValue, aState.v)) {
+              addPendingAtom(pendingPair, atom, aState)
+              recomputeDependents(pendingPair, atom)
+            }
           }
         }
         flushPending(pendingPair)