Swift 4.1中如何在扩展中实现对象销毁时的清理逻辑？

这个问题问得很到位！确实在Swift里，扩展是没办法直接声明deinit()的——而且就算语法允许，多个扩展都加deinit()也会因为重复声明报错。不过咱们有变通方案，能实现类似的效果，还支持多个扩展各自添加清理逻辑不冲突。

核心思路：关联对象+辅助类 #

利用Objective-C的**关联对象（Associated Objects）**特性，给目标对象绑定一个私有的辅助类实例。这个辅助类会在目标对象销毁时一同被销毁，我们把清理逻辑放在辅助类的deinit()里，就能间接实现扩展中的销毁清理。

具体实现步骤 #

1. 定义一个私有的辅助类，用来存储和执行清理任务：

private class CleanupHelper {
    // 存储多个扩展注册的清理闭包
    var cleanupTasks: [() -> Void] = []
    
    deinit {
        // 当辅助类被销毁时，执行所有注册的清理任务
        cleanupTasks.forEach { task in
            task()
        }
    }
}

2. 在目标类的扩展中，添加注册清理任务的方法，通过关联对象绑定辅助类：

// 以UIViewController为例，你可以替换成任何@objc兼容的类（继承NSObject或标记@objc）
extension UIViewController {
    // 关联对象的唯一标识key
    private static var cleanupHelperKey = 0
    
    /// 注册一个销毁时执行的清理任务
    func registerCleanupTask(_ task: @escaping () -> Void) {
        // 尝试获取已绑定的辅助类实例
        var helper = objc_getAssociatedObject(self, &Self.cleanupHelperKey) as? CleanupHelper
        
        // 如果没有，就创建一个并绑定
        if helper == nil {
            helper = CleanupHelper()
            objc_setAssociatedObject(
                self,
                &Self.cleanupHelperKey,
                helper,
                .OBJC_ASSOCIATION_RETAIN_NONATOMIC
            )
        }
        
        // 添加当前扩展的清理任务
        helper?.cleanupTasks.append(task)
    }
    
    // 举个例子：某个扩展的业务逻辑，注册对应的清理任务
    func setupTempNotificationObserver() {
        // 初始化逻辑：添加通知监听
        NotificationCenter.default.addObserver(
            self,
            selector: #selector(handleNotification),
            name: NSNotification.Name("TempNotification"),
            object: nil
        )
        
        // 注册清理任务：移除通知监听
        registerCleanupTask { [weak self] in
            guard let self = self else { return }
            NotificationCenter.default.removeObserver(self)
            print("通知监听已清理")
        }
    }
    
    @objc private func handleNotification() {
        // 处理通知逻辑
    }
}

为什么这个方案可行？ #

• 关联对象会和目标对象保持生命周期绑定：当目标对象被销毁时，绑定的CleanupHelper实例也会被释放，触发它的deinit()方法。
• 支持多个扩展：不同的扩展都可以通过registerCleanupTask()添加自己的清理闭包，所有任务会在对象销毁时依次执行，不会互相冲突。

注意事项 #

• 这个方案只适用于**@objc兼容的类**（比如继承自NSObject的类，或者用@objc标记的自定义Swift类）。纯Swift类（没有Objective-C兼容层）无法使用关联对象，这种情况如果是自定义类，建议在原类中添加一个清理任务数组，在原类的deinit()中执行，扩展里往这个数组加任务；如果是无法修改的系统纯Swift类，可能只能通过组合模式替代扩展。
• 清理闭包里要注意避免循环引用，建议用[weak self]捕获弱引用。

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