WPF TreeView刷新时如何保留节点IsExpanded状态？

这个问题我之前做XML编辑器类的功能时也碰到过——核心原因是调用XmlDataProvider.Refresh()时，整个数据源会被重新加载，TreeView的所有节点都会被重新生成，之前手动展开的状态自然就丢失了。下面给你两个实用的解决方向：

方向一：记录展开节点路径，刷新后恢复状态 #

这是最通用的方案，原理就是在刷新前先把所有展开节点的唯一标识（比如XML节点的XPath）记下来，刷新完成后再根据这些标识找到对应的TreeViewItem并重新展开。

步骤1：编写工具方法记录展开节点的XPath #

因为XML可能存在同名节点，所以生成XPath时要加上节点索引来保证唯一性：

// 递归收集所有展开节点的XPath
private List<string> GetExpandedNodePaths(TreeViewItem treeItem)
{
    var expandedPaths = new List<string>();
    if (treeItem.IsExpanded)
    {
        var xmlNode = treeItem.DataContext as XmlNode;
        if (xmlNode != null)
        {
            expandedPaths.Add(GetUniqueNodeXPath(xmlNode));
        }
        // 遍历子节点
        foreach (TreeViewItem childItem in treeItem.Items)
        {
            expandedPaths.AddRange(GetExpandedNodePaths(childItem));
        }
    }
    return expandedPaths;
}

// 生成包含索引的唯一XPath
private string GetUniqueNodeXPath(XmlNode node)
{
    // 根节点直接返回名称
    if (node.ParentNode == null || node.ParentNode.NodeType == XmlNodeType.Document)
    {
        return $"/{node.Name}";
    }
    // 计算当前节点在父节点中的索引（从1开始，符合XPath规则）
    int nodeIndex = node.ParentNode.ChildNodes.Cast<XmlNode>()
        .ToList().FindIndex(n => n == node) + 1;
    // 递归生成父节点的XPath，再拼接当前节点的名称和索引
    return $"{GetUniqueNodeXPath(node.ParentNode)}/{node.Name}[{nodeIndex}]";
}

步骤2：编写方法根据XPath恢复展开状态 #

需要遍历TreeView找到对应XML节点的容器，然后设置IsExpanded为true：

private void RestoreExpandedNodes(TreeView treeView, List<string> expandedPaths)
{
    foreach (var path in expandedPaths)
    {
        // 根据XPath找到XML节点
        var xmlProvider = (XmlDataProvider)treeView.DataContext;
        var targetNode = xmlProvider.Document.SelectSingleNode(path);
        if (targetNode == null) continue;
        
        // 找到对应的TreeViewItem并展开
        var targetItem = FindTreeViewItem(treeView, targetNode);
        if (targetItem != null)
        {
            targetItem.IsExpanded = true;
        }
    }
}

// 递归查找对应数据项的TreeViewItem容器
private TreeViewItem FindTreeViewItem(ItemsControl parentContainer, object dataItem)
{
    if (parentContainer.DataContext == dataItem)
    {
        return parentContainer as TreeViewItem;
    }
    // 遍历所有子项
    foreach (var item in parentContainer.Items)
    {
        var childContainer = parentContainer.ItemContainerGenerator.ContainerFromItem(item) as ItemsControl;
        var foundItem = FindTreeViewItem(childContainer, dataItem);
        if (foundItem != null)
        {
            return foundItem;
        }
    }
    return null;
}

步骤3：修改刷新逻辑，整合记录和恢复 #

在调用Refresh()前后加入记录和恢复的代码，注意要等待TreeView完成节点重建后再执行恢复：

// 第一步：记录当前所有展开节点的XPath
var expandedPaths = new List<string>();
foreach (var rootItem in OutlineWindowInstance.TreeItems.Items)
{
    var rootTreeItem = OutlineWindowInstance.TreeItems.ItemContainerGenerator.ContainerFromItem(rootItem) as TreeViewItem;
    if (rootTreeItem != null)
    {
        expandedPaths.AddRange(GetExpandedNodePaths(rootTreeItem));
    }
}

// 第二步：刷新XML数据源
var prov = (XmlDataProvider)OutlineWindowInstance.TreeItems.DataContext;
prov.Refresh();

// 第三步：等待TreeView加载完成后恢复展开状态
OutlineWindowInstance.TreeItems.Dispatcher.BeginInvoke(new Action(() =>
{
    RestoreExpandedNodes(OutlineWindowInstance.TreeItems, expandedPaths);
}), System.Windows.Threading.DispatcherPriority.Loaded);

方向二：优化刷新逻辑，避免全量节点重建 #

如果你的XML文件修改频率高且内容较大，全量刷新可能影响性能，可以尝试只更新变化的部分：

• 不要直接调用XmlDataProvider.Refresh()，而是手动读取修改后的XML内容，对比原XML的差异，只更新TreeView中变化的节点
• 或者使用ObservableCollection包装XML节点，当XML文件修改时，更新ObservableCollection中的项，这样TreeView只会更新变化的节点，展开状态自然保留

不过这个方案实现起来更复杂，需要自己处理XML的差异对比和节点映射，适合对性能要求较高的场景。

内容的提问来源于stack exchange，提问作者TheLogan