perf: 优化明细表滚动性能

antvis · Feb 26, 2024 · 66e4d03 · 66e4d03
1 parent 1d52f89
commit 66e4d03
Showing 1 changed file with 31 additions and 33 deletions.
diff --git a/packages/s2-core/src/facet/header/table-col.ts b/packages/s2-core/src/facet/header/table-col.ts
@@ -27,19 +27,38 @@ export class TableColHeader extends ColHeader {
 
   public frozenTrailingColGroup: IGroup;
 
+  private finalColCount: number;
+
+  private finalTrailingColCount: number;
+
   constructor(cfg: ColHeaderConfig) {
     super(cfg);
     const { frozenColCount, frozenTrailingColCount } =
       this.headerConfig.spreadsheet?.options;
 
-    if (frozenColCount) {
+    const topLevelNodes =
+      this.headerConfig.spreadsheet?.facet?.layoutResult.colNodes.filter(
+        (cell) => {
+          return isTopLevelNode(cell);
+        },
+      );
+    const { colCount, trailingColCount } = getFrozenLeafNodesCount(
+      topLevelNodes,
+      frozenColCount,
+      frozenTrailingColCount,
+    );
+
+    this.finalColCount = colCount;
+    this.finalTrailingColCount = trailingColCount;
+
+    if (colCount) {
       this.frozenColGroup = this.addGroup({
         name: KEY_GROUP_COL_FROZEN,
         zIndex: FRONT_GROUND_GROUP_FROZEN_Z_INDEX,
       });
     }
 
-    if (frozenTrailingColCount) {
+    if (trailingColCount) {
       this.frozenTrailingColGroup = this.addGroup({
         name: KEY_GROUP_COL_FROZEN_TRAILING,
         zIndex: FRONT_GROUND_GROUP_FROZEN_Z_INDEX,
@@ -82,24 +101,16 @@ export class TableColHeader extends ColHeader {
 
   protected getCellGroup(node: Node) {
     const { spreadsheet } = this.headerConfig;
-    const { frozenColCount, frozenTrailingColCount } = spreadsheet?.options;
-    const topLevelNodes = spreadsheet?.facet?.layoutResult.colNodes.filter(
-      (cell) => {
-        return isTopLevelNode(cell);
-      },
-    );
-    const { colCount, trailingColCount } = getFrozenLeafNodesCount(
-      topLevelNodes,
-      frozenColCount,
-      frozenTrailingColCount,
-    );
-    if (isFrozenCol(getLeftLeafNode(node).colIndex, colCount)) {
+
+    const leafNode = getLeftLeafNode(node).colIndex;
+
+    if (isFrozenCol(leafNode, this.finalColCount)) {
       return this.frozenColGroup;
     }
     if (
       isFrozenTrailingCol(
-        getLeftLeafNode(node).colIndex,
-        trailingColCount,
+        leafNode,
+        this.finalTrailingColCount,
         spreadsheet?.facet?.layoutResult.colLeafNodes.length,
       )
     ) {
@@ -110,25 +121,12 @@ export class TableColHeader extends ColHeader {
 
   protected isColCellInRect(item: Node): boolean {
     const { spreadsheet } = this.headerConfig;
-    const { frozenColCount, frozenTrailingColCount } = spreadsheet?.options;
     const colLength = spreadsheet?.facet?.layoutResult.colLeafNodes.length;
-    const topLevelNodes = spreadsheet?.facet?.layoutResult.colNodes.filter(
-      (cell) => {
-        return isTopLevelNode(cell);
-      },
-    );
-    const { colCount, trailingColCount } = getFrozenLeafNodesCount(
-      topLevelNodes,
-      frozenColCount,
-      frozenTrailingColCount,
-    );
+
+    const leafNode = getLeftLeafNode(item).colIndex;
     if (
-      isFrozenCol(getLeftLeafNode(item).colIndex, colCount) ||
-      isFrozenTrailingCol(
-        getLeftLeafNode(item).colIndex,
-        trailingColCount,
-        colLength,
-      )
+      isFrozenCol(leafNode, this.finalColCount) ||
+      isFrozenTrailingCol(leafNode, this.finalTrailingColCount, colLength)
     ) {
       return true;
     }