fix: Fix correct behavior of function setState parameter (#598)

mrousavy · Nov 7, 2023 · f391f9b · f391f9b
1 parent 6327d85
commit f391f9b
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/hooks.ts b/src/hooks.ts
@@ -59,13 +59,11 @@ function createMMKVHook<
   ): [value: T, setValue: (value: TSetAction) => void] => {
     const mmkv = instance ?? getDefaultInstance();
     const [value, setValue] = useState(() => getter(mmkv, key));
-    const valueRef = useRef<T>(value);
-    valueRef.current = value;
 
     // update value by user set
     const set = useCallback(
       (v: TSetAction) => {
-        const newValue = typeof v === 'function' ? v(valueRef.current) : v;
+        const newValue = typeof v === 'function' ? v(getter(mmkv, key)) : v;
         switch (typeof newValue) {
           case 'number':
           case 'string':