chore: optimize codes logic

Author: bbb169

src/Cell/index.tsx

@@ -182,16 +182,19 @@ function Cell<RecordType>(
     additionalProps?.onMouseLeave?.(event);
   });
 
-  let mergedLastFixLeft = lastFixLeft;
-  const { current } = headerCellRefs;
-  const dom = current[colIndex];
-  if (lastFixLeft && dom) {
-    const offsetLeft =
-      dom.getBoundingClientRect().x - dom.parentElement.getBoundingClientRect().x || 0;
-
-    // should not be tagged as lastFixLeft if cell is not stickying;
-    mergedLastFixLeft = typeof fixLeft === 'number' && offsetLeft === fixLeft + bodyScrollLeft;
-  }
+  const mergedLastFixLeft = React.useMemo(() => {
+    const { current } = headerCellRefs;
+    const dom = current[colIndex];
+
+    if (lastFixLeft && dom && typeof fixLeft === 'number') {
+      const offsetLeft =
+        dom.getBoundingClientRect().x - dom.parentElement.getBoundingClientRect().x || 0;
+
+      // should not be tagged as lastFixLeft if cell is not stickying;
+      return offsetLeft === fixLeft + bodyScrollLeft;
+    }
+    return lastFixLeft;
+  }, [bodyScrollLeft, colIndex, fixLeft, headerCellRefs, lastFixLeft]);
   // ====================== Render ======================
   if (mergedColSpan === 0 || mergedRowSpan === 0) {
     return null;

src/Header/HeaderRow.tsx

@@ -47,6 +47,11 @@ const HeaderRow = <RecordType extends any>(props: RowProps<RecordType>) => {
 
   const columnsKey = getColumnsKey(cells.map(cell => cell.column));
 
+  const handleHeaderCellRef =
+    (cellIndex: number) => (headerCellRef: HTMLTableCellElement | null) => {
+      headerCellRefs.current[cellIndex] = headerCellRef;
+    };
+
   return (
     <RowComponent {...rowProps}>
       {cells.map((cell: CellType<RecordType>, cellIndex) => {
@@ -67,7 +72,7 @@ const HeaderRow = <RecordType extends any>(props: RowProps<RecordType>) => {
         return (
           <Cell
             colIndex={cellIndex}
-            ref={headerCellRef => (headerCellRefs.current[cellIndex] = headerCellRef)}
+            ref={handleHeaderCellRef(cellIndex)}
             {...cell}
             scope={column.title ? (cell.colSpan > 1 ? 'colgroup' : 'col') : null}
             ellipsis={column.ellipsis}