fix next.js warning for useLayoutEffect

theKashey · Jan 17, 2024 · f423da2 · f423da2
1 parent 365ee2b
commit f423da2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/useMergeRef.ts b/src/useMergeRef.ts
@@ -4,6 +4,8 @@ import { assignRef } from './assignRef';
 import { ReactRef } from './types';
 import { useCallbackRef } from './useRef';
 
+const useIsomorphicLayoutEffect = typeof window !== 'undefined' ? React.useLayoutEffect : React.useEffect;
+
 const currentValues = new WeakMap<any, ReactRef<any>[]>();
 
 /**
@@ -26,7 +28,7 @@ export function useMergeRefs<T>(refs: ReactRef<T>[], defaultValue?: T): React.Mu
   );
 
   // handle refs changes - added or removed
-  React.useLayoutEffect(() => {
+  useIsomorphicLayoutEffect(() => {
     const oldValue = currentValues.get(callbackRef);
 
     if (oldValue) {