Gorilla WebSocket NextWriter与WriteMessage差异及并发写入Panic排查

问题背景 #

我编写了如下write函数：

func write() {
    defer func() {
        serverConn.Close()
    }()
    for message := range msgChan {
        w, err := serverConn.NextWriter(websocket.TextMessage)
        if err != nil {
            return
        }
        bmessage, err := json.Marshal(message)
        if err != nil {
            return
        }

        _, err = w.Write(bmessage)
        if err != nil {
            fmt.Println(err)
        }

        if err := w.Close(); err != nil {
            return
        }
    }
}

运行时出现如下panic信息：

panic: concurrent write to websocket connection

我有两个疑问：

1. 明明只有这一个单实例的写函数负责WebSocket写入，为何会出现并发写入的panic？
2. 使用NextWriter而非conn.WriteMessage的意义是什么？当消息量较大时，NextWriter是否会堆积并导致并发写入？

问题解答 #

疑问1：为何出现并发写入panic？ #

你认为只有单个写函数操作连接，但实际可能存在以下情况：

• 存在其他写入路径：检查代码中是否有其他地方调用了serverConn的写入方法（比如WriteMessage、NextWriter，甚至直接操作底层net.Conn），哪怕是错误处理、超时逻辑里的隐性写入，都会触发并发写入panic。
• 写函数被多次启动：确认write()是否被多次执行（比如重复启动go write()），导致多个goroutine同时进入循环，各自操作同一个连接。
• 错误处理漏洞：当前代码中，若w.Write(bmessage)出错仅打印错误却未终止当前循环或关闭writer，后续执行w.Close()时，若存在其他并发写入操作（比如意外启动的goroutine），也可能引发冲突，但前两种情况概率更高。

疑问2：NextWriter的意义与消息量问题 #

• NextWriter与WriteMessage的区别：
  • WriteMessage是封装好的一次性写入接口，内部流程就是调用NextWriter、写入数据、关闭writer，适合简单的单次消息发送。
  • NextWriter的核心优势是支持分批写入单个消息帧：比如发送大文件时，无需将整个文件加载到内存，可分多次调用w.Write()写入，最后用w.Close()完成帧发送，大幅降低内存占用；另外，若需设置自定义WebSocket帧头（如扩展字段），也可通过该writer操作。
• 消息量大时的并发风险：
  NextWriter本身不会导致并发写入。只要所有写入操作（无论用NextWriter还是WriteMessage）都在同一个goroutine执行，或通过sync.Mutex严格串行化，就不会出现并发问题。消息堆积仅指msgChan内消息过多，但单例写入goroutine会按顺序处理，每个NextWriter的创建、写入、关闭流程都是串行的，不会出现多个writer同时操作连接的情况。

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