perf: perf context

react-component · Jan 22, 2024 · 2d0e818 · 2d0e818
1 parent 8bded0c
commit 2d0e818
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/src/VirtualTable/BodyLine.tsx b/src/VirtualTable/BodyLine.tsx
@@ -28,7 +28,7 @@ const BodyLine = React.forwardRef<HTMLDivElement, BodyLineProps>((props, ref) =>
     TableContext,
     ['prefixCls', 'flattenColumns', 'fixColumn', 'componentWidth', 'scrollX'],
   );
-  const { getComponent } = useContext(StaticContext);
+  const { getComponent } = useContext(StaticContext, ['getComponent']);
 
   const rowInfo = useRowInfo(record, rowKey, index, indent);
 

diff --git a/src/VirtualTable/index.tsx b/src/VirtualTable/index.tsx
@@ -1,6 +1,6 @@
 import type { CompareProps } from '@rc-component/context/lib/Immutable';
 import classNames from 'classnames';
-import { warning } from 'rc-util';
+import { useEvent, warning } from 'rc-util';
 import * as React from 'react';
 import { INTERNAL_HOOKS } from '../constant';
 import { makeImmutable } from '../context/TableContext';
@@ -55,9 +55,8 @@ function VirtualTable<RecordType>(props: VirtualTableProps<RecordType>, ref: Rea
     }
   }
 
-  const getComponent = React.useCallback<GetComponent>(
+  const getComponent = useEvent<GetComponent>(
     (path, defaultComponent) => getValue(components, path) || defaultComponent,
-    [components],
   );
 
   // ========================= Context ==========================