解决Go语言Websocket应用程序内存泄漏的方法

func MyHandler(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println(err)
        return
    }

    go func() {
        for {
            mt, message, err := conn.ReadMessage()
            if err != nil {
                log.Println(err)
                break
            }
            log.Printf("recv: %s", message)
            // 处理消息逻辑

            // 如果不再需要该连接，可以调用conn.Close()关闭连接
            if shouldClose {
                conn.Close()
                break
            }
        }
    }()
}

var wg sync.WaitGroup

func MyHandler(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println(err)
        return
    }

    wg.Add(1)
    go func() {
        defer wg.Done()
        // 处理连接逻辑
    }()
}

func main() {
    http.HandleFunc("/", MyHandler)
    go http.ListenAndServe(":8080", nil)

    // 等待所有连接处理完成
    wg.Wait()
}

import (
    "os"
    "runtime/pprof"
)

func main() {
    f, err := os.Create("heap_profile.pprof")
    if err != nil {
        log.Fatal(err)
    }
    defer f.Close()

    pprof.WriteHeapProfile(f)
}