如何比較兩個枚舉實例與NO原始類型？

Question

考慮：

enum Example { case Step1 case Step2(data: String) } 

和：

let a: Example = .Step1 
let b: Example = .Step2(data: "hi") 

我如何使這項工作？

print(a == b) // ERROR: Binary operator '==' cannot be applied to two 'Example' operands 

請注意，我不能放棄自定義枚舉（它不能包含原始值）

Answer 1

落實Equatable協議的枚舉。

enum Example: Equatable { 
    case Step1 
    case Step2(data: String) 

    static func == (lhs: Example, rhs: Example) -> Bool { 
     switch(lhs) { 
     case Step1: 
      switch(rhs) { 
      case Step1: 
       return true 
      default: 
       return false 
      } 
     case Step2(data: leftString): 
      switch(rhs) { 
      case Step2(data: rightString): 
       return leftString == rightString 
      default: 
       return false 
      } 
     } 
    }  
}