golang defer

Posted 2023-03-08 yunweigo

Go语言中的defer关键字用于在函数返回前执行一些特定的操作。可以将defer看作是一种后置语句，在函数中的任何位置都可以使用。

下面是一个使用defer的例子：

func foo() 
    defer fmt.Println("Done")
    fmt.Println("Hello")

在上面的例子中，当函数foo被调用时，它会先输出"Hello"，然后再输出"Done"，因为"Done"被包装在defer语句中，会在函数返回前执行。

如果在一个函数中有多个defer语句，它们的执行顺序是后进先出的，也就是说，最后一个defer语句会最先执行，而第一个defer语句会最后执行。

例如，下面的代码中有三个defer语句，它们的执行顺序是3、2、1：

func bar() 
    defer fmt.Println("1")
    defer fmt.Println("2")
    defer fmt.Println("3")

在Go语言中，defer关键字可以用于以下几个应用场景：

资源管理：使用defer语句可以确保资源（如文件、网络连接等）被及时关闭和释放，以避免资源泄漏和占用过多的系统资源。

延迟函数调用：通过defer语句可以将函数调用延迟到当前函数返回之前执行，这在一些需要在函数执行结束前执行一些清理或者统计操作的场景下很有用。

错误处理：通过defer语句可以捕获和处理一些可能发生的错误，确保程序在出现异常情况时也能够正常退出并清理资源。

统计和调试：通过defer语句可以在函数执行过程中记录一些统计信息或者调试信息，以便在需要时进行分析和排查。

应用实例1 错误异常的处理

func readFile(filename string) (string, error) 
    file, err := os.Open(filename)
    if err != nil 
        return "", err
    
    defer func() 
        if err := file.Close(); err != nil 
            log.Println("Error closing file:", err)
        
    ()
    content, err := ioutil.ReadAll(file)
    if err != nil 
        return "", err
    
    return string(content), nil

在上面的例子中，我们在打开文件后使用了defer语句将文件关闭的代码放在了一个匿名函数中。如果文件读取操作发生错误，这个匿名函数会在函数返回前被执行，确保文件被关闭并且任何可能的资源泄漏被避免。

需要注意的是，如果在defer语句中使用了函数参数，需要在defer语句执行的时候将它们赋值为函数返回值。在上面的例子中，我们需要将file.Close()的返回值赋给err变量，以便检查文件是否成功关闭。

应用实例2 代码统计和调试中的使用

func processRequest(req *http.Request) (*http.Response, error) 
    startTime := time.Now()

    defer func() 
        log.Printf("Request processed in %v", time.Since(startTime))
    ()

    // Process the request here
    // ...

    resp, err := http.DefaultClient.Do(req)
    if err != nil 
        return nil, err
    

    return resp, nil

在上面的示例代码中，我们使用defer语句记录了函数执行的开始时间，然后在函数返回前打印出函数执行的耗时。这样可以方便地进行性能分析和调试，以便优化代码并找出潜在的性能瓶颈。当函数返回时，defer语句会被执行，打印出函数执行的耗时。由于defer语句的特性，无论函数执行过程中是否发生了异常，这个defer语句都会被执行，确保我们能够正确地记录函数的执行时间。

应用实例3 HTTP请求中的异常处理

func handleRequest(w http.ResponseWriter, r *http.Request) 
    // 将处理异常的代码放在defer语句中
    defer func() 
        if r := recover(); r != nil 
            log.Printf("Recovered from panic: %v", r)
            http.Error(w, "Internal Server Error", http.StatusInternalServerError)
        
    ()

    // 处理HTTP请求
    // ...

    // 在代码中可能会发生异常
    // ...

在上面的示例代码中，我们在处理HTTP请求的代码中可能会发生异常，为了确保在异常发生时仍能够释放资源和清理状态，我们使用defer语句将异常处理的代码放在了一个匿名函数中。当发生异常时，defer语句会被执行，捕获异常并打印出异常信息，并向客户端返回一个500错误响应。

需要注意的是，如果在defer语句中使用了函数参数，需要在defer语句执行的时候将它们赋值为函数返回值。在上面的示例代码中，我们需要将recover()的返回值赋给r变量，以便检查是否发生了异常。同时，由于Go语言中的异常处理机制使用较少，因此在处理HTTP请求时应该特别注意异常情况的处理。

注意

需要注意的是，defer语句应该被谨慎使用，特别是在性能敏感的场景下。因为defer语句会增加代码的执行时间和内存消耗，特别是在循环和递归等高频执行的代码中，过多的defer语句可能会影响代码的性能和稳定性。