perf(tree): code perf

Tencent · Dec 23, 2024 · c10fdaa · c10fdaa
1 parent 84bf0d3
commit c10fdaa
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/tree/hooks/useTreeVirtualScroll.ts b/src/tree/hooks/useTreeVirtualScroll.ts
@@ -3,6 +3,7 @@ import useVirtualScroll from '../../hooks/useVirtualScroll';
 import TreeNode from '../../_common/js/tree-v1/tree-node';
 import { TScroll } from '../../common';
 import type { TdTreeProps } from '../type';
+import useEventCallback from '../../hooks/useEventCallback';
 
 export default function useTreeVirtualScroll({
   treeRef,
@@ -48,7 +49,7 @@ export default function useTreeVirtualScroll({
   });
 
   let lastScrollY = -1;
-  const onInnerVirtualScroll = (e: WheelEvent) => {
+  const onInnerVirtualScroll = useEventCallback((e: WheelEvent) => {
     onScroll?.({ e });
     if (!isVirtual) {
       return;
@@ -62,7 +63,7 @@ export default function useTreeVirtualScroll({
       lastScrollY = -1;
     }
     lastScrollY = top;
-  };
+  });
 
   useEffect(() => {
     const treeList = treeRef?.current;
@@ -72,8 +73,7 @@ export default function useTreeVirtualScroll({
       // 卸载时取消监听
       treeList?.removeEventListener?.('scroll', onInnerVirtualScroll);
     };
-    // eslint-disable-next-line react-hooks/exhaustive-deps
-  }, [onInnerVirtualScroll]);
+  }, [treeRef, onInnerVirtualScroll]);
 
   const cursorStyle = {
     position: 'absolute',