Ant Design Charts 组织结构图节点展开/折叠控制详解

Ant Design Charts 是一个基于 G6 图可视化引擎的 React 图表库，其中的组织结构图(Organization Chart)组件常用于展示层级关系数据。在实际应用中，我们经常需要控制节点的默认展开或折叠状态，本文将详细介绍两种实现方式。

通过数据控制节点状态

最直接的方式是在数据结构中指定节点的初始状态。每个节点对象可以包含一个 style 属性，其中设置 collapsed: true 即可使该节点默认处于折叠状态：

const data = {
  nodes: [
    {
      id: 'root',
      children: ['node1', 'node2'],
      style: { collapsed: true }  // 默认折叠
    },
    {
      id: 'node1',
      children: ['node3']
    },
    {
      id: 'node2'
    },
    {
      id: 'node3'
    }
  ]
}

这种方式适合静态数据场景，当图表初次渲染时就会按照指定的状态显示。折叠的节点会显示一个展开图标，用户可以点击交互。

通过编程方式控制节点状态

对于需要动态控制的场景，我们可以通过获取图表实例来调用相关方法：

import { useRef, useEffect } from 'react';
import { OrganizationChart } from '@ant-design/charts';

function OrgChartWithControl() {
  const chartRef = useRef();
  
  useEffect(() => {
    if (chartRef.current) {
      // 折叠指定节点
      chartRef.current.collapseElement('node1');
      
      // 或者展开指定节点
      // chartRef.current.expandElement('node1');
    }
  }, [chartRef.current]);

  return (
    <OrganizationChart 
      data={data}
      onReady={(graph) => {
        chartRef.current = graph;
      }}
    />
  );
}

这种方式更加灵活，可以在组件生命周期中的任何时机调用，例如响应某个按钮点击事件后改变节点状态。

实际应用建议

1. 混合使用：可以结合两种方式，初始状态通过数据定义，后续交互通过编程控制

2. 性能考虑：对于大型组织结构图，批量操作节点时建议使用 collapseAll 或 expandAll 方法

3. 状态同步：如果需要在节点状态变化时执行操作，可以监听 'node:collapse' 和 'node:expand' 事件

4. 视觉反馈：可以为折叠/展开操作添加动画效果，提升用户体验

通过合理运用这些控制方法，可以创建出交互丰富、用户体验良好的组织结构图应用。