refactor: stateRef to useLatest (alibaba#2407)

raotaohub · Mar 28, 2024 · 935b8b4 · 935b8b4
1 parent 15675cb
commit 935b8b4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/packages/hooks/src/useGetState/index.ts b/packages/hooks/src/useGetState/index.ts
@@ -1,5 +1,6 @@
 import type { Dispatch, SetStateAction } from 'react';
-import { useState, useRef, useCallback } from 'react';
+import { useState, useCallback } from 'react';
+import useLatest from '../useLatest';
 
 type GetStateAction<S> = () => S;
 
@@ -13,8 +14,7 @@ function useGetState<S = undefined>(): [
 ];
 function useGetState<S>(initialState?: S) {
   const [state, setState] = useState(initialState);
-  const stateRef = useRef(state);
-  stateRef.current = state;
+  const stateRef = useLatest(state);
 
   const getState = useCallback(() => stateRef.current, []);