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

Ant Design Charts 是一个基于 G6 的关系数据可视化库，其中的组织结构图(Organization Chart)组件提供了展示层级关系的功能。在实际业务场景中，我们经常需要控制节点的默认展开或折叠状态，本文将详细介绍如何通过数据和API两种方式实现这一功能。

通过数据控制节点状态

在数据结构中，我们可以直接为节点指定collapsed属性来控制其初始状态：

const data = {
  nodes: [
    {
      id: 'root',
      children: ['child1', 'child2'],
      style: { collapsed: true }  // 初始状态为折叠
    },
    {
      id: 'child1',
      children: ['grandchild1']
    },
    {
      id: 'child2'
    }
  ]
};

这种方式适合静态数据场景，当图表初始化时就会按照指定的状态渲染。需要注意的是，collapsed属性需要放在节点的style对象中。

通过API动态控制节点状态

对于需要动态控制的场景，我们可以使用G6提供的API方法：

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

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

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

实际应用建议

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

2. 性能优化：对于大型组织结构图，批量操作节点时建议使用graph.collapseAll()或graph.expandAll()

3. 用户体验：在动态控制时，可以添加动画效果提升体验，G6默认会带有展开/折叠动画

4. 状态同步：如果应用需要保持节点状态，建议在回调函数中记录当前展开/折叠状态

通过这两种方式，开发者可以灵活控制组织结构图的展示层级，满足各种业务场景的需求。Ant Design Charts基于G6的强大能力，为复杂关系数据的可视化提供了便捷的解决方案。