fix: [tree] 修复设置树节点样式与类名不生效问题 fix kingdee#930

hxh2010 · Dec 6, 2024 · da0baff · da0baff
1 parent fe013e3
commit da0baff
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/components/tree/tree.tsx b/components/tree/tree.tsx
@@ -132,8 +132,8 @@ const InternalTree = React.forwardRef((props: TreeProps, ref: any): React.Functi
     onDrop,
     onSelect,
     selectedKeys: selectedKeysProps,
-    setTreeNodeStyle = () => ({}),
-    setTreeNodeClassName = () => '',
+    setTreeNodeStyle,
+    setTreeNodeClassName,
     estimatedItemSize: innerEstimatedItemSize,
     style,
     className,
@@ -495,8 +495,8 @@ const InternalTree = React.forwardRef((props: TreeProps, ref: any): React.Functi
     item.switcherIcon = item.switcherIcon || switcherIcon
     item.estimatedItemSize = estimatedItemSize
     item.draggable = draggable
-    item.className = setTreeNodeClassName(Object.assign({}, item))
-    item.style = setTreeNodeStyle(Object.assign({}, item))
+    item.className = setTreeNodeClassName?.(Object.assign({}, item)) || item?.className
+    item.style = setTreeNodeStyle?.(Object.assign({}, item)) || item?.style
     item.getDragNode = getDragNode
     item.setDragNode = setDragNode
     item.dragOver = dragOverNodeKey === item.key && dropPosition === 0