fix: fix type

react-component · Sep 28, 2024 · 99f61c8 · 99f61c8
1 parent 5524853
commit 99f61c8
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/VirtualTable/BodyLine.tsx b/src/VirtualTable/BodyLine.tsx
@@ -41,7 +41,15 @@ const BodyLine = React.forwardRef<HTMLDivElement, BodyLineProps>((props, ref) =>
   let expandRowNode: React.ReactElement;
   if (rowSupportExpand && expanded) {
     const expandContent = expandedRowRender(record, index, indent + 1, expanded);
-    const computedExpandedRowClassName = expandedRowClassName?.(record, index, indent);
+
+    let computedExpandedRowClassName = '';
+
+    if (typeof expandedRowClassName === 'string') {
+      computedExpandedRowClassName = expandedRowClassName;
+    }
+    if (typeof expandedRowClassName === 'function') {
+      computedExpandedRowClassName = expandedRowClassName(record, index, indent);
+    }
 
     let additionalProps: React.TdHTMLAttributes<HTMLElement> = {};
     if (fixColumn) {