WPF应用退出前如何对自定义派生控件执行存储与清理操作

嘿，这个需求其实挺常见的，核心就是要找到应用里所有存活的MyCustomControl实例，然后统一执行存储和清理逻辑。我给你两个实用的方案，你可以根据自己的场景选：

方案一：遍历可视化树查找控件 #

如果你的MyCustomControl都是在可视化树里（也就是已经加载到窗口/页面中），可以通过遍历所有窗口的可视化树来收集实例。这种方式不需要额外维护实例列表，简单直接。

首先写一个通用的可视化树遍历方法：

using System.Windows;
using System.Windows.Media;
using System.Collections.Generic;
using System.Linq;

public static class VisualTreeHelperExtensions
{
    public static IEnumerable<T> FindVisualChildren<T>(DependencyObject depObj) where T : DependencyObject
    {
        if (depObj == null) yield break;

        for (int i = 0; i < VisualTreeHelper.GetChildrenCount(depObj); i++)
        {
            var child = VisualTreeHelper.GetChild(depObj, i);
            if (child is T targetControl)
            {
                yield return targetControl;
            }

            // 递归遍历子控件
            foreach (var nestedChild in FindVisualChildren<T>(child))
            {
                yield return nestedChild;
            }
        }
    }
}

然后在App_OnExit方法里调用这个方法，遍历所有窗口的控件：

private void App_OnExit(object sender, ExitEventArgs e)
{
    // 遍历当前应用所有打开的窗口
    foreach (Window window in Application.Current.Windows)
    {
        // 查找当前窗口下所有的MyCustomControl
        var customControls = VisualTreeHelperExtensions.FindVisualChildren<MyCustomControl>(window);
        
        foreach (var control in customControls)
        {
            // 执行存储和清理操作（这里假设你已经在MyCustomControl里定义了对应的方法）
            control.SaveUserState();
            control.ReleaseResources();
        }
    }
}

优缺点：

• ✅ 优点：无需额外维护实例列表，实现简单
• ❌ 缺点：只能找到处于可视化树中的控件，如果控件被隐藏、未加载或者动态创建后没加入可视化树，会被遗漏

方案二：维护全局实例列表（更可靠） #

如果你的MyCustomControl可能存在不在可视化树里的情况，或者你想确保所有存活的实例都被处理，最好在基类里维护一个静态实例列表，在控件创建时加入，卸载时移除。

首先修改你的MyBaseControl：

using System.Collections.Generic;
using System.Windows.Controls;

public abstract class MyBaseControl : ContentControl
{
    // 静态列表存储所有存活的基类实例，加锁保证线程安全
    private static readonly List<MyBaseControl> _allActiveInstances = new List<MyBaseControl>();
    private static readonly object _lockObj = new object();

    protected MyBaseControl()
    {
        // 控件创建时加入列表
        lock (_lockObj)
        {
            _allActiveInstances.Add(this);
        }

        // 控件卸载时从列表移除，避免内存泄漏
        Unloaded += (s, e) =>
        {
            lock (_lockObj)
            {
                _allActiveInstances.Remove(this);
            }
        };
    }

    // 提供静态方法获取所有MyCustomControl实例
    public static IEnumerable<MyCustomControl> GetAllCustomControls()
    {
        lock (_lockObj)
        {
            // 转成List避免遍历过程中列表被修改
            return _allActiveInstances.OfType<MyCustomControl>().ToList();
        }
    }

    // 定义抽象方法，强制子类实现存储和清理逻辑
    public abstract void SaveState();
    public abstract void Cleanup();
}

然后让MyCustomControl实现这两个抽象方法：

public class MyCustomControl : MyBaseControl
{
    public override void SaveState()
    {
        // 这里写你的存储逻辑，比如保存用户输入、配置到本地文件或数据库
    }

    public override void Cleanup()
    {
        // 这里写清理逻辑，比如释放非托管资源、取消事件订阅、清理定时器等
    }
}

最后在App_OnExit里直接调用：

private void App_OnExit(object sender, ExitEventArgs e)
{
    var allCustomControls = MyBaseControl.GetAllCustomControls();
    foreach (var control in allCustomControls)
    {
        control.SaveState();
        control.Cleanup();
    }
}

优缺点：

• ✅ 优点：能找到所有存活的MyCustomControl实例，不管是否在可视化树里
• ✅ 优点：通过抽象方法强制子类实现必要的逻辑，规范统一
• ❌ 缺点：需要维护静态列表，要注意线程安全和内存泄漏问题（通过Unloaded事件移除实例解决）

额外注意事项 #

1. 异步操作处理：如果你的存储/清理逻辑是异步的，建议把App_OnExit改成异步方法，并确保操作完成前应用不会关闭。比如：

private async void App_OnExit(object sender, ExitEventArgs e)
{
    var allCustomControls = MyBaseControl.GetAllCustomControls();
    foreach (var control in allCustomControls)
    {
        // 假设SaveStateAsync是异步方法
        await control.SaveStateAsync();
        control.Cleanup();
    }
}

不过要注意，WPF的Exit事件默认不会等待异步操作完成，如果需要确保操作完成，可以考虑延迟关闭或者用Application.ShutdownMode配合逻辑控制。

2. 线程安全：如果你的控件可能在多线程环境下创建/销毁，一定要记得用锁保护静态列表，避免并发修改导致的异常。

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