Go语言bytes.makeSlice导致内存泄漏:如何避免服务器端大内存占用?

Go语言bytes.makeSlice导致内存泄漏:如何避免服务器端大内存占用?

go语言bytes.makeSlice与内存泄漏:剖析及解决方案

高效的内存管理对于Go语言程序至关重要。本文将分析一个案例,探讨bytes.makeSlice函数与内存泄漏的关联,并提供有效的解决方案。

问题描述:一个基于Fiber框架的Go http服务器,其/test路由生成一个包含百万个”123″字符串的大型字节缓冲区并返回给客户端。当客户端并发发起大量请求时,go tool pprof分析显示bytes.makeSlice占据大量内存,且程序结束后内存未能完全释放。

服务器端代码(示例):

立即学习go语言免费学习笔记(深入)”;

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("")         for i := 0; i < 1000000; i++ {             buffer.WriteString("123")         }         return c.SendString(buffer.String())     })     app.Listen(":3000") }

客户端代码(示例):

package main  import (     "fmt"     "io"     "net/http"     "sync" )  func main() {     var wg sync.WaitGroup     for i := 0; i < 500; 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             }             defer resp.Body.Close() //关键:关闭响应体             io.Copy(io.Discard, resp.Body)         }()     }     wg.Wait() }

go tool pprof分析结果表明bytes.makeSlice占据大量内存。根本原因在于:服务器端每次请求都创建大型字节缓冲区,而客户端未正确关闭响应体(resp.Body.Close()),导致服务器端分配的内存无法被垃圾回收。

问题根源:客户端代码缺少resp.Body.Close()。resp.Body是io.ReadCloser,使用完毕后必须调用Close()释放底层资源,从而允许服务器释放相关内存。 忽略Close()会导致内存泄漏,即使bytes.makeSlice本身没有问题。

解决方案:在客户端代码中,确保在读取完响应体后调用resp.Body.Close(),正确释放资源,避免内存泄漏。 改进后的客户端代码如上所示,包含了defer resp.Body.Close()。 这行代码确保在函数返回前,无论是否发生错误,响应体都被正确关闭。

以上就是Go语言bytes.makeSlice导致内存泄漏:如何避免服务器端大

© 版权声明
THE END
喜欢就支持一下吧
点赞8 分享