Question

我有两个班级成绩和老师。

class Grade: NSObject {

    var grade:String = ""
    var id:String = ""

    init(from dictionary:Parameters) {

        self.grade = dictionary["grade"] as? String ?? ""
        self.id = dictionary["grade_id"] as? String ?? ""
    }
}

class Teacher: NSObject {

    var teacher:String = ""
    var id:String = ""

    init(from dictionary:Parameters) {

        self.teacher = dictionary["teacher"] as? String ?? ""
        self.id = dictionary["teacher_id"] as? String ?? ""
    }
}

因此，我想创建一个将包含id和title的通用类。它可以从两个类中获取数据。

我正在使用以下方法，但我认为这不是实现的最佳方法。让我知道什么是最好的方法。

   func prepareDataSourceForPicker(pickerType:PickerType) {

        var pickerArr = [PickerItem]()
        if pickerType == .grade {

            for value in self.arrGrades {
                let pickerItem:PickerItem = PickerItem()
                pickerItem.title = value.grade
                pickerItem.id = value.id
                pickerArr.append(pickerItem)
            }

        } else if pickerType == .teacher {
            for value in self.arrTeachers {
                let pickerItem:PickerItem = PickerItem()
                pickerItem.title = value.teacher
                pickerItem.id = value.id
                pickerArr.append(pickerItem)
            }
        }
    }

Answer 1

创建一个名为PickerItemProtocol的协议：

protocol PickerItemProtocol {
    var title: String { get }
    var id: String { get }
}

然后使Grade和Teacher符合它：

extension Grade: PickerItemProtocol {
    var title: String { return grade }
}

extension Teacher : PickerItemProtocol {
    var title: String { return teacher }
}

现在在prepareDataSourceForPicker中，您可以执行以下操作：

func prepareDataSourceForPicker(pickerType:PickerType) {

    let dataSource: [PickerItemProtocol]
    switch pickerType {
        case .grade: dataSource = self.arrGrades
        case .teacher: dataSource = self.arrTeachers 
    }
    let pickerArr = dataSource.map { item in
        let pickerItem = PickerItem()
        pickerItem.title = item.title
        pickerItem.id = item.id
        return pickerItem
    }
    // you might want to do something with pickerArr here...
}

Answer 2

我认为最好的方法是在OOP的最基本概念中，即继承。我的意思是：

创建具有基本属性（每个子类中需要的公共属性）的BaseModel类。您的情况是类型，ID和标题。
然后将教师和年级设置为该类的子类，并将剩余的属性添加到每个班级（教师和年级）。
然后，您将在教师和年级模型中都拥有类型，ID和标题。

如何将一种类型的对象数组转换为另一种？

2 个答案: