Swift实现收藏按钮：关联TableView与Core Data及状态切换问题

Hey there! Let's fix that finicky favorite button issue you're dealing with. The problem almost always comes down to not properly syncing the button's UI state with your Core Data object, and forgetting to update both the data and the interface when the button is tapped. Here's a step-by-step solution that should get your toggle working reliably:

First, make sure your Core Data entity has a boolean property to track the favorite state—let's call it isFavorited for this example. This is the single source of truth for whether an item is favorited.

Step 1: Build the Table View Cell with a Bindable Button #

Your cell needs to know which Core Data object it's displaying, and update its button state based on that object's isFavorited value. We'll also add a callback to pass the tap event up to your view controller.

class FavoriteItemCell: UITableViewCell {
    // Favorite button (using SF Symbols for simplicity)
    private let favoriteButton = UIButton(type: .system)
    
    // Weak reference to avoid retain cycles
    weak var currentItem: NSManagedObject?
    var onFavoriteToggle: ((NSManagedObject) -> Void)?
    
    override init(style: UITableViewCell.CellStyle, reuseIdentifier: String?) {
        super.init(style: style, reuseIdentifier: reuseIdentifier)
        setupButton()
    }
    
    required init?(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    
    private func setupButton() {
        favoriteButton.frame = CGRect(x: 0, y: 0, width: 44, height: 44)
        accessoryView = favoriteButton
        favoriteButton.addTarget(self, action: #selector(toggleFavorite), for: .touchUpInside)
    }
    
    // Configure the cell with your Core Data object
    func configure(with item: NSManagedObject) {
        currentItem = item
        // Update button state based on the object's isFavorited property
        let isFavorited = item.value(forKey: "isFavorited") as? Bool ?? false
        let iconName = isFavorited ? "star.fill" : "star"
        favoriteButton.setImage(UIImage(systemName: iconName), for: .normal)
        favoriteButton.tintColor = isFavorited ? .systemYellow : .systemGray
    }
    
    @objc private func toggleFavorite() {
        guard let item = currentItem else { return }
        onFavoriteToggle?(item)
    }
}

Step 2: Handle the Toggle in Your View Controller #

In your view controller, you'll update the Core Data object's state, save the context, and refresh the cell's UI immediately to reflect the change.

class ItemsViewController: UIViewController, UITableViewDataSource {
    @IBOutlet weak var tableView: UITableView!
    private let context = (UIApplication.shared.delegate as! AppDelegate).persistentContainer.viewContext
    private var items: [NSManagedObject] = []
    
    override func viewDidLoad() {
        super.viewDidLoad()
        tableView.dataSource = self
        tableView.register(FavoriteItemCell.self, forCellReuseIdentifier: "FavoriteItemCell")
        fetchItemsFromCoreData()
    }
    
    // Fetch items from Core Data to populate your table
    private func fetchItemsFromCoreData() {
        let fetchRequest = NSFetchRequest<NSManagedObject>(entityName: "YourEntityName")
        do {
            items = try context.fetch(fetchRequest)
            tableView.reloadData()
        } catch {
            print("Failed to fetch items: \(error.localizedDescription)")
        }
    }
    
    // MARK: - Table View Data Source
    func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
        return items.count
    }
    
    func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
        let cell = tableView.dequeueReusableCell(withIdentifier: "FavoriteItemCell", for: indexPath) as! FavoriteItemCell
        let item = items[indexPath.row]
        
        cell.configure(with: item)
        // Set up the toggle callback
        cell.onFavoriteToggle = { [weak self] tappedItem in
            self?.updateFavoriteState(for: tappedItem)
            // Refresh the specific cell to update the button UI
            tableView.reloadRows(at: [indexPath], with: .fade)
        }
        
        return cell
    }
    
    // Update Core Data and save the change
    private func updateFavoriteState(for item: NSManagedObject) {
        let currentState = item.value(forKey: "isFavorited") as? Bool ?? false
        item.setValue(!currentState, forKey: "isFavorited")
        
        do {
            try context.save()
            // If you have a separate favorites table view, refresh its data here too
        } catch {
            print("Failed to save favorite state: \(error.localizedDescription)")
            // Rollback if save fails to keep UI and data in sync
            context.rollback()
        }
    }
}

Common Pitfalls to Avoid #

• Forgetting to sync initial state: Always set the button's icon/tint based on the Core Data object's isFavorited value when configuring the cell—don't hardcode a default state.
• Not refreshing the UI after save: Calling reloadRows(at:with:) ensures the button updates immediately after the tap, instead of waiting for the next table refresh.
• Retain cycles: Use [weak self] in the cell's callback to prevent your view controller from being held in memory unnecessarily.
• Ignoring save errors: If Core Data fails to save, rollback the context to keep the object's state in sync with what the user sees.

Give this approach a go—your favorite button should toggle smoothly every time, and the state will stay persisted in Core Data even after closing the app.

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