fix: offsetHeight null error (react-component#259)

bagrinsergiu · Jul 3, 2024 · 6f9d98a · 6f9d98a
1 parent 20e940a
commit 6f9d98a
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/hooks/useHeights.tsx b/src/hooks/useHeights.tsx
@@ -1,6 +1,5 @@
 import * as React from 'react';
 import { useRef, useEffect } from 'react';
-import findDOMNode from 'rc-util/lib/Dom/findDOMNode';
 import raf from 'rc-util/lib/raf';
 import type { GetKey } from '../interface';
 import CacheMap from '../utils/CacheMap';
@@ -30,10 +29,10 @@ export default function useHeights<T>(
     const doCollect = () => {
       instanceRef.current.forEach((element, key) => {
         if (element && element.offsetParent) {
-          const htmlElement = findDOMNode<HTMLElement>(element);
-          const { offsetHeight } = htmlElement;
+          const { offsetHeight } = element;
+
           if (heightsRef.current.get(key) !== offsetHeight) {
-            heightsRef.current.set(key, htmlElement.offsetHeight);
+            heightsRef.current.set(key, offsetHeight);
           }
         }
       });