swift - 在对象 deinit 的扩展中调用代码？

Question

在 Swift（我使用的是 4.1）中，当对象被破坏时，有没有办法在扩展中进行一些清理？我想到了会进入的那种代码deinit()，但是扩展不能声明deinit()。（除此之外，如果需要多个扩展来执行此操作，则会有多个deinit()声明。）

Answer 1

我没有找到一种方法来准确地得到你想要的，但也许这段代码会有所帮助。我从未尝试过，所以也许更多地使用它作为灵感。简而言之，它允许您添加将在取消初始化时执行的代码位。

/// This is a simple object whose job is to execute
/// some closure when it deinitializes
class DeinitializationObserver {

    let execute: () -> ()

    init(execute: @escaping () -> ()) {
        self.execute = execute
    }

    deinit {
        execute()
    }
}

/// We're using objc associated objects to have this `DeinitializationObserver`
/// stored inside the protocol extension
private struct AssociatedKeys {
    static var DeinitializationObserver = "DeinitializationObserver"
}

/// Protocol for any object that implements this logic
protocol ObservableDeinitialization: AnyObject {

    func onDeinit(_ execute: @escaping () -> ())

}

extension ObservableDeinitialization {

    /// This stores the `DeinitializationObserver`. It's fileprivate so you
    /// cannot interfere with this outside. Also we're using a strong retain
    /// which will ensure that the `DeinitializationObserver` is deinitialized
    /// at the same time as your object.
    fileprivate var deinitializationObserver: DeinitializationObserver {
        get {
            return objc_getAssociatedObject(self, &AssociatedKeys.DeinitializationObserver) as! DeinitializationObserver
        }
        set {
            objc_setAssociatedObject(
                self,
                &AssociatedKeys.DeinitializationObserver,
                newValue,
                objc_AssociationPolicy.OBJC_ASSOCIATION_RETAIN_NONATOMIC
            )    
        }
    }

    /// This is what you call to add a block that should execute on `deinit`
    func onDeinit(_ execute: @escaping () -> ()) {
        deinitializationObserver = DeinitializationObserver(execute: execute)
    }
}

---

这个怎么运作

现在让我们假设你有你的对象，让我们调用它A并且你想在扩展中创建一些代码，你可以onDeinit像这样添加你的清理代码：

extension A {

    func someSetup() {

        // Do your thing...


        onDeinit {
            /// This will be executed when A is deinitialized
            print("Execute this")
        }
    }
}

您也可以从扩展外部添加它，

let a = A()
a.onDeinit {
    print("Deinitialized")
}

---

讨论

• 这与您想要的不同，因为deinit您需要调用一个函数（而不是定义一个函数）。但无论如何，在协议扩展中，您永远无法真正使用底层对象的生命周期，因此在大多数情况下应该没问题。
• 我不确定A'sdeinit和 DeinitializationObserver's之间的执行顺序deinit。A当您在闭包内编写代码时，您可能需要放弃仍在内存中的假设。
• 如果您需要多个，onDeinit您可以更新关联对象以保留一个数组DeinitializationObserver
• 在我的代码中，我通常使用 ReactiveCocoa 来Lifetime处理这类事情。但是，它比我在这里写的要复杂。相反，看起来他们混合了dealloc选择器。如果你有兴趣可以看看里面 - https://github.com/ReactiveCocoa/ReactiveCocoa/blob/master/ReactiveCocoa/NSObject+Lifetime.swift