MulticastDelegate.swift

class WeakObject<T: AnyObject>: Hashable {
    private(set) weak var object: T?
    let identifier: ObjectIdentifier

    static func == (lhs: WeakObject<T>, rhs: WeakObject<T>) -> Bool {
        return lhs.identifier == rhs.identifier
    }

    func hash(into hasher: inout Hasher) {
        hasher.combine(identifier)
    }

    init(object: T) {
        self.object = object
        self.identifier = ObjectIdentifier(object)
    }
}

class WeakDelegateCollection<T> {
    typealias Canceller = () -> Void
    private var objects: Set<WeakObject<AnyObject>> = []

    @discardableResult
    func add(_ delegate: T) -> Canceller {
        //WeakObject chỉ hoạt động với reference type
        guard type(of: delegate as Any) is AnyClass else {
            fatalError("Class misused: WeakDelegateCollection should only be use with class")
        }

        //Chủ động cast delegate to be AnyObject bởi swift đang nhận diện delegate thuộc type T - vừa có thể là class vừa có thể là struct
        //Sau khi cast ta có thể dùng [weak delegate] để tránh retain cycle trong Canceller
        let delegate = delegate as AnyObject
        let object = WeakObject(object: delegate)
        objects.insert(object)

        //Sau khi gọi add có thể lưu lại Canceller để chủ động huỷ đăng ký
        return { [weak self, weak delegate] in
            guard let self = self, let delegate = delegate as? T else { return }
            self.remove(delegate)
        }
    }

    func trigger(_ action: (T) -> Void) {
        for weakObject in objects {
            //Nếu object trong WeakObject chưa deinit => chạy closure
            if let object = weakObject.object as? T {
                action(object)
            } else {
                //Xoá WeakObject có object đã được deinit
                objects.remove(weakObject)
            }
        }
    }

    func remove(_ delegate: T) {
        guard type(of: delegate as Any) is AnyClass else { return }
        objects.remove(WeakObject(object: delegate as AnyObject))
    }
}