💄 style: 优化第一次水合逻辑

aleung · Aug 10, 2023 · fefae61 · fefae61
1 parent 226a875
commit fefae61
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/store/session/hooks/useSessionHydrated.ts b/src/store/session/hooks/useSessionHydrated.ts
@@ -3,12 +3,15 @@ import { useEffect, useState } from 'react';
 import { useSessionStore } from '@/store/session';
 
 export const useSessionHydrated = () => {
-  const [isInit, setInit] = useState(false);
+  // 根据 sessions 是否有值来判断是否已经初始化
+  const hasInited = !!Object.values(useSessionStore.getState().sessions).length;
+
+  const [isInit, setInit] = useState(hasInited);
 
   useEffect(() => {
     const hasRehydrated = useSessionStore.persist.hasHydrated();
 
-    if (hasRehydrated) {
+    if (hasRehydrated && !isInit) {
       setInit(true);
     }
   }, []);