Swift枚举的一些小用法总结

时间:2022-02-16 15:09:26

前言

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