Golang中Gin框架结合WebSocket处理高并发-Golang-PHP中文网

必须显式校验origin，生产环境严禁return true或不配置checkorigin；需用精确白名单比对（如https://app.example.com），禁用前缀匹配；同时应设置readbuffersize和writebuffersize为65536以降低gc压力。

golang中gin框架结合websocket处理高并发

gorilla/websocket.Upgrader 配置必须显式校验 Origin

生产环境不设 CheckOrigin 或直接返回 true，等于把 WebSocket 接口暴露给任意域名——攻击者可伪造 Origin 发起连接，劫持用户会话或发起 CSRF 式广播。这不是“开发阶段才管”的事，而是上线前必须堵死的入口。

正确做法是白名单校验，比如只允许前端部署在 https://app.example.com 和 https://staging.app.example.com：

var upgrader = websocket.Upgrader{
    CheckOrigin: func(r *http.Request) bool {
        origin := r.Header.Get("Origin")
        return origin == "https://app.example.com" ||
               origin == "https://staging.app.example.com"
    },
}

不要用正则或字符串前缀匹配（如 strings.HasPrefix(origin, "https://example.com")），易被绕过
若前端走 CDN 或多子域，建议统一提取 Host 或从 JWT claim 中解析可信 domain
开发时可用 os.Getenv("ENV") == "dev" 区分环境，但禁止硬编码 return true

ReadBufferSize / WriteBufferSize 不设默认值会放大内存压力

gorilla/websocket 默认 ReadBufferSize 和 WriteBufferSize 均为 0，触发 runtime 按需分配小缓冲区（通常 4KB 以下）。高并发下每连接都 malloc 小块内存，GC 压力陡增，且频繁拷贝导致 CPU 占用飙升。

实测中，设为 65536（64KB）能显著降低分配频次和 GC pause：

var upgrader = websocket.Upgrader{
    ReadBufferSize:  65536,
    WriteBufferSize: 65536,
}

若业务消息普遍
缓冲区大小不影响协议帧解析逻辑，只影响底层 io.ReadFull 和 bufio.Writer 行为
别盲目设过大（如 1MB），单连接内存占用翻倍，连接数上万时 OOM 风险剧增

Upgrade 后必须立刻设 ReadDeadline / WriteDeadline

没设 deadline 的 *websocket.Conn 在 ReadMessage() 或 WriteMessage() 上可能永久阻塞——网络抖动、客户端假死、NAT 超时都会让 goroutine 卡住，最终耗尽 stack 内存或触发调度器饥饿。

Gin框架 1.9.0

Gin框架 1.9.0版本源码包下载，版本号 1.9.0，适合需要 sonic JSON 支持、路由修复和内容协商改进的 Go Web 开发场景。

下载

典型错误写法：升级完就开 goroutine 读消息，却不设超时：

// ❌ 危险
go func() {
    for {
        _, msg, err := conn.ReadMessage()
        if err != nil { /* 忽略错误处理 */ }
        // ...
    }
}()

正确姿势是每次读/写前重置 deadline（尤其读操作）：

conn.SetReadDeadline(time.Now().Add(30 * time.Second))
_, msg, err := conn.ReadMessage()
if err != nil {
    if !websocket.IsCloseError(err, websocket.CloseAbnormalClosure) {
        log.Printf("read error: %v", err)
    }
    return
}
// 处理消息后，再次设置下一轮读超时
conn.SetReadDeadline(time.Now().Add(30 * time.Second))