使用WriteBufferPool时Gorilla Websocket WriteMessage引发内存溢出

问题原因 #

你配置了WriteBufferPool但内存消耗无变化，核心问题在于两处不匹配：

1. 直接将sync.Pool赋值给Upgrader.WriteBufferPool，但该字段要求的是websocket库定义的WriteBufferPool接口类型，两者类型不兼容。
2. 从你贴出的beginMessage代码来看，内部期望从池中获取私有类型writePoolData，但你的sync.Pool返回的是[]byte，导致类型断言wpd, ok := c.writePool.Get().(writePoolData)失败，最终每次都会重新创建新缓冲区，池完全未起到复用作用。

解决方案 #

1. 正确实现WriteBufferPool接口 #

以gorilla/websocket为例，需自定义实现WriteBufferPool接口的结构体，内部用sync.Pool管理缓冲区：

type bufferPool struct {
    pool sync.Pool
}

// Get 从池中获取缓冲区
func (p *bufferPool) Get() []byte {
    buf, ok := p.pool.Get().([]byte)
    if !ok {
        return make([]byte, 1024)
    }
    return buf
}

// Put 将缓冲区放回池中
func (p *bufferPool) Put(buf []byte) {
    // 只回收指定容量的缓冲区，避免池内混入不同大小的缓冲
    if cap(buf) != 1024 {
        return
    }
    p.pool.Put(buf)
}

// 初始化Upgrader时使用自定义的缓冲池
var upgrader = websocket.Upgrader{
    ReadBufferSize:    1024,
    WriteBufferPool:   &bufferPool{
        pool: sync.Pool{
            New: func() interface{} {
                return make([]byte, 1024)
            },
        },
    },
    EnableCompression: false,
}

2. 适配库内部类型要求（针对你贴出的beginMessage逻辑） #

如果你的websocket库确实要求返回writePoolData类型（尽管它是私有），不建议依赖私有类型的实现细节，优先查看库的官方文档或升级到最新稳定版本，避免后续版本迭代导致代码失效。

3. 检查消息写入逻辑 #

你当前的WriteMessage调用逻辑本身无问题，只要池配置正确，websocket库会在消息写入完成后自动将缓冲区放回池中。如果连接频繁创建销毁，可能降低池的复用效率，可结合连接池进一步优化。

内容的提问来源于stack exchange，提问作者Sanket Bijawe