Golang学习笔记之延迟函数(defer)的使用小结

脚本专栏 2024/12/25 佚名

3 2 1

golang的defer优雅又简洁，是golang的亮点之一。defer在声明时不会立即执行，而是在函数return后，再按照先进后出的原则依次执行每个defer，一般用于释放资源、清理数据、记录日志、异常处理等。

关键字defer于注册延迟调用。这些调用直到 ret 前才被执行，通常用于释放资源或错误处理。

一、当defer被声明时，其参数就会被实时解析

func a() {
  i := 0
  defer fmt.Println(i) //输出0，因为i此时就是0
  i++
  defer fmt.Println(i) //输出1，因为i此时就是1
  return
}

二、当一个函数内多次调用 defer 时，Go 会把 defer 调用放入到一个栈中，随后按照先进后出的顺序执行。

func main() {
  defer fmt.Println("1")
  defer fmt.Println("2")
  defer fmt.Println("3")
  defer fmt.Println("4")
}

输出顺序为4321

使用defer逆序输出字符串

name := "Naveen你好"
  fmt.Printf("%s\n", string(name))
  fmt.Printf("逆序:")
  defer fmt.Printf("\n")
  for _, v := range []rune(name) {
    defer fmt.Printf("%c", v)
  }

输出：

Naveen你好

逆序:好你neevaN

三、defer 的实际应用

func (r rect) area(wg *sync.WaitGroup) {
  if r.length < 0 {
    fmt.Printf("rect %v's length should be greater than zero\n", r)
    wg.Done()
    return
  }
  if r.width < 0 {
    fmt.Printf("rect %v's width should be greater than zero\n", r)
    wg.Done()
    return
  }
  area := r.length * r.width
  fmt.Printf("rect %v's area %d\n", r, area)
  wg.Done()
}

我们会发现上面wg.Done()被调用多次我们可以使用defer来优化代码

func (r rect) area(wg *sync.WaitGroup) {
  defer wg.Done()
  if r.length < 0 {
    fmt.Printf("rect %v's length should be greater than zero\n", r)
    return
  }
  if r.width < 0 {
    fmt.Printf("rect %v's width should be greater than zero\n", r)
    return
  }
  area := r.length * r.width
  fmt.Printf("rect %v's area %d\n", r, area)
}

使用defer延时调用，会在程序结束之前调用

下面举个例子：

package main 
import "fmt" 
func deferTest(number int) int {
 defer func() {
 number++
 fmt.Println("three:", number)
 }()
 
 defer func() {
 number++
 fmt.Println("two:", number)
 }()
 
 defer func() {
 number++
 fmt.Println("one:", number)
 }()
 
 return number
}
 
func main() {
 fmt.Println("函数返回值：", deferTest(0))
}

上面的代码打印的结果是：