Swift Enum 关联值嵌套的一些实践

前言

Swift 中的枚举很强大，算是一等公民。可以定义函数，也可以遵守协议、实现 extension 等等。

关联值也是 Swift 枚举的一大特性。基本用法如下：

enum RequestResult {
    case success
    case failure(Error)
}

let result = RequestResult.failure(URLError(URLError.timedOut))
switch result {
case .success:
    print("请求成功")
case .failure(let error):
    print(error)
}

1、在需要关联值的 case 中声明关联值的类型。

2、在 switch 的 case 中声明一个常量或者变量来接收。

遇到的问题

一般情况下，上述的代码是清晰明了的。但在实际开发的过程中，遇到了以下的情况：关联值的类型也是枚举，而且嵌套不止一层。

比如下面的代码：

enum EnumT1 {
    case test1(EnumT2)
    case other
}

enum EnumT2 {
    case test2(EnumT3)
    case other2
}

enum EnumT3 {
    case test3(EnumT4)
    case test4
}

根据我们的需求，需要进行多次嵌套来进行类型细化。当进行枚举的声明时，代码还是正常的，简单明了。但当进行 case 判断时，代码就变得丑陋难写了。

比如，我只想处理 EnumT3 中的 test4 的情况，在 switch 中我需要进行 switch 的嵌套来处理：

let t1: EnumT1? = .test1(.test2(.test4))
switch t1 {
case .test1(let t2):
    switch t2 {
    case .test2(let t3):
        switch t3 {
        case .test4:
            print("test4")
        case default:
            print("default")
        }
    default:
        print("default")
    }
default:
    print("default")
}

这种写法，对于一个程序员来说是无法忍受的。它存在两个问题：一是代码臃肿，我的本意是只处理某一种情况，但我需要显式的嵌套多层 switch；二是枚举本身是不推荐使用 default 的，官方推荐是显式的写出所有的 case，以防出现难以预料的问题。

废话不多说，下面开始简化之路。

实践一

首先能想到的是，因为是对某一种情况进行处理，考虑使用 if + == 的判断来进行处理，比如下面这种写法：

if t1 == .test1(.test2(.test4)) { }

这样处理有两个不足之处。首先，如果对枚举用 == 操作符的话，需要对每一个枚举都遵守 Equatable 协议，这为我们带来了工作量。其次最重要的是，这种处理方式无法应对 test3 这种带有关联值的情况。

if t1 == .test1(.test2(.test3) { } 

如果这样写的话，编译器会报错，因为 test3 是需要传进去一个 Int 值的。

if t1 == .test1(.test2(.test3(20))) { }

如果这样写的话也不行，因为我们的需求是处理 test3 的统一情况（所有的关联值），而不是某一个具体的关联值。

实践二

经过在网上的一番搜查，发现可以用 if-case 关键字来简化写法：

if case .test1(.test2(.test3)) = t1 { }

这样就能统一处理 test3 这个 case 的所有情况了。如果想获取关联值，可以用下面的写法：

if case .test1(.test2(.test3(let i))) = t1 {
    print(i)
}

对比上面的 switch 写法，可以看到，下面的这种写法既易懂又好写😁。

总结来说，当我们遇到关联值多层枚举嵌套的时候，又需要对某一种情况进行处理。那么可以采用实践二的做法来进行代码简化。

参考链接