Сравнение двух перечислимых переменных независимо от их связанных значений - PullRequest
0 голосов
/ 28 января 2019

Рассмотрим это перечисление:

enum DataType {
    case One (data: Int)
    case Two (value: String)
}

Swift имеет сопоставление с шаблоном для сравнения перечисления со связанными значениями, например так:

let var1 = DataType.One(data: 123)
let var2 = DataType.One(data: 456)

if case DataType.One(data: _) = var2 {
    print ("var2 is DataType.One")
}

Как можно было бы сравнить неодна переменная против типа перечисления, но сравнивая тип перечисления с двумя переменными? Я видел массу похожих вопросов, но ни одна не была сосредоточена на случае, когда у вас есть две переменные.

Что я в основном хочу, так это:

if case var1 = var2 {
    print ("var1 is the same enum type as var2")
}

Ответы [ 3 ]

0 голосов
/ 28 января 2019

Обновленный подход:

Я думаю, что нет никакой встроенной поддержки для этого.Но вы можете достичь этого, определив пользовательский оператор (предпочтительно с помощью протокола, но вы можете сделать это и напрямую).Примерно так:

protocol EnumTypeEquatable {
    static func ~=(lhs: Self, rhs: Self) -> Bool
}

extension DataType: EnumTypeEquatable {
    static func ~=(lhs: DataType, rhs: DataType) -> Bool {
        switch (lhs, rhs) {
        case (.one, .one), 
             (.two, .two): 
            return true
        default: 
            return false
        }
    }
}

А затем использовать его как:

let isTypeEqual = DataType.One(value: 1) ~= DataType.One(value: 2)
print (isTypeEqual) // true



Старый подход:

protocol EnumTypeEquatable {
    var enumCaseIdentifier: String { get }
}

extension DataType: EnumTypeEquatable {
    var enumCaseIdentifier: String {
        switch self {
        case .one: return "ONE"
        case .two: return "TWO"
        }
    }
}

func ~=<T>(lhs: T, rhs: T) -> Bool where T: EnumTypeEquatable {
    return lhs.enumCaseIdentifier == rhs.enumCaseIdentifier
}

Старая версия зависит от времени выполнения и может быть снабжена реализацией enumCaseIdentifier по умолчанию в зависимости от String(describing: self), что не рекомендуется.(поскольку String(describing: self) работает с протоколом CustromStringConvertible и может быть изменен)

0 голосов
/ 28 февраля 2019

Это сработало для меня:

enum DataType {
    case one (data: Int)
    case two (value: String)
}

protocol EnumTypeEquatable {
    static func sameType(lhs: Self, rhs: Self) -> Bool
}

extension DataType: EnumTypeEquatable {
    static func sameType(lhs: DataType, rhs: DataType) -> Bool {
        if let caseLhs = Mirror(reflecting: lhs).children.first?.label, let caseRhs = Mirror(reflecting: rhs).children.first?.label {
            return (caseLhs == caseRhs)
        } else { return false }
    }
}

let isTypeEqual = DataType.sameType(lhs: .one(data: 1), rhs: .one(data: 2))
print (isTypeEqual) // true
0 голосов
/ 28 января 2019

Просто подтвердите Equatable, как показано ниже

extension DataType: Equatable {
    static func == (lhs: DataType, rhs: DataType) -> Bool {
        switch (lhs, rhs) {
        case (.One, .Two), (.Two, .One):
            return false
        case (.One, .One), (.Two, .Two):
            return true
        }
    }
}

Если вы не хотите внедрять Equatable, просто переместите содержимое в метод экземпляра:

extension DataType{
    func isSame(_ other: DataType) -> Bool {
        switch (self, other) {
        case (.One, .Two), (.Two, .One):
            return false
        case (.One, .One), (.Two, .Two):
            return true
        }
    }
}

Использование:

let isTypeEqual = DataType.One(value: 1).isSame(DataType.One(value: 2))
print (isTypeEqual) // true
...