[feat] Support customized ref in useDimensions (#2409)

* [FEAT] Support customized ref in useDimensions Signed-off-by: Ihor Dykhta <dikhta.igor@gmail.com> * deps Signed-off-by: Ihor Dykhta <dikhta.igor@gmail.com> --------- Signed-off-by: Ihor Dykhta <dikhta.igor@gmail.com> Co-authored-by: Rui Wang <64480766+albatross97@users.noreply.github.com>
keplergl · Oct 27, 2023 · 79002ea · 79002ea
1 parent 4d72331
commit 79002ea
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/utils/src/observe-dimensions.ts b/src/utils/src/observe-dimensions.ts
@@ -95,9 +95,10 @@ function getSize(node, entry): Dimensions | null {
  * @returns
  */
 export default function useDimensions<T extends Element>(
+  nodeRef?: RefObject<T>,
   throttleDelay = DEFAULT_THROTTLE_DELAY
 ): [RefObject<T>, Dimensions | null] {
-  const ref = useRef<T>(null);
+  const ref = nodeRef ?? useRef<T>(null);
   const [size, setSize] = useState(null);
 
   useEffect(() => {
@@ -124,7 +125,7 @@ export default function useDimensions<T extends Element>(
       unobserveDimensions(current);
     };
     // eslint-disable-next-line react-hooks/exhaustive-deps
-  }, [throttleDelay]);
+  }, [throttleDelay, ref?.current]);
 
   return [ref, size];
 }