Update packages/components/src/utils/hooks/use-prop-ref.ts

Co-authored-by: Haz <hazdiego@gmail.com>
WordPress · Jul 1, 2021 · 2e8b22d · 2e8b22d
1 parent 3559eca
commit 2e8b22d
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/packages/components/src/utils/hooks/use-prop-ref.ts b/packages/components/src/utils/hooks/use-prop-ref.ts
@@ -41,12 +41,12 @@ import { useEffect, useRef } from '@wordpress/element';
  * @param  prop
  * @return The prop reference.
  */
-export function usePropRef< T >( prop: T ): MutableRefObject< T > {
-	const propRef = useRef( prop );
+export function useLatestRef< T >( value: T ): RefObject< T > {
+	const ref = useRef( value );
 
-	useEffect( () => {
-		propRef.current = prop;
-	}, [ prop ] );
+	useLayoutEffect( () => {
+		ref.current = prop;
+	} );
 
-	return propRef;
+	return ref;
 }