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