Go语言中关键字defer使用方法

在Go语言中，可以使用关键字defer向函数注册退出调用，即主函数退出时，defer后的函数才被调用。defer语句的作用是不管程序是否出现异常，均在函数退出时自动执行相关代码。

但是，你还可以使用defer在任何函数开始后和结束前执行配对的代码。这个隐藏的功能在网上的教程和书籍中很少提到。要使用此功能，需要创建一个函数并使它本身返回另一个函数，返回的函数将作为真正的延迟函数。在 defer 语句调用父函数后在其上添加额外的括号来延迟执行返回的子函数如下所示：

func main() {
 defer greet()()
 fmt.Println("Some code here...")
}

func greet() func() {
 fmt.Println("Hello!")
 return func() { fmt.Println("Bye!") } // this will be deferred
}

输出以下内容：

Hello!
Some code here...
Bye!

父函数返回的函数将是实际的延迟函数。父函数中的其他代码将在函数开始时（由 defer 语句放置的位置决定）立即执行。

这为开发者提供了什么能力？因为在函数内定义的匿名函数可以访问完整的词法环境（lexical environment），这意味着在函数中定义的内部函数可以引用该函数的变量。在下一个示例中看到的，参数变量在measure函数第一次执行和其延迟执行的子函数内都能访问到：

func main() {
 example()
 otherExample()
}

func example(){
 defer measure("example")()
 fmt.Println("Some code here")
}

func otherExample(){
 defer measure("otherExample")()
 fmt.Println("Some other code here")
}

func measure(name string) func() {
 start := time.Now()
 fmt.Printf("Starting function %s\n", name)
 return func(){ fmt.Printf("Exiting function %s after %s\n", name, time.Since(start)) }
}

输出以下内容：

Starting example
Some code here
Exiting example after 0s
Starting otherExample
Some other code here
Exiting otherExample after 0s

此外函数命名的返回值也是函数内的局部变量，所以上面例子中的measure函数如果接收命名返回值作为参数的话，那么命名返回值在延迟执行的函数中访问到，这样就能将measure函数改造成记录入参和返回值的工具函数。

下面的示例是引用《go 语言程序设计》中的代码段:

func bigSlowOperation() {
 defer trace("bigSlowOperation")() // don't forget the extra parentheses
 // ...lots of work…
 time.Sleep(10 * time.Second) // simulate slow
 operation by sleeping
}
func trace(msg string) func() {
 start := time.Now()
 log.Printf("enter %s", msg)
 return func() {
   log.Printf("exit %s (%s)", msg,time.Since(start))
 }
}

可以想象，将代码延迟在函数的入口和出口使用是非常有用的功能，尤其是在调试代码的时候。

以上就是良许教程网为各位朋友分享的Linu系统相关内容。想要了解更多Linux相关知识记得关注公众号“良许Linux”，或扫描下方二维码进行关注，更多干货等着你 ！