WPF虚拟化TreeView：自定义搜索框选中项后选中状态消失问题

背景 #

自定义TreeView控件使用VirtualizingStackPanel展示约8000项，与自定义SearchBox同处一个GroupBox内，用户可通过搜索框查找并选中TreeView项。搜索框和TreeView的SelectedItem依赖属性均绑定到ViewModel的同一个SelectedItem属性。

为适配虚拟化，参考官方文档实现了自定义逻辑查找匹配的TreeViewItem，相关代码如下：

获取TreeViewItem的核心代码 #

private static TreeViewItem GetTreeViewItem(ItemsControl container, object item)
{
    if (container == null) 
        return null;

    var rootFolder = (Folder)container.DataContext;
    var folderTreeItem = (IFolderTreeItem)item;

    var parentStack = new Stack<Folder>();
    var parentFolder = folderTreeItem.ParentFolder;
    while (parentFolder != rootFolder)
    {
        parentStack.Push(parentFolder);
        parentFolder = parentFolder.ParentFolder;
    }

    var rootTreeViewItem = (TreeViewItem)container;
    rootTreeViewItem.BringIntoView();
    rootTreeViewItem.IsExpanded = true;
    
    var virtualizingPanel = GetVirtualizingStackPanel(container);
    var currentFolder = rootFolder;
    while (parentStack.Count > 0)
    {
        var childFolder = parentStack.Pop();
        
        var childIndex = currentFolder.SubFolders.IndexOf(childFolder);
        
        if (childIndex == -1)
            throw new Exception();
        
        virtualizingPanel.BringIntoView(childIndex);
        var currentTreeViewItem = (TreeViewItem)rootTreeViewItem.ItemContainerGenerator.ContainerFromIndex(childIndex);
        currentTreeViewItem.IsExpanded = true;
        currentFolder = childFolder;
        rootTreeViewItem = currentTreeViewItem;
        virtualizingPanel = GetVirtualizingStackPanel(currentTreeViewItem);
    }
    
    var currentIndex = rootTreeViewItem.ItemContainerGenerator.Items.Cast<IFolderTreeItem>().ToList().FindIndex(tv => tv.Equals(folderTreeItem));
    
    virtualizingPanel.BringIntoView(currentIndex);
    var treeView = (TreeViewItem)rootTreeViewItem.ItemContainerGenerator.ContainerFromIndex(currentIndex);
    
    return treeView;
}

数据模型定义 #

/// <summary>
/// Interface that propages path in order for a class to be displayed in a folder tree view
/// </summary>
internal interface IFolderTreeItem
{
    /// <summary>
    /// Gets the path folder tree where item located
    /// </summary>
    string FolderTreePath { get; }
    
    /// <summary>
    /// Gets or sets the folder where this item lives
    /// </summary>
    Folder ParentFolder { get; set;  }
    
    /// <summary>
    /// Gets the name of the object that will be displayed
    /// </summary>
    string Name { get; }
    
    /// <summary>
    /// Gets or sets the list of items to display on the tree view
    /// </summary>
    IEnumerable AllItems { get; }
}

注：Folder是IFolderTreeItem的实例；使用AllItems属性绑定HierarchicalDataTemplate；GetVirtualizingStackPanel()和GetVisualChild()实现与官方文档一致；未使用行为或附加属性（仅拖拽库，疑似无关）。

当前问题 #

所有功能基本正常，但部分通过搜索框选中的项，其TreeViewItem选中状态会在显示后立刻消失。

排查建议 #

• 检查绑定时序与容器生命周期：虚拟化下TreeViewItem容器会被复用，若在容器未完全生成/绑定完成时设置选中状态，后续容器复用或绑定更新可能覆盖选中状态。尝试在BringIntoView后，用Dispatcher.BeginInvoke以Render优先级延迟执行选中操作，确保容器已就绪。
• 验证SelectedItem绑定同步：确认ViewModel的SelectedItem属性正确触发PropertyChanged事件，TreeView的SelectedItem绑定设置了Mode=TwoWay。排查搜索框更新ViewModel属性的时机是否与TreeView自身选中逻辑存在竞争，导致状态被覆盖。
• 确认ItemContainerGenerator状态：调用ContainerFromIndex前，检查ItemContainerGenerator.Status是否为ContainersGenerated，若容器未生成，获取到的TreeViewItem可能无效。可监听ItemContainerGenerator.StatusChanged事件，等待容器生成完成后再设置选中。
• 调整BringIntoView后的操作时机：virtualizingPanel.BringIntoView(currentIndex)触发滚动或容器加载后，立即调用UpdateLayout()强制完成布局更新，再设置选中状态，避免项未完全加载导致状态丢失。
• 检查Equals方法逻辑：代码中通过tv.Equals(folderTreeItem)查找索引，确认IFolderTreeItem的Equals实现是否正确，是否存在误匹配导致选中错误项，随后被ViewModel的正确SelectedItem覆盖状态。
• 排查TreeView自定义选中逻辑：检查TreeView是否有额外的SelectedItemChanged事件处理逻辑，是否存在选中后自动取消选中的操作。

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