如何关闭 HTTP 的响应体？

八股文_Go基础 0 29

参考回答

在 Go 中，使用 http 包发起 HTTP 请求后，必须关闭 HTTP 响应体（Response.Body），否则会导致资源泄露，尤其是对于长时间运行的程序（如服务器）或高并发请求场景。

关闭响应体的方法是调用 defer resp.Body.Close()，通常紧跟在 HTTP 请求之后，确保资源会被及时释放。

详细讲解与示例

1. 为什么需要关闭响应体？

资源管理：
- HTTP 响应体（Response.Body）是一个流，需要手动关闭来释放底层的网络连接和文件描述符。
连接复用：
- 在使用 HTTP 长连接时（默认开启），关闭响应体后连接才会被复用。如果未关闭，连接可能会耗尽，导致性能下降或程序崩溃。

2. 如何正确关闭响应体？

使用 defer 关键字确保响应体在不需要时被关闭，即使函数中出现错误或提前返回。

示例：

package main

import (
    "fmt"
    "io"
    "net/http"
)

func main() {
    // 发起 HTTP 请求
    resp, err := http.Get("https://example.com")
    if err != nil {
        fmt.Println("Error:", err)
        return
    }
    // 确保响应体被关闭
    defer resp.Body.Close()

    // 读取并打印响应体
    body, err := io.ReadAll(resp.Body)
    if err != nil {
        fmt.Println("Error reading body:", err)
        return
    }

    fmt.Println(string(body))
}

3. 注意事项

(1) 确保总是关闭

无论是否读取或处理响应体，都需要关闭。
使用 defer resp.Body.Close() 是最佳实践，因为它确保函数退出时关闭响应体，即使发生错误。

(2) 忽略响应体内容

如果你不需要读取响应体内容，可以使用 io.Copy 或 io.Discard 快速读取并丢弃内容。
这样可以避免连接被挂起。

示例：忽略响应体内容

package main

import (
    "io"
    "net/http"
)

func main() {
    resp, err := http.Get("https://example.com")
    if err != nil {
        panic(err)
    }
    defer resp.Body.Close()

    // 丢弃响应体
    io.Copy(io.Discard, resp.Body)
}

(3) 在自定义请求中

对于 http.Client 的自定义请求（如 POST 请求），关闭响应体的方式相同。

示例：自定义 POST 请求

package main

import (
    "bytes"
    "fmt"
    "io"
    "net/http"
)

func main() {
    client := &http.Client{}
    data := []byte(`{"key": "value"}`)
    req, err := http.NewRequest("POST", "https://example.com", bytes.NewBuffer(data))
    if err != nil {
        fmt.Println("Error creating request:", err)
        return
    }

    resp, err := client.Do(req)
    if err != nil {
        fmt.Println("Error making request:", err)
        return
    }
    defer resp.Body.Close()

    body, err := io.ReadAll(resp.Body)
    if err != nil {
        fmt.Println("Error reading body:", err)
        return
    }

    fmt.Println(string(body))
}

4. 常见错误及避免方法

未关闭响应体：
- 如果 defer resp.Body.Close() 被遗忘，连接不会释放，最终可能耗尽资源。

错误使用 defer：

在循环中使用 defer 可能导致资源释放延迟，应在每次循环结束时手动关闭响应体。

示例（错误的写法）：

for _, url := range urls {
 resp, err := http.Get(url)
 if err != nil {
     continue
 }
 defer resp.Body.Close() // 不要在循环中使用 defer
}

正确的写法：
“`go
for _, url := range urls {
resp, err := http.Get(url)
if err != nil {
continue
}
io.Copy(io.Discard, resp.Body)
resp.Body.Close() // 手动关闭
}
“`

总结

关闭响应体的方法：
- 使用 defer resp.Body.Close() 是推荐的方式，可以确保函数退出时释放资源。
注意事项：
- 即使不处理响应体内容，也需要关闭它（如使用 io.Copy(io.Discard, resp.Body)）。
- 在循环中避免直接使用 defer，需手动关闭每次响应体。
最佳实践：
- 无论是否读取响应体，始终在请求完成后关闭 Body，避免资源泄漏和连接复用问题。