欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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

程序员文章站 2023-10-11 12:33:19
golang的defer优雅又简洁, 是golang的亮点之一。defer在声明时不会立即执行,而是在函数return后,再按照先进后出的原则依次执行每个defer,一般用...

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))
}

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

one: 1
two: 2
three: 3
函数返回值: 0

ps:defer 有一个重要的特性,即便函数抛出了异常,defer也会被执行的。这样就不会因为程序出现了错误,导致资源不会被释放了。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。