swift - 检查继承类和 super 类

Question

我有一个类:

class User: NSObject {
    var name = "User_Name"
    // Other variables and methods
}

还有一个继承自“user”的第二个类:

class Employee: User {
    var job = "Manager"
    // other variables and methods
}

我希望能够在函数中检查两者之间的情况，并对它们采取稍微不同的操作。

我相信我可以在使用之前使用保护语句将“AnyObject”参数转换为可疑类型，但不知道如何在两个不同的类类型之间进行检查。例如，我相信我可以对一种类型执行以下操作:

func addUser(user : AnyObject) {
    guard let user = user as? User else {return}
        // do something with 'user' which is known to be of type 'User'
    }

但无法弄清楚如何检查发送的参数是否是 Employee(如果它不是“User”类型)。如果防护语句的评估结果为 true，我想要执行非常相似的操作，无论它是用户还是员工，因此我认为必须有一种方法可以在不重复代码的情况下执行此操作。

有人可以指出我在两种类型之间进行测试的正确方向吗？

Answer 1

使用 AnyObject 很少是正确的方法。通常，当您希望根据输入参数的类型特定实现同一函数时，您会希望实现通用函数。但是，由于您只需要函数处理具有相同父类(super class)的对象，因此您可以简单地使用父类(super class)作为输入参数类型，然后使用可选的转换来查看参数实际上是否属于更具体的类型。

func addUser(user: User) {
    if let employee = user as? Employee {
        print(employee.job)
    } else {
        print(user.name)
    }
}

addUser(user: User()) //prints "User_Name"
addUser(user: Employee()) //prints "Manager"

对于这个简单的情况，不需要泛型函数，特别是因为在这种情况下，泛型函数将采用与非泛型函数几乎相同的形式:

func addUser<T:User>(user: T){
    if let employee = user as? Employee {
        print(employee.job)
    } else {
        print(user.name)
    }
}

addUser(user: User()) //prints "User_Name"
addUser(user: Employee()) //prints "Manager"

或者，如果您想扩展它以适用于不继承自 User 的类:

func addUser<T>(user: T){
    if let user = user as? User {
        if let employee = user as? Employee {
            print(employee.job)
        } else {
            print(user.name)
        }
    } else {
        print("Not a user")
    }
}

addUser(user: User()) //prints "User_Name"
addUser(user: Employee()) //prints "Manager"
addUser(user: "Me") //prints "Not a user"