fix: Reduce size of useControllerState().

jwalton · Aug 10, 2021 · 12a4297 · 12a4297
1 parent 6016c69
commit 12a4297
Showing 1 changed file with 11 additions and 11 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -129,17 +129,17 @@ export function useControllerState<S, R = Immutable<S>>(
     }
 
     // Update resultRef and force a re-render if the store changes.
-    useEffect(() => {
-        const unsubscribe = store.subscribe((state: Immutable<S>) => {
-            const newResult = latestSelector.current(state);
-            if (!latestIsEqual.current(resultRef.current, newResult)) {
-                resultRef.current = newResult;
-                setResult(newResult);
-            }
-        });
-
-        return () => unsubscribe();
-    }, [latestIsEqual, latestSelector, store]);
+    useEffect(
+        () =>
+            store.subscribe((state: Immutable<S>) => {
+                const newResult = latestSelector.current(state);
+                if (!latestIsEqual.current(resultRef.current, newResult)) {
+                    resultRef.current = newResult;
+                    setResult(newResult);
+                }
+            }),
+        [latestIsEqual, latestSelector, store]
+    );
 
     return resultRef.current;
 }