如何在一个集合视图中实现不同的集合视图单元格类型？

import UIKit

class ViewController: UIViewController, UICollectionViewDataSource, UICollectionViewDelegateFlowLayout {
    
    let collectionView: UICollectionView = {
        let layout = UICollectionViewFlowLayout()
        let collectionView = UICollectionView(frame: .zero, collectionViewLayout: layout)
        collectionView.translatesAutoresizingMaskIntoConstraints = false
        collectionView.backgroundColor = .white
        return collectionView
    }()
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        // 设置集合视图的数据源和代理
        collectionView.dataSource = self
        collectionView.delegate = self
        
        // 注册集合视图单元格类型
        collectionView.register(TextCell.self, forCellWithReuseIdentifier: "TextCell")
        collectionView.register(ImageCell.self, forCellWithReuseIdentifier: "ImageCell")
        
        // 添加集合视图到视图层级中
        view.addSubview(collectionView)
        
        // 设置集合视图的约束
        NSLayoutConstraint.activate([
            collectionView.topAnchor.constraint(equalTo: view.topAnchor),
            collectionView.leadingAnchor.constraint(equalTo: view.leadingAnchor),
            collectionView.trailingAnchor.constraint(equalTo: view.trailingAnchor),
            collectionView.bottomAnchor.constraint(equalTo: view.bottomAnchor)
        ])
    }
    
    // MARK: - UICollectionViewDataSource
    
    func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
        return 10 // 假设有10个数据项
    }
    
    func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
        let data = getData(for: indexPath) // 获取对应索引路径的数据
        
        if data.type == .text {
            let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "TextCell", for: indexPath) as! TextCell
            cell.configure(with: data.text)
            return cell
        } else if data.type == .image {
            let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "ImageCell", for: indexPath) as! ImageCell
            cell.configure(with: data.image)
            return cell
        }
        
        return UICollectionViewCell()
    }
    
    // MARK: - UICollectionViewDelegateFlowLayout
    
    func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> CGSize {
        let data = getData(for: indexPath) // 获取对应索引路径的数据
        
        if data.type == .text {
            return CGSize(width: collectionView.bounds.width, height: 50)
        } else if data.type == .image {
            return CGSize(width: collectionView.bounds.width, height: 200)
        }
        
        return CGSize.zero
    }
    
    // MARK: - Helper Methods
    
    func getData(for indexPath: IndexPath) -> Data {
        // 根据索引路径返回相应的数据
        // 这里假设有一个名为Data的自定义结构体，包含type、text和image属性
        // 根据数据的类型来决定使用哪种集合视图单元格类型
        // 可以根据实际需求进行修改
        return Data(type: .text, text: "Text", image: nil)
    }
}

// 自定义集合视图单元格类型

class TextCell: UICollectionViewCell {
    
    let label: UILabel = {
        let label = UILabel()
        label.translatesAutoresizingMaskIntoConstraints = false
        return label
    }()
    
    override init(frame: CGRect) {
        super.init(frame: frame)
        
        contentView.addSubview(label)
        
        NSLayoutConstraint.activate([
            label.topAnchor.constraint(equalTo: contentView.topAnchor),
            label.leadingAnchor.constraint(equalTo: contentView.leadingAnchor),
            label.trailingAnchor.constraint(equalTo: contentView.trailingAnchor),
            label.bottomAnchor.constraint(equalTo: contentView.bottomAnchor)
        ])
    }
    
    required init?(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    
    func configure(with text: String) {
        label.text = text
    }
}

class ImageCell: UICollectionViewCell {
    
    let imageView: UIImageView = {
        let imageView = UIImageView()
        imageView.translatesAutoresizingMaskIntoConstraints = false
        return imageView
    }()
    
    override init(frame: CGRect) {
        super.init(frame: frame)
        
        contentView.addSubview(imageView)
        
        NSLayoutConstraint.activate([
            imageView.topAnchor.constraint(equalTo: contentView.topAnchor),
            imageView.leadingAnchor.constraint(equalTo: contentView.leadingAnchor),
            imageView.trailingAnchor.constraint(equalTo: contentView.trailingAnchor),
            imageView.bottomAnchor.constraint(equalTo: contentView.bottomAnchor)
        ])
    }
    
    required init?(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    
    func configure(with image: UIImage?) {
        imageView.image = image
    }
}

// 自定义数据类型

struct Data {
    enum DataType {
        case text
        case image
    }
    
    let type: DataType
    let text: String?
    let image: UIImage?
}