17.1 逗号 ok 模式

在学习本书第二部分和第三部分时,我们经常在一个表达式返回 2 个参数时使用这种模式:something, ok,第一个参数是一个值或者 nil,第二个参数是 true/false 或者一个错误 error。在一个需要赋值的 if 条件语句中,使用这种模式去检测第二个参数值会让代码显得优雅简洁。这种模式在 Go 语言编码规范中非常重要。下面总结了所有使用这种模式的例子:

(1)在函数返回时检测错误(参考第 5.2 小节):

  1. value, err := pack1.Func1(param1)
  2. if err != nil {
  3. fmt.Printf("Error %s in pack1.Func1 with parameter %v", err.Error(), param1)
  4. return err
  5. }
  6. // 函数Func1没有错误:
  7. Process(value)
  8. e.g.: os.Open(file) strconv.Atoi(str)

这段代码中的函数将错误返回给它的调用者,当函数执行成功时,返回的错误是 nil,所以使用这种写法:

  1. func SomeFunc() error {
  2. if value, err := pack1.Func1(param1); err != nil {
  3. return err
  4. }
  5. return nil
  6. }

这种模式也常用于通过 defer 使程序从 panic 中恢复执行(参考第 17.2(4) 小节)。

要实现简洁的错误检测代码,更好的方式是使用闭包,参考第 16.10.2 小节

(2)检测映射中是否存在一个键值(参考第 8.2 小节):key1 在映射 map1 中是否有值?

  1. if value, isPresent = map1[key1]; isPresent {
  2. Process(value)
  3. }
  4. // key1 不存在

(3)检测一个接口类型变量 varI 是否包含了类型 T:类型断言(参考第 11.3 小节):

  1. if value, ok := varI.(T); ok {
  2. Process(value)
  3. }
  4. // 接口类型 varI 没有包含类型 T

(4)检测一个通道 ch 是否关闭(参考第 14.3 小节):

  1. for input := range ch {
  2. Process(input)
  3. }

或者:

  1. for {
  2. if input, open := <-ch; !open {
  3. break // 通道是关闭的
  4. }
  5. Process(input)
  6. }

链接