iOS Swift中动态修改UITableView自定义分区头部圆角问题

我能理解你遇到的困扰——明明在layoutSubviews里改了圆角，结果完全没生效。这种情况通常是几个细节没处理到位，咱们一步步来排查修复：

1. 先确认你的UIView扩展方法是否正确实现 #

很多时候问题出在圆角设置的核心逻辑上，指定圆角需要用CAShapeLayer做遮罩，而不是直接改layer.cornerRadius。下面是正确的setRoundCorners实现：

extension UIView {
    func setRoundCorners(cornerRadius: CGFloat, corners: UIRectCorner) {
        let path = UIBezierPath(roundedRect: self.bounds,
                                byRoundingCorners: corners,
                                cornerRadii: CGSize(width: cornerRadius, height: cornerRadius))
        let maskLayer = CAShapeLayer()
        maskLayer.path = path.cgPath
        self.layer.mask = maskLayer
        // 别忘了开启裁剪，否则圆角不会生效
        self.clipsToBounds = true
    }
}

2. 给自定义HeaderFooter添加状态标记并触发重布局 #

在你的自定义UITableViewHeaderFooterView子类里，添加一个布尔属性来标记当前分区是否有单元格，并且在属性变化时强制触发布局更新：

class CustomSectionHeader: UITableViewHeaderFooterView {
    // 标记当前分区是否包含单元格
    var hasCells: Bool = false {
        didSet {
            // 属性变化时通知系统重新布局
            setNeedsLayout()
        }
    }
    
    override func layoutSubviews() {
        super.layoutSubviews()
        // 这里必须先调用super，确保view的bounds已经正确计算
        
        let targetCorners: UIRectCorner
        if hasCells {
            // 有单元格时，只保留左上、右上圆角
            targetCorners = [.topLeft, .topRight]
        } else {
            // 无单元格时，四个角都设圆角
            targetCorners = [.topLeft, .topRight, .bottomLeft, .bottomRight]
        }
        self.setRoundCorners(cornerRadius: 5, corners: targetCorners)
    }
    
    // 其他自定义代码...
}

3. 在TableView数据源中更新Header的状态 #

你需要在TableView的数据源方法里，判断对应分区是否有单元格，然后给Header设置hasCells属性：

func tableView(_ tableView: UITableView, viewForHeaderInSection section: Int) -> UIView? {
    let header = tableView.dequeueReusableHeaderFooterView(withIdentifier: "CustomSectionHeader") as! CustomSectionHeader
    // 判断当前分区的单元格数量
    let rowCount = tableView.numberOfRows(inSection: section)
    header.hasCells = rowCount > 0
    return header
}

4. 排查可能的额外问题 #

• 确保Header的frame正确：如果你的Header用了AutoLayout，一定要保证在layoutSubviews调用时，bounds已经是最终的正确尺寸——因为我们的遮罩路径是基于bounds创建的。
• 避免重复设置mask：如果你的其他代码里也修改了layer.mask，会覆盖这里的设置，要检查有没有冲突的操作。
• TableView的SectionHeader高度：如果Header高度设置不正确，也可能导致圆角显示异常，确保高度是你预期的值。

这样调整后，当分区添加/移除单元格时，只要刷新对应的Section（比如调用tableView.reloadSections([section], with: .automatic)），Header就会自动更新圆角状态了。

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