Swift 枚举的几个小用法

在 Swift 中，枚举是一个很是方便也很是强大的类型。咱们在平常使用中也常常会使用到它。

例如，咱们最多见的 optional：

enum Optional<T> {
  case Some(T)
  case None
}
复制代码

这里不许备介绍枚举的基本用法，只是记录两个比较好用的枚举用法。

关联值

关联值是将额外信息附加到 enum case 中的一种极好的方式。

例如，当咱们须要将一系列的值传到下一个类中时，通常状况下咱们像下方代码同样写出几个设置的方法：

struct MyStruct {
    var value: Int

    init(_ value: Int?) {
        if let val = value {
            self.value = val
        } else {
            self.value = Int(INT_MAX)
        }
    }
}

class Two {
    var value1: String?
    var value2: Int?
    var value3: MyStruct?

    func setValue1(value: String?) { }
    func setValue2(value: Int?) { }
    func setValue2(value: MyStruct?) { }
}
复制代码

这样当须要传的值变多时，代码无疑就会变得没那么好看了。咱们能够用枚举来简化：

enum ValueBind {
    case bindStringValue(str: String)
    case bindIntValue(num: Int)
    case bindModel(model: MyStruct)
}

class Two {
    var value1: String?
    var value2: Int?
    var value3: MyStruct?

    func setValueBind(value: ValueBind) {
        switch value {
        case .bindStringValue(let str):
            print(str)
        case .bindModel(let model):
            print(model.value)
        case .bindIntValue(let num)
            print(num)
        }
    }
}
复制代码

利用枚举关联值以后，我们的代码立刻就简洁了很多。

自定义枚举类型

日常咱们使用枚举时，咱们在为枚举定义 value 时，通常就只用了几种基本的类型：

enum Direction {
    case left
    case top
    case right
    case bottom
}

enum StringEnum: String {
    case hello = "hello"
    case world = "world"
}

enum IntEnum: Int {
    case one = 1
    case two = 2
}
复制代码

可是，若是咱们须要在枚举类型放入咱们自定义的类型的话，咱们就须要为枚举加一些东西了。

enum CustomEnum: RawRepresentable {
    typealias RawValue = MyStruct

    case null
    case one
    case two
    
    init?(rawValue: MyStruct) {
        switch rawValue.value {
        case 1:
            self = .one
        case 2:
            self = .two
        default:
            self = .null
        }
    }

    var rawValue: MyStruct {
        switch self {
        case .one:
            return MyStruct(1)
        case .two:
            return MyStruct(2)
        default:
            return MyStruct(nil)
        }
    }
}
复制代码

咱们让枚举遵照 RawRepresentable 协议，并实现协议的一些属性及方法：

/*
    将枚举的 RawValue 关联为本身但愿的类型
**/
associatedtype RawValue

/*
    利用本身关联的类型生成枚举的实例
**/
init?(rawValue: Self.RawValue)

/*
    将本身定义的类型的做为 RawValue 返回
**/
var rawValue: Self.RawValue { get }
复制代码