golang 之 defer（统计函数执行时间）

package main

import (
    "fmt"
    "time"
)

func sum(a ...int) int {
    defer trace("sum")() // note:不要忘记defer语句后的圆括号，否则本该在进入时执行的操作会在退出时执行，而本该在退出时执行的，永远不会执行
    total := 0
    for _, val := range a {
        total += val
    }
    return total
}

func trace(msg string) func() {
    start := time.Now()
    fmt.Printf("enter %s
", msg)
    return func() {
        fmt.Printf("exit %s (%s)
", msg, time.Since(start))
    }
}

func main() {
    count := sum(3, 5, 9)
    fmt.Printf("%d
", count)
}

执行结果：

enter sum
exit sum (194.764µs)
17

 利用defer修改函数的返回值：

package main

import (
    "fmt"
    "time"
)

func sum(a, b int) (result int) {
    //defer trace("sum")() // note:不要忘记defer语句后的圆括号，否则本该在进入时执行的操作会在退出时执行，而本该在退出时执行的，永远不会执行
    defer func() { result += a }() //被延时执行的匿名函数甚至可以修改函数返回给调用者的返回值
    result = 0
    result += a
    result += b
    return result
}

func trace(msg string) func() {
    start := time.Now()
    fmt.Printf("enter %s
", msg)
    return func() {
        fmt.Printf("exit %s (%s)
", msg, time.Since(start))
    }
}

func main() {
    count := sum(3, 8)
    fmt.Printf("%d
", count)
}

执行结果：

[root@docker pro]# go run arg.go 
[root@docker pro]# go run arg.go 
14