如何在类层次结构中正确实现Equatable协议?

时间:2016-10-07 05:08:11

标签: swift subclass swift3 equatable

我试图在基类中实现==运算符(来自Equatable)及其在Swift 3中的子类。所有类只会在Swift中使用所以我这样做不想涉及NSObjectNSCopying协议。

我从一个基类和一个子类开始:

class Base {
    var x : Int
}

class Subclass : Base {
    var y : String
}

现在我想将Equatable==运算符添加到Base。看起来很简单。从文档中复制==运算符签名:

class Base : Equatable {
    var x : Int

    static func == (lhs: Base, rhs: Base) -> Bool {
        return lhs.x == rhs.x
    }
}

到目前为止一切顺利。现在为子类:

class Subclass : Base {
    static override func == (lhs: Base, rhs: Base) -> Bool {
        return true
    }
}

但这会导致错误:

  

操作员功能会覆盖最后一个'操作员功能

行。经过一些研究(仍在学习Swift 3)后,我了解到static可以替换为class,以表明可以覆盖类型方法。

所以我尝试在static中将class更改为Base

class Base : Equatable {
    var x : Int

    class func == (lhs: Base, rhs: Base) -> Bool {
        return lhs.x == rhs.x
    }
}

但这会导致新的错误:

  

运营商' =='在非决赛阶段宣布' Base'必须是最后的

唉。这远比应该复杂得多。

如何在基类和子类中正确实现Equatable协议和==运算符?

4 个答案:

答案 0 :(得分:12)

经过大量的研究和一些反复试验,我终于想出了一个有效的解决方案。第一步是将==运算符从类内部移动到全局范围。这修复了staticfinal的错误。

对于基类,这变为:

func == (lhs: Base, rhs: Base) -> Bool {
    return lhs.x == rhs.x
}

class Base : Equatable {
    var x : Int
}

对于子类:

func == (lhs: Subclass, rhs: Subclass) -> Bool {
    return true
}

class Subclass : Base {
    var y : String
}

现在剩下的唯一部分是弄清楚如何从子类的==运算符调用基类的==运算符。这导致我找到了最终的解决方案:

func == (lhs: Subclass, rhs: Subclass) -> Bool {
    if lhs.y == rhs.y {
        if lhs as Base == rhs as Base {
            return true
        }
    }

    return false
}

第一个if语句导致调用基类中的==运算符。

最终解决方案:

Base.swift:

func == (lhs: Base, rhs: Base) -> Bool {
    return lhs.x == rhs.x
}

class Base : Equatable {
    var x : Int
}

Subclass.swift:

func == (lhs: Subclass, rhs: Subclass) -> Bool {
    if lhs.y == rhs.y {
        if lhs as Base == rhs as Base {
            return true
        }
    }

    return false
}

class Subclass : Base {
    var y : String
}

答案 1 :(得分:1)

我知道问题发布已经有一段时间了,但我希望我的答案有所帮助。

TLDR - 您提供自定义比较方法,而不是尝试覆盖==,请==调用它,并在需要时覆盖自定义比较方法。

所以你说

  

所有类只会在Swift中使用,所以我不想涉及NSObjectNSCopying协议。

但如果你 继承NSObject,你将如何编写自定义比较方法?你会覆盖isEqual(Any?),对吗?如果您尝试在子类中符合Equatable协议,编译器会抱怨“冗余符合协议Equatable”,因为NSObject已经符合Equatable

现在,我们提供了一些关于NSObject如何处理此问题的提示 - 它提供了一个自定义比较方法isEqual(Any?),在==内调用它,如果需要,它的子类可以覆盖它。您可以在自己的基类中执行相同的操作。

不用多说,让我们做一些实验(在Swift 4中)。定义一些类

class Grandpa: Equatable {
    var x = 0

    static func ==(lhs: Grandpa, rhs: Grandpa) -> Bool {
        return lhs.isEqual(to: rhs)
    }

    func isEqual(to object: Any?) -> Bool {
        guard object != nil && type(of: object!) == Grandpa.self else {
            return false
        }
        let value = object as! Grandpa
        return x == value.x
    }
}

class Father: Grandpa {
    var y = 0

    override func isEqual(to object: Any?) -> Bool {
        guard object != nil && type(of: object!) == Father.self else {
            return false
        }
        let value = object as! Father
        return x == value.x && y == value.y
    }
}

class Son: Father {
    var z = 0

    override func isEqual(to object: Any?) -> Bool {
        guard object != nil && type(of: object!) == Son.self else {
            return false
        }
        let value = object as! Son
        return x == value.x && y == value.y && z == value.z
    }
}

并编写一些测试代码

let grandpa1 = Grandpa()
let grandpa2 = Grandpa()
let grandpa3: Grandpa? = nil
let grandpa4: Grandpa? = nil
let father1 = Father()
let father2 = Father()
let father3 = Father()
father3.y = 1
let son1 = Son()
let son2 = Son()
let son3 = Son()
son3.z = 1

print("grandpa1 == grandpa2: \(grandpa1 == grandpa2)")
print("grandpa1 == grandpa3: \(grandpa1 == grandpa3)")
print("grandpa3 == grandpa4: \(grandpa3 == grandpa4)")
print("grandpa1 == father1: \(grandpa1 == father1)")
print("father1 == father2: \(father1 == father2)")
print("father1 == father3: \(father1 == father3)")
print("son1 == son2: \(son1 == son2)")
print("son1 == son3: \(son1 == son3)")

运行它,你应该

grandpa1 == grandpa2: true
grandpa1 == grandpa3: false
grandpa3 == grandpa4: true
grandpa1 == father1: false
father1 == father2: true
father1 == father3: false
son1 == son2: true
son1 == son3: false

答案 2 :(得分:-1)

rmaddy's answer之后,我想出了一种用于测试平等的守卫方法:

Base.swift

static func ==(lhs: Base, rhs: Base) -> Bool {
    // ensure class properties match
    guard lhs.x == rhs.x else {
        return false
    }

    return true

}

Subclass.swift

static func ==(lhs: Subclass, rhs: Subclass) -> Bool {
    // ensure base class properties match
    guard lhs as Base == rhs as Base else {
        return false
    }

    // ensure class properties match
    guard lhs.y == rhs.y else {
        return false
    }

    return true
}

```

答案 3 :(得分:-1)

在其他答案之后,我想到了这一点:

class Base : Equatable {
    var x : Int
    static func == (lhs: Base, rhs: Base) -> Bool {
        return lhs.x == rhs.x
    }
}

class Subclass : Base {
    var y : String
    static func == (lhs: Subclass, rhs: Subclass) -> Bool {
        return lhs.y == rhs.y && (lhs as Base) == (rhs as Base)
    }
}