Catch Retain Cycles In Table View Subclass

retainCycle3.swift

import UIKit

class TableView: UITableView {
    private var allCells: [WeakBox<UITableViewCell>] = []

    private func appendIfNotPresent(_ cell: UITableViewCell) {
        if allCells.first(where: { $0.unbox === cell }) == nil {
            allCells.append(WeakBox(cell))
        }
    }

    override func dequeueReusableCell(withIdentifier identifier: String, for indexPath: IndexPath) -> UITableViewCell {
        let cell = super.dequeueReusableCell(withIdentifier: identifier, for: indexPath)
        appendIfNotPresent(cell)
        return cell
    }

    deinit {
        allCells.forEach { MemoryChecker.verifyDealloc(object: $0.unbox) }
    }
}