refactor(store): minor

exuanbo · Sep 16, 2024 · 5eb1648 · 5eb1648
1 parent 4c327bd
commit 5eb1648
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/src/app/store/enhancers/subscribeChange.ts b/src/app/store/enhancers/subscribeChange.ts
@@ -9,23 +9,22 @@ type SubscribeChange = Store['subscribe']
 export const subscribeChange = injectStoreExtension<{ subscribeChange: SubscribeChange }>(
   <State>(store: Store<State>) => {
     const NIL = Symbol('NIL')
-    let stateSnapshot: State | typeof NIL = NIL
+    let snapshot: State | typeof NIL = NIL
 
     const dispatch: typeof store.dispatch = (action) => {
-      invariant(stateSnapshot === NIL)
+      invariant(snapshot === NIL)
+      snapshot = store.getState()
       try {
-        stateSnapshot = store.getState()
         return store.dispatch(action)
       } finally {
-        stateSnapshot = NIL
+        snapshot = NIL
       }
     }
 
     const subscribeChange: SubscribeChange = (listener) =>
       store.subscribe(() => {
-        invariant(stateSnapshot !== NIL)
-        const state = store.getState()
-        if (state !== stateSnapshot) {
+        invariant(snapshot !== NIL)
+        if (snapshot !== store.getState()) {
           listener()
         }
       })