前言
在 Swift 中,枚举是一个非常方便也非常强大的类型。我们在日常使用中也经常会使用到它。
例如,我们最常见的 optional:
1
2
3
4
|
enum Optional<T> {
case Some(T)
case None
}
|
这里不准备介绍枚举的基本用法,只是记录两个比较好用的枚举用法。
关联值
关联值是将额外信息附加到 enum case 中的一种极好的方式。
例如,当我们需要将一系列的值传到下一个类中时,一般情况下我们像下方代码一样写出几个设置的方法:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
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?) { }
}
|
这样当需要传的值变多时,代码无疑就会变得没那么好看了。我们可以用枚举来简化:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
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 时,一般就只用了几种基本的类型:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
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
}
|
但是,如果我们需要在枚举类型放入我们自定义的类型的话,我们就需要为枚举加一些东西了。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
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 协议,并实现协议的一些属性及方法:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
/*
将枚举的 RawValue 关联为自己希望的类型
**/
associatedtype RawValue
/*
利用自己关联的类型生成枚举的实例
**/
init?(rawValue: Self.RawValue)
/*
将自己定义的类型的作为 RawValue 返回
**/
var rawValue: Self.RawValue { get }
|
总结
以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,谢谢大家对服务器之家的支持。
原文链接:https://juejin.im/post/5d1abc646fb9a07ef81a15ab