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