如何区分自定义单元格中的两个textFields？

override init(style: UITableViewCell.CellStyle, reuseIdentifier: String?) {
    super.init(style: style, reuseIdentifier: reuseIdentifier)

    let firstTextField = UITextField()
    firstTextField.tag = 1
    // 设置其他属性和约束

    let secondTextField = UITextField()
    secondTextField.tag = 2
    // 设置其他属性和约束

    addSubview(firstTextField)
    addSubview(secondTextField)
}

func textFieldDidEndEditing(_ textField: UITextField) {
    if textField.tag == 1 {
        // 第一个文本字段
    } else if textField.tag == 2 {
        // 第二个文本字段
    }
}

protocol CustomCellDelegate: class {
    func firstTextFieldDidEndEditing(cell: CustomCell)
    func secondTextFieldDidEndEditing(cell: CustomCell)
}

func textFieldDidEndEditing(_ textField: UITextField) {
    if textField == firstTextField {
        delegate?.firstTextFieldDidEndEditing(cell: self)
    } else if textField == secondTextField {
        delegate?.secondTextFieldDidEndEditing(cell: self)
    }
}

func firstTextFieldDidEndEditing(cell: CustomCell) {
    // 第一个文本字段
}

func secondTextFieldDidEndEditing(cell: CustomCell) {
    // 第二个文本字段
}

var firstTextFieldDidEndEditing: (() -> Void)?
var secondTextFieldDidEndEditing: (() -> Void)?

func textFieldDidEndEditing(_ textField: UITextField) {
    if textField == firstTextField {
        firstTextFieldDidEndEditing?()
    } else if textField == secondTextField {
        secondTextFieldDidEndEditing?()
    }
}

customCell.firstTextFieldDidEndEditing = {
    // 第一个文本字段
}

customCell.secondTextFieldDidEndEditing = {
    // 第二个文本字段
}