fix(#2073): the update will not take effect before unmount (#2074)

Co-authored-by: lxr <1076629390@qq.com>
alibaba · Feb 23, 2023 · e7682c1 · e7682c1
1 parent 73ac860
commit e7682c1
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/packages/hooks/src/useCookieState/index.ts b/packages/hooks/src/useCookieState/index.ts
@@ -28,15 +28,15 @@ function useCookieState(cookieKey: string, options: Options = {}) {
       newOptions: Cookies.CookieAttributes = {},
     ) => {
       const { defaultValue, ...restOptions } = { ...options, ...newOptions };
-      setState((prevState) => {
-        const value = isFunction(newValue) ? newValue(prevState) : newValue;
-        if (value === undefined) {
-          Cookies.remove(cookieKey);
-        } else {
-          Cookies.set(cookieKey, value, restOptions);
-        }
-        return value;
-      });
+      const value = isFunction(newValue) ? newValue(state) : newValue;
+
+      setState(value);
+
+      if (value === undefined) {
+        Cookies.remove(cookieKey);
+      } else {
+        Cookies.set(cookieKey, value, restOptions);
+      }
     },
   );