自定义UITableViewCell的动态高度问题

class CustomTableViewCell: UITableViewCell {
    // 自定义的子视图
    var contentLabel: UILabel!
    
    override init(style: UITableViewCell.CellStyle, reuseIdentifier: String?) {
        super.init(style: style, reuseIdentifier: reuseIdentifier)
        
        // 初始化子视图
        contentLabel = UILabel(frame: CGRect(x: 10, y: 10, width: contentView.frame.width - 20, height: 0))
        contentLabel.numberOfLines = 0
        contentView.addSubview(contentLabel)
    }
    
    override func layoutSubviews() {
        super.layoutSubviews()
        
        // 根据内容调整子视图的布局
        contentLabel.sizeToFit()
        contentLabel.frame.size.width = contentView.frame.width - 20
    }
}

class ViewController: UIViewController, UITableViewDelegate, UITableViewDataSource {
    // 数据源
    var data: [String] = ["这是一段测试文本", "这是另一段测试文本", "这是一段很长很长的测试文本，用来测试UITableViewCell的动态高度问题。"]
    
    // UITableView
    var tableView: UITableView!
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        // 初始化UITableView
        tableView = UITableView(frame: view.bounds, style: .plain)
        tableView.delegate = self
        tableView.dataSource = self
        tableView.register(CustomTableViewCell.self, forCellReuseIdentifier: "CustomTableViewCell")
        view.addSubview(tableView)
    }
    
    // UITableViewDelegate和UITableViewDataSource的实现
    func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
        return data.count
    }
    
    func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
        let cell = tableView.dequeueReusableCell(withIdentifier: "CustomTableViewCell", for: indexPath) as! CustomTableViewCell
        cell.contentLabel.text = data[indexPath.row]
        return cell
    }
    
    func tableView(_ tableView: UITableView, heightForRowAt indexPath: IndexPath) -> CGFloat {
        let cell = tableView.dequeueReusableCell(withIdentifier: "CustomTableViewCell") as! CustomTableViewCell
        cell.contentLabel.text = data[indexPath.row]
        cell.setNeedsLayout()
        cell.layoutIfNeeded()
        let height = cell.contentView.systemLayoutSizeFitting(UIView.layoutFittingCompressedSize).height
        return height
    }
}