Go语言bytes.makeSlice导致内存泄漏的根本原因是什么？

Go语言bytes.MakeSlice与内存泄漏：案例分析

在Go语言网络编程中，内存泄漏是一个常见问题。本文分析一个案例，探讨bytes.MakeSlice与内存泄漏的潜在关联，并指出问题的根源并非bytes.MakeSlice本身。

案例：一个使用Fiber框架构建的HTTP服务器，处理/test路径的请求。服务器端在/test路由的处理函数中，使用bytes.NewBufferString创建一个缓冲区，并写入大量数据。客户端并发发送多个请求。使用go tool pprof分析堆内存，发现bytes.MakeSlice占据大量内存，且持续增长，未释放。

服务器端代码（简化版）：

立即学习“go语言免费学习笔记（深入）”；

LobeHub

LobeChat brings you the best user experience of ChatGPT, OLLaMA, Gemini, Claude

下载

package main

import (
    "bytes"

    "github.com/gofiber/fiber/v2"
    "github.com/gofiber/fiber/v2/middleware/pprof"
)

func main() {
    app := fiber.New()
    app.Use(pprof.New())
    app.Get("/test", func(c *fiber.Ctx) error {
        buffer := bytes.NewBufferString("")
        // ...写入大量数据到buffer...
        return c.SendString(buffer.String()) // 直接发送buffer内容
    })
    app.Listen(":3000")
}

客户端代码（简化版）：

package main

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

func main() {
    var wg sync.WaitGroup
    for i := 0; i < 1000; i++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            resp, err := http.Get("http://localhost:3000/test")
            if err != nil {
                fmt.Println("Error:", err)
                return
            }
            // 关键：缺少resp.Body.Close()
            //resp.Body.Close()
            // ...处理resp.Body...
        }()
    }
    wg.Wait()
}

go tool pprof分析结果显示bytes.MakeSlice占据大量内存。然而，这并非bytes.MakeSlice本身的问题。 bytes.MakeSlice是bytes.Buffer内部使用的函数，用于调整缓冲区大小。 真正的根源在于客户端代码中缺少resp.Body.Close()。未关闭的响应体导致服务器端分配的内存（bytes.Buffer）无法被垃圾回收，从而造成内存泄漏。 这最终体现在pprof结果中，显示bytes.MakeSlice相关的内存占用高。

解决方法： 在客户端代码中添加resp.Body.Close()，确保正确释放资源。 改进后的客户端代码如下：

// ... (其他代码不变) ...
resp, err := http.Get("http://localhost:3000/test")
if err != nil {
    fmt.Println("Error:", err)
    return
}
defer resp.Body.Close() // 添加这行代码
// ...处理resp.Body...

通过正确关闭响应体，可以避免内存泄漏，解决go tool pprof显示的bytes.MakeSlice相关内存问题。 这强调了资源管理在Go语言编程中的重要性。