Keep else

swan-io · Dec 6, 2024 · 2714d0a · 2714d0a
1 parent b730a47
commit 2714d0a
Showing 1 changed file with 9 additions and 11 deletions.
diff --git a/packages/lake/src/hooks/usePersistedState.ts b/packages/lake/src/hooks/usePersistedState.ts
@@ -40,18 +40,16 @@ export const usePersistedState = <T>(key: string, defaultValue: T) => {
       if (typeof value !== "function") {
         const rawValue = value != null ? stringifyValue(value) : null;
         setItem(key, rawValue);
-        return setRawValue(rawValue);
+        setRawValue(rawValue);
+      } else {
+        setRawValue(prevState => {
+          const prevValue = parseRawValue(prevState, stableDefaultValue);
+          const nextValue = (value as (prevState: T) => T)(prevValue);
+          const rawValue = stringifyValue(nextValue);
+          setItem(key, rawValue);
+          return rawValue;
+        });
       }
-
-      setRawValue(prevState => {
-        const nextValue = (value as (prevState: T) => T)(
-          parseRawValue(prevState, stableDefaultValue),
-        );
-
-        const rawValue = stringifyValue(nextValue);
-        setItem(key, rawValue);
-        return rawValue;
-      });
     },
     [key, stableDefaultValue],
   );