更新UICollectionViewCell中的高度约束

class CustomCollectionViewCell: UICollectionViewCell {
    @IBOutlet weak var heightConstraint: NSLayoutConstraint!
    
    override func preferredLayoutAttributesFitting(_ layoutAttributes: UICollectionViewLayoutAttributes) -> UICollectionViewLayoutAttributes {
        let newAttributes = super.preferredLayoutAttributesFitting(layoutAttributes)
        
        // 更新高度约束
        heightConstraint.constant = 100 // 设置新的高度
        
        // 返回新的布局属性
        return newAttributes
    }
}

class CustomCollectionViewFlowLayout: UICollectionViewFlowLayout {
    override init() {
        super.init()
        
        // 设置estimatedItemSize为自动大小
        estimatedItemSize = UICollectionViewFlowLayoutAutomaticSize
    }
    
    required init?(coder aDecoder: NSCoder) {
        super.init(coder: aDecoder)
        
        // 设置estimatedItemSize为自动大小
        estimatedItemSize = UICollectionViewFlowLayoutAutomaticSize
    }
}

// 在集合视图的数据源方法中返回更新后的布局属性
func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> CGSize {
    let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "CustomCell", for: indexPath) as! CustomCollectionViewCell
    
    let attributes = UICollectionViewLayoutAttributes(forCellWith: indexPath)
    attributes.frame = CGRect(x: 0, y: 0, width: collectionView.bounds.width, height: 100) // 设置初始高度
    
    let newAttributes = cell.preferredLayoutAttributesFitting(attributes)
    
    return newAttributes.frame.size
}