fix: Table MeasureRow bug when children have refs (#1367)

* fix: Table MeasureRow bug when children have refs

* refactor: remove unused ColumnType import in MeasureCell.tsx

---------

Co-authored-by: keda <[email protected]>

Author: afc163

src/Body/MeasureCell.tsx

@@ -1,20 +1,19 @@
 import * as React from 'react';
 import ResizeObserver from 'rc-resize-observer';
 import useLayoutEffect from 'rc-util/lib/hooks/useLayoutEffect';
-import type { ColumnType } from '../interface';
 
 export interface MeasureCellProps {
   columnKey: React.Key;
   onColumnResize: (key: React.Key, width: number) => void;
-  column?: ColumnType<any>;
   prefixCls: string;
+  title?: React.ReactNode;
 }
 
 export default function MeasureCell({
   columnKey,
   onColumnResize,
-  column,
   prefixCls,
+  title,
 }: MeasureCellProps) {
   const cellRef = React.useRef<HTMLTableCellElement>();
 
@@ -27,7 +26,7 @@ export default function MeasureCell({
   return (
     <ResizeObserver data={columnKey}>
       <th ref={cellRef} className={`${prefixCls}-measure-cell`}>
-        <div className={`${prefixCls}-measure-cell-content`}>{column?.title || '\xa0'}</div>
+        <div className={`${prefixCls}-measure-cell-content`}>{title || '\xa0'}</div>
       </th>
     </ResizeObserver>
   );

src/Body/MeasureRow.tsx

@@ -35,13 +35,17 @@ export default function MeasureRow({
       >
         {columnsKey.map(columnKey => {
           const column = columns.find(col => col.key === columnKey);
+          const rawTitle = column?.title;
+          const titleForMeasure = React.isValidElement<React.RefAttributes<any>>(rawTitle)
+            ? React.cloneElement(rawTitle, { ref: null })
+            : rawTitle;
           return (
             <MeasureCell
               prefixCls={prefixCls}
               key={columnKey}
               columnKey={columnKey}
               onColumnResize={onColumnResize}
-              column={column}
+              title={titleForMeasure}
             />
           );
         })}