Go defer使用

 

defer使用语法

//defer后面必须是函数调用语句或方法调用语句,不能是其他语句,否则编译器会出错。
package main
import (
	"fmt"
)
func foo(n int) int {
	defer n++ 
	//defer fmt.Println(n)
	return n
}
func main() {
	var i int = 100
	foo(i)
}
# command-line-arguments
expression in defer must be function call
syntax error: unexpected ++ at end of statement

  

 

defer 语句的用途

含有 defer 语句的函数,会在该函数将要返回之前,调用另一个函数。

defer后面的函数在defer语句所在的函数执行结束的时候会被调用;

package main
import (  
    "fmt"
)

type person struct {  
    firstName string
    lastName string
}

func (p person) fullName() {  
    fmt.Printf("%s %s",p.firstName,p.lastName)
}

func main() {  
    p := person {
        firstName: "John",
        lastName: "Smith",
    }
    defer p.fullName()
    fmt.Printf("Welcome ")  
}

在上面的程序里的第 11 行,a 的初始值为 5。在第 12 行执行 defer 语句的时候,由于 a 等于 5,因此延迟函数 printA 的实参也等于 5。接着我们在第 13 行将 a 的值修改为 10。下一行会打印出 a 的值。该程序输出:

value of a before deferred function call 10  
value of a in deferred function 5

从上面的输出,我们可以看出,在调用了 defer 语句后,虽然我们将 a 修改为 10,但调用延迟函数 printA(a)后,仍然打印的是 5。

 

defer 栈

当一个函数内多次调用 defer 时,Go 会把 defer 调用放入到一个栈中,随后按照后进先出(Last In First Out, LIFO)的顺序执行。

我们下面编写一个小程序,使用 defer 栈,将一个字符串逆序打印。

package main
import (  
    "fmt"
)

func main() {  
    name := "Naveen"
    fmt.Printf("Orignal String: %s\n", string(name))
    fmt.Printf("Reversed String: ")
    for _, v := range []rune(name) {
        defer fmt.Printf("%c", v)
    }}
Orignal String: Naveen
Reversed String: neevaN

  

defer 的实际应用

  • file对象打开后的自动关闭
func CopyFile(dstName, srcName string) (written int64, err error) {
    src, err := os.Open(srcName)
    if err != nil {
        return
    }
    defer src.Close()

    dst, err := os.Create(dstName)
    if err != nil {
        return
    }
    defer dst.Close()

    // other codes
    return io.Copy(dst, src)
}

 在打开输入文件输出文件后,不管后面的代码流程如何影响,这两个文件能够被自动关闭。 

  • mutex对象锁住后的自动释放
func foo(...) {
    mu.Lock()
    defer mu.Unlock()

    // code logic
}

确保mu锁能够在函数foo退出之后自动释放。

  • WaitGroup的退出
package main
import (
	"fmt"
	"sync"
)

type rect struct {
	length int
	width  int 
}

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

func main() {
	var wg sync.WaitGroup
	r1 := rect{-67, 89}
	r2 := rect{5, -67}
	r3 := rect{8, 9}
	rects := []rect{r1, r2, r3}
	for _, v := range rects {
		wg.Add(1)
		go v.area(&wg)
	}
	wg.Wait()
	fmt.Println("All go routines finished executing")
}

  

 

 

上一篇:defer 延时执行 用于【灯】开灯和关灯写到一起


下一篇:only和defer与select_related和prefetch_Related(面试重点)