如何制作UICollectionView动态高度？

import UIKit

class CustomFlowLayout: UICollectionViewFlowLayout {
    override func prepare() {
        super.prepare()
        // 进行一些初始化操作
    }
    
    override func layoutAttributesForElements(in rect: CGRect) -> [UICollectionViewLayoutAttributes]? {
        let attributesArray = super.layoutAttributesForElements(in: rect)
        
        // 计算每个item的布局属性，并保存在布局属性数组中
        
        return attributesArray
    }
    
    override func layoutAttributesForItem(at indexPath: IndexPath) -> UICollectionViewLayoutAttributes? {
        let attributes = super.layoutAttributesForItem(at: indexPath)
        
        // 计算指定位置的item的布局属性
        
        return attributes
    }
}

class ViewController: UIViewController, UICollectionViewDelegate, UICollectionViewDataSource, UICollectionViewDelegateFlowLayout {
    @IBOutlet weak var collectionView: UICollectionView!
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        // 设置UICollectionView的布局对象为自定义的CustomFlowLayout
        let layout = CustomFlowLayout()
        layout.scrollDirection = .vertical
        collectionView.collectionViewLayout = layout
        
        // 设置UICollectionView的delegate和dataSource
        collectionView.delegate = self
        collectionView.dataSource = self
    }
    
    // 实现UICollectionViewDelegateFlowLayout代理方法，返回每个item的大小
    func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> CGSize {
        // 返回每个item的大小，包括动态计算的高度
        return CGSize(width: collectionView.bounds.width, height: calculateItemHeight(at: indexPath))
    }
    
    // 实现UICollectionViewDataSource代理方法，返回item的数量
    func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
        // 返回item的数量
        return 10
    }
    
    // 实现UICollectionViewDataSource代理方法，返回每个item的内容
    func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
        let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "CustomCell", for: indexPath) as! CustomCell
        
        // 设置每个item的内容
        
        return cell
    }
    
    // 根据实际需求计算每个item的高度
    func calculateItemHeight(at indexPath: IndexPath) -> CGFloat {
        // 根据indexPath或外部数据源计算每个item的高度
        
        return 100.0 // 示例中使用固定高度100.0
    }
}

class CustomCell: UICollectionViewCell {
    // 自定义的UICollectionViewCell，用于展示每个item的内容
}