From d773321666e57c2f535f8bd8a01927b66e5461c3 Mon Sep 17 00:00:00 2001
From: "ouxiaofeng.utlf" <ouxiaofeng.utlf@bytedance.com>
Date: Fri, 24 Oct 2025 16:21:14 +0800
Subject: [PATCH] feat: add forceRender api

---
 docs/examples/expandedRowRender.tsx |  3 +++
 src/Body/BodyRow.tsx                | 11 ++++++++++-
 src/Body/index.tsx                  |  3 +++
 src/Table.tsx                       |  3 +++
 src/context/TableContext.tsx        |  1 +
 src/interface.ts                    |  1 +
 6 files changed, 21 insertions(+), 1 deletion(-)

diff --git a/docs/examples/expandedRowRender.tsx b/docs/examples/expandedRowRender.tsx
index 791f5a7c7..5c4c7d059 100644
--- a/docs/examples/expandedRowRender.tsx
+++ b/docs/examples/expandedRowRender.tsx
@@ -146,6 +146,9 @@ const Demo = () => {
           rowExpandable,
           expandIconColumnIndex: expandIconPosition ? 1 : null,
           fixed: fixExpand,
+          forceRender: record => {
+            return record?.key === 0;
+          },
         }}
         scroll={{ x: fixColumns || scrollX ? 2000 : null, y: fixHeader ? 300 : null }}
         data={data}
diff --git a/src/Body/BodyRow.tsx b/src/Body/BodyRow.tsx
index aae9f4c7b..4271d92a2 100644
--- a/src/Body/BodyRow.tsx
+++ b/src/Body/BodyRow.tsx
@@ -31,6 +31,7 @@ export interface BodyRowProps<RecordType> {
     colSpan: number;
     sticky: number;
   };
+  forceRender?: boolean | ((record: RecordType, index: number) => boolean);
 }
 
 // ==================================================================================
@@ -138,6 +139,7 @@ const BodyRow = <RecordType extends { children?: readonly RecordType[] }>(
     cellComponent,
     scopeCellComponent,
     expandedRowInfo,
+    forceRender,
   } = props;
 
   const rowInfo = useRowInfo(record, rowKey, index, indent);
@@ -157,6 +159,13 @@ const BodyRow = <RecordType extends { children?: readonly RecordType[] }>(
   const expandedRef = React.useRef(false);
   expandedRef.current ||= expanded;
 
+  const shouldForceRender = () => {
+    if (typeof forceRender === 'function') {
+      return forceRender(record, index);
+    }
+    return !!forceRender;
+  };
+
   if (process.env.NODE_ENV !== 'production') {
     devRenderTimes(props);
   }
@@ -224,7 +233,7 @@ const BodyRow = <RecordType extends { children?: readonly RecordType[] }>(
 
   // ======================== Expand Row =========================
   let expandRowNode: React.ReactElement<ExpandedRowProps>;
-  if (rowSupportExpand && (expandedRef.current || expanded)) {
+  if (rowSupportExpand && (expandedRef.current || expanded || shouldForceRender?.())) {
     const expandContent = expandedRowRender(record, index, indent + 1, expanded);
 
     expandRowNode = (
diff --git a/src/Body/index.tsx b/src/Body/index.tsx
index ade0f5bf8..7476ccff5 100644
--- a/src/Body/index.tsx
+++ b/src/Body/index.tsx
@@ -36,6 +36,7 @@ const Body = <RecordType,>(props: BodyProps<RecordType>) => {
     styles,
     expandedRowOffset = 0,
     colWidths,
+    forceRender
   } = useContext(TableContext, [
     'prefixCls',
     'getComponent',
@@ -50,6 +51,7 @@ const Body = <RecordType,>(props: BodyProps<RecordType>) => {
     'expandedRowOffset',
     'fixedInfoList',
     'colWidths',
+    'forceRender'
   ]);
   const { body: bodyCls = {} } = classNames || {};
   const { body: bodyStyles = {} } = styles || {};
@@ -111,6 +113,7 @@ const Body = <RecordType,>(props: BodyProps<RecordType>) => {
           indent={indent}
           // Expanded row info
           expandedRowInfo={expandedRowInfo}
+          forceRender={forceRender}
         />
       );
     });
diff --git a/src/Table.tsx b/src/Table.tsx
index a8431dc4c..9c9d21f0e 100644
--- a/src/Table.tsx
+++ b/src/Table.tsx
@@ -892,6 +892,7 @@ const Table = <RecordType extends DefaultRecordType>(
       expandableType,
       expandRowByClick: expandableConfig.expandRowByClick,
       expandedRowRender: expandableConfig.expandedRowRender,
+      forceRender: expandableConfig.forceRender,
       expandedRowOffset: expandableConfig.expandedRowOffset,
       onTriggerExpand,
       expandIconColumnIndex: expandableConfig.expandIconColumnIndex,
@@ -953,6 +954,8 @@ const Table = <RecordType extends DefaultRecordType>(
       onTriggerExpand,
       expandableConfig.expandIconColumnIndex,
       expandableConfig.indentSize,
+      expandableConfig.forceRender,
+      expandableConfig,
       emptyNode,
 
       // Column
diff --git a/src/context/TableContext.tsx b/src/context/TableContext.tsx
index 19acc6366..a62925cbb 100644
--- a/src/context/TableContext.tsx
+++ b/src/context/TableContext.tsx
@@ -53,6 +53,7 @@ export interface TableContextProps<RecordType = any> {
   expandableType: ExpandableType;
   expandRowByClick: boolean;
   expandedRowRender: ExpandedRowRender<RecordType>;
+  forceRender?: boolean | ((record: RecordType) => boolean);
   expandIcon: RenderExpandIcon<RecordType>;
   onTriggerExpand: TriggerEventHandler<RecordType>;
   expandIconColumnIndex: number;
diff --git a/src/interface.ts b/src/interface.ts
index 2c2d67801..b6cd6375f 100644
--- a/src/interface.ts
+++ b/src/interface.ts
@@ -267,6 +267,7 @@ export interface ExpandableConfig<RecordType> {
   columnWidth?: number | string;
   fixed?: FixedType;
   expandedRowOffset?: number;
+  forceRender?: boolean|((record: RecordType) => boolean);
 }
 
 // =================== Render ===================